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-sfa.png'))
  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         hlayout.addWidget(QLabel("Go to: ", self))
  48         hlayout.addWidget(self.main)
  49         hlayout.addWidget(self.config)
  50         hlayout.addWidget(self.help)
  51         self.setLayout(hlayout)
  52
  53     def setTitle(self, title):
  54         self.title.setText(title)
  55
  56     def setLinks(self, main, config, help):
  57         self.main.setText(main)
  58         self.config.setText(config)
  59         self.help.setText(help)
  60
  61
  62 class Status(QLabel):
  63     def __init__(self, parent=None):
  64         QLabel.__init__(self, "", parent)
  65         self.sliceUpdateDate()
  66
  67     def set(self, msg, timeout):
  68         self.setText(msg)
  69         if timeout:
  70             QTimer.singleShot(timeout, self.reset)
  71
  72     def sliceUpdateDate(self):
  73         rspec_file = config.getSliceRSpecFile()
  74         if not os.path.exists(rspec_file):
  75             return
  76
  77         creation_time = os.stat(rspec_file).st_ctime
  78         last_update = time.ctime(creation_time)
  79         self.set("Slice data last updated on %s" % last_update, timeout=None)
  80
  81     def reset(self):
  82         self.setText("")
  83         QTimer.singleShot(1500, self.sliceUpdateDate)
  84
  85
  86 class MainWindow(QWidget):
  87     def __init__(self, parent=None):
  88         QWidget.__init__(self, parent)
  89
  90         self.logWindow = LogWindow(self)
  91         self.rspecWindow = RSpecWindow(self)
  92
  93         self.pix = QLabel(self)
  94         self.config_screen = ConfigScreen(self)
  95         self.main_screen = MainScreen(self)
  96         self.help_screen = HelpScreen(self)
  97
  98         self.screens = QStackedWidget(self)
  99         self.screens.addWidget(self.main_screen)
 100         self.screens.addWidget(self.config_screen)
 101         self.screens.addWidget(self.help_screen)
 102         self.screens.addWidget(self.pix)
 103         self.next_screen = None
 104
 105         self.nav = Nav(self)
 106         self.nav.setTitle(self.main_screen.getTitleText())
 107         self.nav.setLinks(self.main_screen.getLinkText(),
 108                           self.config_screen.getLinkText(),
 109                           self.help_screen.getLinkText())
 110
 111         self.status = Status(self)
 112         self.log = QLabel("<a href='showlog'>Show Log</a>", self)
 113         self.rspec = QLabel("<a href='showlog'>Show RSpec</a>", self)
 114
 115         hlayout = QHBoxLayout()
 116         hlayout.addWidget(self.status)
 117         hlayout.addStretch()
 118         hlayout.addWidget(self.rspec)
 119         hlayout.addWidget(self.log)
 120
 121         layout = QVBoxLayout()
 122         layout.addWidget(self.nav)
 123         layout.addWidget(self.screens)
 124         layout.addLayout(hlayout)
 125         self.setLayout(layout)
 126         self.resize(800, 500)
 127
 128         for link in (self.nav.main, self.nav.config, self.nav.help):
 129             self.connect(link, SIGNAL('linkActivated(QString)'),
 130                          self.animateToScreen)
 131
 132         self.connect(self.log, SIGNAL('linkActivated(QString)'),
 133                      self.showLogWindow)
 134         self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
 135                      self.showRSpecWindow)
 136
 137     def redirectOutputToLog(self):
 138         self.logWindow.redirectOutput()
 139
 140     def showLogWindow(self, link):
 141         self.logWindow.show()
 142         self.logWindow.resize(800, 200)
 143         self.logWindow.raise_()
 144         self.logWindow.activateWindow()
 145
 146     def showRSpecWindow(self, link):
 147         self.rspecWindow.show()
 148         self.rspecWindow.resize(500, 640)
 149         self.rspecWindow.raise_()
 150         self.rspecWindow.activateWindow()
 151
 152
 153     def animatePixmap(self, y):
 154         self.pix.move(0, y)
 155
 156     def animateToScreen(self, link):
 157         if link == self.config_screen.name:
 158             self.next_screen = self.config_screen
 159         elif link == self.main_screen.name:
 160             self.next_screen = self.main_screen
 161         elif link == self.help_screen.name:
 162             self.next_screen = self.help_screen
 163
 164         curr_screen = self.screens.currentWidget()
 165
 166         if self.next_screen == curr_screen:
 167             self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
 168             return
 169
 170         # This is an optimization to have a smoother animation. We
 171         # render the widget into a pixmap and animate that instead of
 172         # moving the whole widget around.
 173         pixmap = QPixmap(self.screens.size())
 174         curr_screen.render(pixmap)
 175         self.screens.setCurrentWidget(self.pix)
 176         self.pix.setPixmap(pixmap)
 177
 178         timeLine = QTimeLine(500, self)
 179         timeLine.setFrameRange(0, self.screens.height());
 180         self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
 181         self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
 182         timeLine.start()
 183
 184
 185     def toNextScreen(self):
 186         if self.next_screen == self.config_screen:
 187             self.toConfigScreen()
 188         elif self.next_screen == self.main_screen:
 189             self.toMainScreen()
 190         elif self.next_screen == self.help_screen:
 191             self.toHelpScreen()
 192
 193     def toConfigScreen(self):
 194         self.screens.setCurrentWidget(self.config_screen)
 195         self.nav.setTitle(self.config_screen.getTitleText())
 196
 197     def toHelpScreen(self):
 198         self.screens.setCurrentWidget(self.help_screen)
 199         self.nav.setTitle(self.help_screen.getTitleText())
 200
 201     def toMainScreen(self):
 202         self.screens.setCurrentWidget(self.main_screen)
 203         self.nav.setTitle(self.main_screen.getTitleText())
 204
 205     def setStatus(self, msg, timeout):
 206         self.status.set(msg, timeout)
 207
 208     def nodeSelectionChanged(self, hostname):
 209         if self.rspecWindow.isVisible():
 210             self.rspecWindow.showNode(hostname)