X-Git-Url: http://git.onelab.eu/?p=sface.git;a=blobdiff_plain;f=sface%2Fmainwindow.py;h=cefec17f11d9b4e277bc022b2284baed840101b8;hp=c5e0183358b5900424a82b9f20a120d8d85e8631;hb=ca5d17301939ddd993f658a3acae09d1aa56e5c9;hpb=306a8cababce34b45507e7c1d9b05fe0486c56af diff --git a/sface/mainwindow.py b/sface/mainwindow.py index c5e0183..cefec17 100644 --- a/sface/mainwindow.py +++ b/sface/mainwindow.py @@ -1,16 +1,16 @@ import os, os.path import sys import time +import traceback from PyQt4.QtCore import * from PyQt4.QtGui import * +import sface.screens from sface.config import config from sface.logwindow import LogWindow from sface.rspecwindow import RSpecWindow -from sface.screens.configscreen import ConfigScreen -from sface.screens.mainscreen import MainScreen -from sface.screens.helpscreen import HelpScreen +from sface.screens.sfascreen import SfaScreen # depending on the platform.. # could probably use Qt's resource system but looks overkill for just one file... @@ -20,25 +20,57 @@ def locate_image_file (filename): attempt=os.path.join(dir,'images',"%s.%s"%(filename,suffix)) if os.path.isfile(attempt) : return attempt return os.path.join('/could/not/locate/image/file',filename) - + +def load_screens(dirname): + modnames = [] + for fn in os.listdir(dirname): + if not fn.endswith(".py"): + continue + modname = fn.rsplit(".py",1)[0] + if modname == "sfascreen": + # ignore this, it's the base class, not a screen + continue + modnames.append(modname) + + # we want the stock screens to show up in a specific order. plugins can + # show up in any order afterward. + + sort_order = ["mainscreen", "configscreen", "helpscreen"] + sort_order.reverse() + for modname in sort_order: + if modname in modnames: + modnames.remove(modname) + modnames.insert(0,modname) + + # import each module and find whatever class(es) is descendant from + # SfaScreen within the module. Might be a better idea to just define a + # screens=[] variable in each module + + screens = [] + for modname in modnames: + try: + mod = __import__("sface.screens." + modname, fromlist=["modname"]) + except: + print "Exception while importing screen", modname + traceback.print_exc() + continue + + for object in dir(mod): + object = getattr(mod, object) + if hasattr(object, "__bases__") and sface.screens.sfascreen.SfaScreen in object.__bases__: + screens.append(object) + + return screens class Nav(QWidget): - def __init__(self, parent=None): + def __init__(self, screens, parent=None): QWidget.__init__(self, parent) - + self.title = QLabel("", self) scene=QGraphicsScene() pixmap = QPixmap(locate_image_file('graphic-sfa64')) logolabel=QLabel("",self) logolabel.setPixmap(pixmap) - self.main = QLabel("", self) - self.main.setAlignment(Qt.AlignRight) - - self.config = QLabel("", self) - self.config.setAlignment(Qt.AlignRight) - - self.help = QLabel("", self) - self.help.setAlignment(Qt.AlignRight) hlayout = QHBoxLayout() hlayout.addWidget(logolabel) @@ -47,19 +79,19 @@ class Nav(QWidget): gotolabel=QLabel("Go to: ", self) gotolabel.setAlignment(Qt.AlignRight) hlayout.addWidget(gotolabel) - hlayout.addWidget(self.main) - hlayout.addWidget(self.config) - hlayout.addWidget(self.help) + + self.screenLabels = [] + for screen in screens: + label = QLabel(screen.getLinkText(), self) + label.setAlignment(Qt.AlignRight) + self.screenLabels.append(label) + hlayout.addWidget(label) + self.setLayout(hlayout) def setTitle(self, title): self.title.setText(title) - def setLinks(self, main, config, help): - self.main.setText(main) - self.config.setText(config) - self.help.setText(help) - class Status(QLabel): def __init__(self, parent=None): @@ -93,22 +125,21 @@ class MainWindow(QWidget): self.rspecWindow = RSpecWindow(self) self.pix = QLabel(self) - self.config_screen = ConfigScreen(self) - self.main_screen = MainScreen(self) - self.help_screen = HelpScreen(self) + + screenClasses = load_screens(os.path.dirname(sface.screens.__file__)) + self.screenWidgets = [] self.screens = QStackedWidget(self) - self.screens.addWidget(self.main_screen) - self.screens.addWidget(self.config_screen) - self.screens.addWidget(self.help_screen) + for screen in screenClasses: + screenWidget = screen(self) + self.screenWidgets.append(screenWidget) + self.screens.addWidget(screenWidget) + self.screens.addWidget(self.pix) self.next_screen = None - self.nav = Nav(self) - self.nav.setTitle(self.main_screen.getTitleText()) - self.nav.setLinks(self.main_screen.getLinkText(), - self.config_screen.getLinkText(), - self.help_screen.getLinkText()) + self.nav = Nav(self.screenWidgets, self) + self.nav.setTitle(self.screenWidgets[0].getTitleText()) self.status = Status(self) self.log = QLabel("Show Log", self) @@ -127,7 +158,7 @@ class MainWindow(QWidget): self.setLayout(layout) self.resize(800, 500) - for link in (self.nav.main, self.nav.config, self.nav.help): + for link in self.nav.screenLabels: self.connect(link, SIGNAL('linkActivated(QString)'), self.animateToScreen) @@ -156,12 +187,9 @@ class MainWindow(QWidget): self.pix.move(0, y) def animateToScreen(self, link): - if link == self.config_screen.name: - self.next_screen = self.config_screen - elif link == self.main_screen.name: - self.next_screen = self.main_screen - elif link == self.help_screen.name: - self.next_screen = self.help_screen + for screen in self.screenWidgets: + if link == screen.name: + self.next_screen = screen curr_screen = self.screens.currentWidget() @@ -185,24 +213,8 @@ class MainWindow(QWidget): def toNextScreen(self): - if self.next_screen == self.config_screen: - self.toConfigScreen() - elif self.next_screen == self.main_screen: - self.toMainScreen() - elif self.next_screen == self.help_screen: - self.toHelpScreen() - - def toConfigScreen(self): - self.screens.setCurrentWidget(self.config_screen) - self.nav.setTitle(self.config_screen.getTitleText()) - - def toHelpScreen(self): - self.screens.setCurrentWidget(self.help_screen) - self.nav.setTitle(self.help_screen.getTitleText()) - - def toMainScreen(self): - self.screens.setCurrentWidget(self.main_screen) - self.nav.setTitle(self.main_screen.getTitleText()) + self.screens.setCurrentWidget(self.next_screen) + self.nav.setTitle(self.next_screen.getTitleText()) def setStatus(self, msg, timeout): self.status.set(msg, timeout)