sface/mainwindow.py

   1 import os, os.path
   2 import sys
   3 import time
   4
   5 from PyQt4.QtCore import *
   6 from PyQt4.QtGui import *
   7
   8 from sface.config import config
   9 from sface.logwindow import LogWindow
  10 from sface.rspecwindow import RSpecWindow
  11 from sface.screens.configscreen import ConfigScreen
  12 from sface.screens.mainscreen import MainScreen
  13 from sface.screens.helpscreen import HelpScreen
  14
  15 # depending on the platform..
  16 # could probably use Qt's resource system but looks overkill for just one file...
  17 def locate_image_file (filename):
  18     for dir in [ '/usr/share', '/Applications/sface.app/Contents/Resources/sface' ] :
  19         for suffix in ['png','jpg']:
  20             attempt=os.path.join(dir,'images',"%s.%s"%(filename,suffix))
  21             if os.path.isfile(attempt) : return attempt
  22     return os.path.join('/could/not/locate/image/file',filename)
  23
  24
  25 class Nav(QWidget):
  26     def __init__(self, parent=None):
  27         QWidget.__init__(self, parent)
  28
  29         self.title = QLabel("", self)
  30         scene=QGraphicsScene()
  31         pixmap = QPixmap(locate_image_file('graphic-sfa64'))
  32         logolabel=QLabel("",self)
  33         logolabel.setPixmap(pixmap)
  34         self.main = QLabel("", self)
  35         self.main.setAlignment(Qt.AlignRight)
  36
  37         self.config = QLabel("", self)
  38         self.config.setAlignment(Qt.AlignRight)
  39
  40         self.help = QLabel("", self)
  41         self.help.setAlignment(Qt.AlignRight)
  42
  43         hlayout = QHBoxLayout()
  44         hlayout.addWidget(logolabel)
  45         hlayout.addWidget(self.title)
  46         hlayout.addStretch()
  47         gotolabel=QLabel("Go to: ", self)
  48         gotolabel.setAlignment(Qt.AlignRight)
  49         hlayout.addWidget(gotolabel)
  50         hlayout.addWidget(self.main)
  51         hlayout.addWidget(self.config)
  52         hlayout.addWidget(self.help)
  53         self.setLayout(hlayout)
  54
  55     def setTitle(self, title):
  56         self.title.setText(title)
  57
  58     def setLinks(self, main, config, help):
  59         self.main.setText(main)
  60         self.config.setText(config)
  61         self.help.setText(help)
  62
  63
  64 class Status(QLabel):
  65     def __init__(self, parent=None):
  66         QLabel.__init__(self, "", parent)
  67         self.sliceUpdateDate()
  68
  69     def set(self, msg, timeout):
  70         self.setText(msg)
  71         if timeout:
  72             QTimer.singleShot(timeout, self.reset)
  73
  74     def sliceUpdateDate(self):
  75         rspec_file = config.getSliceRSpecFile()
  76         if not os.path.exists(rspec_file):
  77             return
  78
  79         creation_time = os.stat(rspec_file).st_ctime
  80         last_update = time.ctime(creation_time)
  81         self.set("Slice data last updated on %s" % last_update, timeout=None)
  82
  83     def reset(self):
  84         self.setText("")
  85         QTimer.singleShot(1500, self.sliceUpdateDate)
  86
  87
  88 class MainWindow(QWidget):
  89     def __init__(self, parent=None):
  90         QWidget.__init__(self, parent)
  91
  92         self.logWindow = LogWindow(self)
  93         self.rspecWindow = RSpecWindow(self)
  94
  95         self.pix = QLabel(self)
  96         self.config_screen = ConfigScreen(self)
  97         self.main_screen = MainScreen(self)
  98         self.help_screen = HelpScreen(self)
  99
 100         self.screens = QStackedWidget(self)
 101         self.screens.addWidget(self.main_screen)
 102         self.screens.addWidget(self.config_screen)
 103         self.screens.addWidget(self.help_screen)
 104         self.screens.addWidget(self.pix)
 105         self.next_screen = None
 106
 107         self.nav = Nav(self)
 108         self.nav.setTitle(self.main_screen.getTitleText())
 109         self.nav.setLinks(self.main_screen.getLinkText(),
 110                           self.config_screen.getLinkText(),
 111                           self.help_screen.getLinkText())
 112
 113         self.status = Status(self)
 114         self.log = QLabel("<a href='showlog'>Show Log</a>", self)
 115         self.rspec = QLabel("<a href='showlog'>Show RSpec</a>", self)
 116
 117         hlayout = QHBoxLayout()
 118         hlayout.addWidget(self.status)
 119         hlayout.addStretch()
 120         hlayout.addWidget(self.rspec)
 121         hlayout.addWidget(self.log)
 122
 123         layout = QVBoxLayout()
 124         layout.addWidget(self.nav)
 125         layout.addWidget(self.screens)
 126         layout.addLayout(hlayout)
 127         self.setLayout(layout)
 128         self.resize(800, 500)
 129
 130         for link in (self.nav.main, self.nav.config, self.nav.help):
 131             self.connect(link, SIGNAL('linkActivated(QString)'),
 132                          self.animateToScreen)
 133
 134         self.connect(self.log, SIGNAL('linkActivated(QString)'),
 135                      self.showLogWindow)
 136         self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
 137                      self.showRSpecWindow)
 138
 139     def redirectOutputToLog(self):
 140         self.logWindow.redirectOutput()
 141
 142     def showLogWindow(self, link):
 143         self.logWindow.show()
 144         self.logWindow.resize(800, 200)
 145         self.logWindow.raise_()
 146         self.logWindow.activateWindow()
 147
 148     def showRSpecWindow(self, link):
 149         self.rspecWindow.show()
 150         self.rspecWindow.resize(500, 640)
 151         self.rspecWindow.raise_()
 152         self.rspecWindow.activateWindow()
 153
 154
 155     def animatePixmap(self, y):
 156         self.pix.move(0, y)
 157
 158     def animateToScreen(self, link):
 159         if link == self.config_screen.name:
 160             self.next_screen = self.config_screen
 161         elif link == self.main_screen.name:
 162             self.next_screen = self.main_screen
 163         elif link == self.help_screen.name:
 164             self.next_screen = self.help_screen
 165
 166         curr_screen = self.screens.currentWidget()
 167
 168         if self.next_screen == curr_screen:
 169             self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
 170             return
 171
 172         # This is an optimization to have a smoother animation. We
 173         # render the widget into a pixmap and animate that instead of
 174         # moving the whole widget around.
 175         pixmap = QPixmap(self.screens.size())
 176         curr_screen.render(pixmap)
 177         self.screens.setCurrentWidget(self.pix)
 178         self.pix.setPixmap(pixmap)
 179
 180         timeLine = QTimeLine(500, self)
 181         timeLine.setFrameRange(0, self.screens.height());
 182         self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
 183         self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
 184         timeLine.start()
 185
 186
 187     def toNextScreen(self):
 188         if self.next_screen == self.config_screen:
 189             self.toConfigScreen()
 190         elif self.next_screen == self.main_screen:
 191             self.toMainScreen()
 192         elif self.next_screen == self.help_screen:
 193             self.toHelpScreen()
 194
 195     def toConfigScreen(self):
 196         self.screens.setCurrentWidget(self.config_screen)
 197         self.nav.setTitle(self.config_screen.getTitleText())
 198
 199     def toHelpScreen(self):
 200         self.screens.setCurrentWidget(self.help_screen)
 201         self.nav.setTitle(self.help_screen.getTitleText())
 202
 203     def toMainScreen(self):
 204         self.screens.setCurrentWidget(self.main_screen)
 205         self.nav.setTitle(self.main_screen.getTitleText())
 206
 207     def setStatus(self, msg, timeout):
 208         self.status.set(msg, timeout)
 209
 210     def nodeSelectionChanged(self, hostname):
 211         if self.rspecWindow.isVisible():
 212             self.rspecWindow.showNode(hostname)