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
  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.trackerWindow = init_tracker(parent)
 131
 132         self.pix = QLabel(self)
 133
 134         screenClasses = load_screens(os.path.dirname(sface.screens.__file__))
 135         self.screenWidgets = []
 136
 137         self.screens = QStackedWidget(self)
 138         for screen in screenClasses:
 139             # use a try/catch block to isolate the screen. Third-party plugins
 140             # could be buggy.
 141             try:
 142                 screenWidget = screen(self)
 143                 self.screenWidgets.append(screenWidget)
 144                 self.screens.addWidget(screenWidget)
 145             except:
 146                 print "Exception while creating screen", screen.__name__
 147                 traceback.print_exc()
 148
 149         self.screens.addWidget(self.pix)
 150         self.next_screen = None
 151
 152         self.nav = Nav(self.screenWidgets, self)
 153
 154         if self.screenWidgets:
 155             self.nav.setTitle(self.screenWidgets[0].getTitleText())
 156
 157         self.status = Status(self)
 158         self.tracker = QLabel("<a href='showtracker'>Show Xmlrpc</a>", self)
 159         self.log = QLabel("<a href='showlog'>Show Log</a>", self)
 160         self.rspec = QLabel("<a href='showlog'>Show RSpec</a>", self)
 161
 162         hlayout = QHBoxLayout()
 163         hlayout.addWidget(self.status)
 164         hlayout.addStretch()
 165         hlayout.addWidget(self.tracker)
 166         hlayout.addWidget(self.rspec)
 167         hlayout.addWidget(self.log)
 168
 169         layout = QVBoxLayout()
 170         layout.addWidget(self.nav)
 171         layout.addWidget(self.screens)
 172         layout.addLayout(hlayout)
 173         self.setLayout(layout)
 174         self.resize(800, 500)
 175
 176         for link in self.nav.screenLabels:
 177             self.connect(link, SIGNAL('linkActivated(QString)'),
 178                          self.animateToScreen)
 179
 180         self.connect(self.tracker, SIGNAL('linkActivated(QString)'),
 181                      self.showTrackerWindow)
 182         self.connect(self.log, SIGNAL('linkActivated(QString)'),
 183                      self.showLogWindow)
 184         self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
 185                      self.showRSpecWindow)
 186
 187     def redirectOutputToLog(self):
 188         self.logWindow.redirectOutput()
 189
 190     def showTrackerWindow(self):
 191         tracker = get_tracker()
 192         tracker.show()
 193         tracker.resize(500, 640)
 194         tracker.raise_()
 195         tracker.activateWindow()
 196
 197     def showLogWindow(self, link):
 198         self.logWindow.show()
 199         self.logWindow.resize(800, 200)
 200         self.logWindow.raise_()
 201         self.logWindow.activateWindow()
 202
 203     def showRSpecWindow(self, link):
 204         self.rspecWindow.show()
 205         self.rspecWindow.resize(500, 640)
 206         self.rspecWindow.raise_()
 207         self.rspecWindow.activateWindow()
 208
 209     def animatePixmap(self, y):
 210         self.pix.move(0, y)
 211
 212     def animateToScreen(self, link):
 213         for screen in self.screenWidgets:
 214             if link == screen.name:
 215                 self.next_screen = screen
 216
 217         curr_screen = self.screens.currentWidget()
 218
 219         if self.next_screen == curr_screen:
 220             self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
 221             return
 222
 223         # This is an optimization to have a smoother animation. We
 224         # render the widget into a pixmap and animate that instead of
 225         # moving the whole widget around.
 226         pixmap = QPixmap(self.screens.size())
 227         curr_screen.render(pixmap)
 228         self.screens.setCurrentWidget(self.pix)
 229         self.pix.setPixmap(pixmap)
 230
 231         timeLine = QTimeLine(500, self)
 232         timeLine.setFrameRange(0, self.screens.height());
 233         self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
 234         self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
 235         timeLine.start()
 236
 237     def toNextScreen(self):
 238         self.screens.setCurrentWidget(self.next_screen)
 239         self.nav.setTitle(self.next_screen.getTitleText())
 240
 241     def setStatus(self, msg, timeout):
 242         self.status.set(msg, timeout)
 243
 244     def nodeSelectionChanged(self, hostname):
 245         if self.rspecWindow.isVisible():
 246             self.rspecWindow.showNode(hostname)
 247
 248     def closeEvent(self, event):
 249         # give the screens an opportunity to veto the close
 250         for screen in self.screenWidgets:
 251             if not screen.canClose():
 252                 event.ignore()
 253                 return
 254
 255         # give the screens an opportunity to close gracefully
 256         for screen in self.screenWidgets:
 257             screen.mainWindowClose()
 258
 259         QWidget.closeEvent(self, event)
 260