Source code for appbase.MainWindow

# -*- coding: utf-8 -*-
#this pkg:
from mainWindowRessources.menubar import MenuBar
from appbase.Application import Application
    
#foreign:
from PyQt4 import QtGui


[docs]class MainWindow(QtGui.QMainWindow):
    '''
    template for QMainWindow including:
    
    * a menu bar with all common features
    * fullscreen with F11
    * changed window title when saved under a new name
    * Preferences in Menubar->File
    * autosave
    * close dialog
    '''
    def __init__(self, title=''):
        super(MainWindow, self).__init__()
        self.app = QtGui.QApplication.instance()
        if not isinstance(self.app, Application):
            print 'Error: QApp is no instance from appbase.Application'
            return
        fn = lambda state: self.setTitlePath(self.app.session.path)
        
        self.app.session.sigRestore.connect(fn)
        self.app.session.sigSave.connect(fn)

        self._window_title = title
        self._window_title_additive = ''
        self._window_title_path = ''
        m = MenuBar()
        self.setMenuBar(m)
        self._setTitle()
        #Shortcuts
        sc = m.shortcutsWidget
        sc.addShortcut('Alt+F4', 'Close current window')
        sc.addShortcut('F11', 'Fullscreen')
        k  = QtGui.QKeySequence
        sc.addShortcut(k(k.Undo).toString(), 'Undo last action')
        sc.addShortcut(k(k.Redo).toString(), 'Redo last action')
        sc.addShortcut(k(k.Save).toString(), 'Save current session')
        s = k(k.SaveAs).toString()
        if s:
            #not every OS has this shortcut:
            sc.addShortcut(s, 'Save current session under a new name')
    

[docs]    def setTitleAdditive(self, value=''):
        if value:
            self._window_title_additive = '- %s' %value
        else:
            self._window_title_additive = value        
        self._setTitle()


[docs]    def setTitlePath(self, value=''):
        if value:
            self._window_title_path = '- %s' %value
        else:
            self._window_title_path = ''
        self._setTitle()
        
        
    def _setTitle(self):
        self.setWindowTitle('%s %s %s' %(self._window_title, 
                                        self._window_title_additive, 
                                        self._window_title_path) )


[docs]    def closeEvent(self, evt):
        if self.app.session.app_opts['showCloseDialog']:
            b = _CloseDialog(self)
            ret = b.exec_()
            if ret == QtGui.QMessageBox.Save:
                self.app.session.save()
            elif ret == QtGui.QMessageBox.Cancel:
                return evt.ignore()
        return evt.accept()
        
            
            
class _CloseDialog(QtGui.QMessageBox):    
    def __init__(self, mainWindow):
        QtGui.QMessageBox.__init__(self, mainWindow)
        self.setIcon(QtGui.QMessageBox.Warning)
        self.setText("Close the program...")
        self.setInformativeText("Save changes?")
        self.setStandardButtons(QtGui.QMessageBox.Save | QtGui.QMessageBox.Discard | QtGui.QMessageBox.Cancel)


        c = QtGui.QCheckBox("don't ask me again")
        c.clicked.connect(lambda val: mainWindow.app.session.app_opts.__setitem__('showCloseDialog', not val))
        self.layout().addWidget(c,4,0,7,0)




if __name__ == '__main__':
    def save(session):
        print('saveTest')
        print(session.path)

    def restore(session):
        print ('restore')
    
    import sys
    app = Application([])
    win = MainWindow(title='Hello World')
    #CONNECT OWN SAVE/RESTORE FUNCTIONS TO THE SESSION
    app.session.sigSave.connect(save)
    app.session.sigRestore.connect(restore)
    
    win.show()
    sys.exit(app.exec_())