sface/mainwindow.py

   1
   2 import os
   3 import sys
   4 import time
   5
   6 from PyQt4.QtCore import *
   7 from PyQt4.QtGui import *
   8
   9 from sface.config import config
  10 from sface.logwindow import LogWindow
  11 from sface.rspecwindow import RSpecWindow
  12 from sface.screens.configscreen import ConfigScreen
  13 from sface.screens.mainscreen import MainScreen
  14
  15 class Nav(QWidget):
  16     def __init__(self, parent=None):
  17         QWidget.__init__(self, parent)
  18
  19         self.title = QLabel("", self)
  20         self.main = QLabel("", self)
  21         self.main.setAlignment(Qt.AlignRight)
  22
  23         self.config = QLabel("", self)
  24         self.config.setAlignment(Qt.AlignRight)
  25
  26         hlayout = QHBoxLayout()
  27         hlayout.addWidget(self.title)
  28         hlayout.addStretch()
  29         hlayout.addWidget(QLabel("Go to: ", self))
  30         hlayout.addWidget(self.main)
  31         hlayout.addWidget(self.config)
  32         self.setLayout(hlayout)
  33
  34     def setTitle(self, title):
  35         self.title.setText(title)
  36
  37     def setLinks(self, main, config):
  38         self.main.setText(main)
  39         self.config.setText(config)
  40
  41
  42 class Status(QLabel):
  43     def __init__(self, parent=None):
  44         QLabel.__init__(self, "", parent)
  45         self.sliceUpdateDate()
  46
  47     def set(self, msg, timeout):
  48         self.setText(msg)
  49         if timeout:
  50             QTimer.singleShot(timeout, self.reset)
  51
  52     def sliceUpdateDate(self):
  53         rspec_file = config.getSliceRSpecFile()
  54         if not os.path.exists(rspec_file):
  55             return
  56
  57         creation_time = os.stat(rspec_file).st_ctime
  58         last_update = time.ctime(creation_time)
  59         self.set("Slice data last updated on %s" % last_update, timeout=None)
  60
  61     def reset(self):
  62         self.setText("")
  63         QTimer.singleShot(1500, self.sliceUpdateDate)
  64
  65 class MainWindow(QWidget):
  66     def __init__(self, parent=None):
  67         QWidget.__init__(self, parent)
  68
  69         self.logWindow = LogWindow(self)
  70         self.rspecWindow = RSpecWindow(self)
  71
  72         self.pix = QLabel(self)
  73         self.config_screen = ConfigScreen(self)
  74         self.main_screen = MainScreen(self)
  75
  76         self.screens = QStackedWidget(self)
  77         self.screens.addWidget(self.main_screen)
  78         self.screens.addWidget(self.config_screen)
  79         self.screens.addWidget(self.pix)
  80         self.next_screen = None
  81
  82         self.nav = Nav(self)
  83         self.nav.setTitle(self.main_screen.getTitleText())
  84         self.nav.setLinks(self.main_screen.getLinkText(),
  85                           self.config_screen.getLinkText())
  86
  87         self.status = Status(self)
  88         self.log = QLabel("<a href='showlog'>Show Log</a>", self)
  89         self.rspec = QLabel("<a href='showlog'>Show RSpec</a>", self)
  90
  91         hlayout = QHBoxLayout()
  92         hlayout.addWidget(self.status)
  93         hlayout.addStretch()
  94         hlayout.addWidget(self.rspec)
  95         hlayout.addWidget(self.log)
  96
  97         layout = QVBoxLayout()
  98         layout.addWidget(self.nav)
  99         layout.addWidget(self.screens)
 100         layout.addLayout(hlayout)
 101         self.setLayout(layout)
 102         self.resize(800, 600)
 103
 104         for link in (self.nav.main, self.nav.config):
 105             self.connect(link, SIGNAL('linkActivated(QString)'),
 106                          self.animateToScreen)
 107
 108         self.connect(self.log, SIGNAL('linkActivated(QString)'),
 109                      self.showLogWindow)
 110         self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
 111                      self.showRSpecWindow)
 112
 113     def redirectOutputToLog(self):
 114         self.logWindow.redirectOutput()
 115
 116     def showLogWindow(self, link):
 117         self.logWindow.show()
 118         self.logWindow.resize(800, 400)
 119         self.logWindow.raise_()
 120         self.logWindow.activateWindow()
 121
 122     def showRSpecWindow(self, link):
 123         self.rspecWindow.show()
 124         self.rspecWindow.resize(500, 640)
 125         self.rspecWindow.raise_()
 126         self.rspecWindow.activateWindow()
 127
 128
 129     def animatePixmap(self, y):
 130         self.pix.move(0, y)
 131
 132     def animateToScreen(self, link):
 133         if link == self.config_screen.name:
 134             self.next_screen = self.config_screen
 135         elif link == self.main_screen.name:
 136             self.next_screen = self.main_screen
 137
 138         curr_screen = self.screens.currentWidget()
 139
 140         if self.next_screen == curr_screen:
 141             self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
 142             return
 143
 144         pixmap = QPixmap(self.screens.size())
 145         curr_screen.render(pixmap)
 146         self.screens.setCurrentWidget(self.pix)
 147         self.pix.setPixmap(pixmap)
 148
 149         timeLine = QTimeLine(500, self)
 150         timeLine.setFrameRange(0, self.screens.height());
 151         self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
 152         self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
 153         timeLine.start()
 154
 155
 156     def toNextScreen(self):
 157         if self.next_screen == self.config_screen:
 158             self.toConfigScreen()
 159         elif self.next_screen == self.main_screen:
 160             self.toMainScreen()
 161
 162     def toConfigScreen(self):
 163         self.screens.setCurrentWidget(self.config_screen)
 164         self.nav.setTitle(self.config_screen.getTitleText())
 165
 166     def toMainScreen(self):
 167         self.screens.setCurrentWidget(self.main_screen)
 168         self.nav.setTitle(self.main_screen.getTitleText())
 169
 170     def setStatus(self, msg, timeout):
 171         self.status.set(msg, timeout)
 172
 173     def nodeSelectionChanged(self, hostname):
 174         self.rspecWindow.showNode(hostname)