hydrus/hydrus/client/gui/ClientGUIFrames.py

import os

from qtpy import QtWidgets as QW

from hydrus.core import HydrusGlobals as HG

from hydrus.client import ClientConstants as CC
from hydrus.client.gui import ClientGUITopLevelWindows
from hydrus.client.gui import QtPorting as QP

class ShowKeys( ClientGUITopLevelWindows.Frame ):
    
    def __init__( self, key_type, keys ):
        
        if key_type == 'registration': title = 'Registration Tokens'
        elif key_type == 'access': title = 'Access Keys'
        
        tlw = HG.client_controller.GetMainTLW()
        
        ClientGUITopLevelWindows.Frame.__init__( self, tlw, HG.client_controller.PrepStringForDisplay( title ) )
        
        self._key_type = key_type
        self._keys = keys
        
        #
        
        self._text_ctrl = QW.QPlainTextEdit( self )
        self._text_ctrl.setLineWrapMode( QW.QPlainTextEdit.NoWrap )
        self._text_ctrl.setReadOnly( True )
        
        self._save_to_file = QW.QPushButton( 'save to file', self )
        self._save_to_file.clicked.connect( self.EventSaveToFile )
        
        self._done = QW.QPushButton( 'done', self )
        self._done.clicked.connect( self.close )
        
        #
        
        if key_type == 'registration': prepend = 'r'
        else: prepend = ''
        
        self._text = os.linesep.join( [ prepend + key.hex() for key in self._keys ] )
        
        self._text_ctrl.setPlainText( self._text )
        
        #
        
        vbox = QP.VBoxLayout()
        
        QP.AddToLayout( vbox, self._text_ctrl, CC.FLAGS_EXPAND_BOTH_WAYS )
        QP.AddToLayout( vbox, self._save_to_file, CC.FLAGS_ON_RIGHT )
        QP.AddToLayout( vbox, self._done, CC.FLAGS_ON_RIGHT )
        
        self.setLayout( vbox )
        
        size_hint = self.sizeHint()
        
        size_hint.setWidth( max( size_hint.width(), 500 ) )
        size_hint.setHeight( max( size_hint.height(), 200 ) )
        
        QP.SetInitialSize( self, size_hint )
        
        self.show()
        
    
    def EventSaveToFile( self ):
        
        filename = 'keys.txt'
        
        with QP.FileDialog( self, acceptMode = QW.QFileDialog.AcceptSave, fileMode = QW.QFileDialog.AnyFile, default_filename = filename ) as dlg:
            
            if dlg.exec() == QW.QDialog.Accepted:
                
                path = dlg.GetPath()
                
                with open( path, 'w', encoding = 'utf-8' ) as f:
                    
                    f.write( self._text )