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