sface/mainwindow.py

   1 import os, os.path
   2 import sys
   3 import time
   4 import traceback
   5
   6 from PyQt4.QtCore import *
   7 from PyQt4.QtGui import *
   8
   9 import sface.screens
  10 from sface.config import config
  11 from sface.logwindow import LogWindow
  12 from sface.rspecwindow import RSpecWindow, ResourcesWindow
  13 from sface.screens.sfascreen import SfaScreen
  14 from sface.xmlrpcwindow import get_tracker, init_tracker
  15
  16 # depending on the platform..
  17 # could probably use Qt's resource system but looks overkill for just one file...
  18 def locate_image_file (filename):
  19     for dir in [ '/usr/share', '/Applications/sface.app/Contents/Resources/sface' ] :
  20         for suffix in ['png','jpg']:
  21             attempt=os.path.join(dir,'images',"%s.%s"%(filename,suffix))
  22             if os.path.isfile(attempt) : return attempt
  23     return os.path.join('/could/not/locate/image/file',filename)
  24
  25 def load_screens(dirname):
  26     modnames = []
  27     for fn in os.listdir(dirname):
  28         if not fn.endswith(".py"):
  29             continue
  30         modname = fn.rsplit(".py",1)[0]
  31         if modname == "sfascreen":
  32             # ignore this, it's the base class, not a screen
  33             continue
  34         modnames.append(modname)
  35
  36     # we want the stock screens to show up in a specific order. plugins can
  37     # show up in any order afterward.
  38
  39     sort_order = ["mainscreen", "userscreen", "configscreen", "helpscreen"]
  40     sort_order.reverse()
  41     for modname in sort_order:
  42         if modname in modnames:
  43             modnames.remove(modname)
  44             modnames.insert(0,modname)
  45
  46     # import each module and find whatever class(es) is descendant from
  47     # SfaScreen within the module. Might be a better idea to just define a
  48     # screens=[] variable in each module
  49
  50     screens = []
  51     for modname in modnames:
  52         try:
  53             mod = __import__("sface.screens." + modname, fromlist=["modname"])
  54         except:
  55             print "Exception while importing screen", modname
  56             traceback.print_exc()
  57             continue
  58
  59         for object in dir(mod):
  60             object = getattr(mod, object)
  61             if hasattr(object, "__bases__") and sface.screens.sfascreen.SfaScreen in object.__bases__:
  62                 screens.append(object)
  63
  64     return screens
  65
  66 class Nav(QWidget):
  67     def __init__(self, screens, parent=None):
  68         QWidget.__init__(self, parent)
  69
  70         self.title = QLabel("", self)
  71         scene=QGraphicsScene()
  72         pixmap = QPixmap(locate_image_file('graphic-sfa64'))
  73         logolabel=QLabel("",self)
  74         logolabel.setPixmap(pixmap)
  75
  76         hlayout = QHBoxLayout()
  77         hlayout.addWidget(logolabel)
  78         hlayout.addWidget(self.title)
  79         hlayout.addStretch()
  80         gotolabel=QLabel("Go to: ", self)
  81         gotolabel.setAlignment(Qt.AlignRight)
  82         hlayout.addWidget(gotolabel)
  83
  84         self.screenLabels = []
  85         for screen in screens:
  86             label = QLabel(screen.getLinkText(), self)
  87             label.setAlignment(Qt.AlignRight)
  88             self.screenLabels.append(label)
  89             hlayout.addWidget(label)
  90
  91         self.setLayout(hlayout)
  92
  93     def setTitle(self, title):
  94         self.title.setText(title)
  95
  96
  97 class Status(QLabel):
  98     def __init__(self, parent=None):
  99         QLabel.__init__(self, "", parent)
 100         self.setMaximumWidth(640)
 101         self.sliceUpdateDate()
 102
 103     def set(self, msg, timeout):
 104         self.setText(msg)
 105         if timeout:
 106             QTimer.singleShot(timeout, self.reset)
 107
 108     def sliceUpdateDate(self):
 109         rspec_file = config.getSliceRSpecFile()
 110         if not os.path.exists(rspec_file):
 111             return
 112
 113         creation_time = os.stat(rspec_file).st_ctime
 114         last_update = time.ctime(creation_time)
 115         self.set("Slice data last refreshed on %s" % last_update, timeout=None)
 116
 117     def reset(self):
 118         self.setText("")
 119         QTimer.singleShot(1500, self.sliceUpdateDate)
 120
 121
 122 class MainWindow(QWidget):
 123     def __init__(self, parent=None):
 124         QWidget.__init__(self, parent)
 125
 126         # These are top-level windows and should be initialized with parent set
 127         # to our parent. Otherwise, getting a segfault on exit in Ubuntu.
 128         self.logWindow = LogWindow(parent)
 129         self.rspecWindow = RSpecWindow(parent)
 130         self.resourcesWindow = ResourcesWindow(parent)
 131         self.trackerWindow = init_tracker(parent)
 132
 133         self.pix = QLabel(self)
 134
 135         screenClasses = load_screens(os.path.dirname(sface.screens.__file__))
 136         self.screenWidgets = []
 137
 138         self.screens = QStackedWidget(self)
 139         for screen in screenClasses:
 140             # use a try/catch block to isolate the screen. Third-party plugins
 141             # could be buggy.
 142             try:
 143                 screenWidget = screen(self)
 144                 self.screenWidgets.append(screenWidget)
 145                 self.screens.addWidget(screenWidget)
 146             except:
 147                 print "Exception while creating screen", screen.__name__
 148                 traceback.print_exc()
 149
 150         self.screens.addWidget(self.pix)
 151         self.next_screen = None
 152
 153         self.nav = Nav(self.screenWidgets, self)
 154
 155         if self.screenWidgets:
 156             self.nav.setTitle(self.screenWidgets[0].getTitleText())
 157
 158         self.status = Status(self)
 159         self.tracker = QLabel("<a href='showtracker'>Xmlrpc</a>", self)
 160         self.log = QLabel("<a href='showlog'>Log</a>", self)
 161         self.rspec = QLabel("<a href='showlog'>RSpec</a>", self)
 162         self.resources = QLabel("<a href='showlog'>Resources</a>", self)
 163
 164         hlayout = QHBoxLayout()
 165         hlayout.addWidget(self.status)
 166         hlayout.addStretch()
 167         hlayout.addWidget(self.tracker)
 168         hlayout.addWidget(self.rspec)
 169         hlayout.addWidget(self.resources)
 170         hlayout.addWidget(self.log)
 171
 172         layout = QVBoxLayout()
 173         layout.addWidget(self.nav)
 174         layout.addWidget(self.screens)
 175         layout.addLayout(hlayout)
 176         self.setLayout(layout)
 177         self.resize(800, 500)
 178
 179         for link in self.nav.screenLabels:
 180             self.connect(link, SIGNAL('linkActivated(QString)'),
 181                          self.animateToScreen)
 182
 183         self.connect(self.tracker, SIGNAL('linkActivated(QString)'),
 184                      self.showTrackerWindow)
 185         self.connect(self.log, SIGNAL('linkActivated(QString)'),
 186                      self.showLogWindow)
 187         self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
 188                      self.showRSpecWindow)
 189         self.connect(self.resources, SIGNAL('linkActivated(QString)'),
 190                      self.showResourcesWindow)
 191
 192     def redirectOutputToLog(self):
 193         self.logWindow.redirectOutput()
 194
 195     def showTrackerWindow(self):
 196         tracker = get_tracker()
 197         tracker.show()
 198         tracker.resize(500, 640)
 199         tracker.raise_()
 200         tracker.activateWindow()
 201
 202     def showLogWindow(self, link):
 203         self.logWindow.show()
 204         self.logWindow.resize(800, 200)
 205         self.logWindow.raise_()
 206         self.logWindow.activateWindow()
 207
 208     def showRSpecWindow(self, link):
 209         self.rspecWindow.show()
 210         self.rspecWindow.resize(500, 640)
 211         self.rspecWindow.raise_()
 212         self.rspecWindow.activateWindow()
 213
 214     def showResourcesWindow(self, link):
 215         self.resourcesWindow.show()
 216         self.resourcesWindow.resize(500, 640)
 217         self.resourcesWindow.raise_()
 218         self.resourcesWindow.activateWindow()
 219
 220     def animatePixmap(self, y):
 221         self.pix.move(0, y)
 222
 223     def animateToScreen(self, link):
 224         for screen in self.screenWidgets:
 225             if link == screen.name:
 226                 self.next_screen = screen
 227
 228         curr_screen = self.screens.currentWidget()
 229
 230         if self.next_screen == curr_screen:
 231             self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
 232             return
 233
 234         # This is an optimization to have a smoother animation. We
 235         # render the widget into a pixmap and animate that instead of
 236         # moving the whole widget around.
 237         pixmap = QPixmap(self.screens.size())
 238         curr_screen.render(pixmap)
 239         self.screens.setCurrentWidget(self.pix)
 240         self.pix.setPixmap(pixmap)
 241
 242         timeLine = QTimeLine(500, self)
 243         timeLine.setFrameRange(0, self.screens.height());
 244         self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
 245         self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
 246         timeLine.start()
 247
 248     def toNextScreen(self):
 249         self.screens.setCurrentWidget(self.next_screen)
 250         self.nav.setTitle(self.next_screen.getTitleText())
 251
 252     def setStatus(self, msg, timeout):
 253         self.status.set(msg, timeout)
 254
 255     def nodeSelectionChanged(self, hostname):
 256         if self.rspecWindow.isVisible():
 257             self.rspecWindow.showNode(hostname)
 258
 259     def closeEvent(self, event):
 260         # give the screens an opportunity to veto the close
 261         for screen in self.screenWidgets:
 262             if not screen.canClose():
 263                 event.ignore()
 264                 return
 265
 266         # give the screens an opportunity to close gracefully
 267         for screen in self.screenWidgets:
 268             screen.mainWindowClose()
 269
 270         QWidget.closeEvent(self, event)
 271