6 from PyQt4.QtCore import *
7 from PyQt4.QtGui import *
9 from distutils.version import LooseVersion
10 from sfa.util.version import version_core
13 from sface.config import config
14 from sface.logwindow import LogWindow
15 from sface.rspecwindow import RSpecWindow, ResourcesWindow
16 from sface.screens.sfascreen import SfaScreen
17 from sface.xmlrpcwindow import get_tracker, init_tracker
19 MINIMUM_SFA_VERSION = "1.0-37"
21 # depending on the platform..
22 # could probably use Qt's resource system but looks overkill for just one file...
23 def locate_image_file (filename):
24 for dir in [ '/usr/share', '/Applications/sface.app/Contents/Resources/sface' ] :
25 for suffix in ['png','jpg']:
26 attempt=os.path.join(dir,'images',"%s.%s"%(filename,suffix))
27 if os.path.isfile(attempt) : return attempt
28 return os.path.join('/could/not/locate/image/file',filename)
30 def load_screens(dirname):
32 for fn in os.listdir(dirname):
33 if not fn.endswith(".py"):
35 modname = fn.rsplit(".py",1)[0]
36 if modname == "sfascreen":
37 # ignore this, it's the base class, not a screen
39 modnames.append(modname)
41 # we want the stock screens to show up in a specific order. plugins can
42 # show up in any order afterward.
44 sort_order = ["mainscreen", "statusscreen", "userscreen", "configscreen", "helpscreen"]
46 for modname in sort_order:
47 if modname in modnames:
48 modnames.remove(modname)
49 modnames.insert(0,modname)
51 # import each module and find whatever class(es) is descendant from
52 # SfaScreen within the module. Might be a better idea to just define a
53 # screens=[] variable in each module
56 for modname in modnames:
58 mod = __import__("sface.screens." + modname, fromlist=["modname"])
60 print "Exception while importing screen", modname
64 for object in dir(mod):
65 object = getattr(mod, object)
66 if hasattr(object, "__bases__") and sface.screens.sfascreen.SfaScreen in object.__bases__:
67 screens.append(object)
72 sfa_version = version_core()['code_tag']
73 if (LooseVersion(sfa_version) < LooseVersion(MINIMUM_SFA_VERSION)):
74 QMessageBox.warning(None, "Old SFA Version", "sfa version %s is required. "
75 "Your installed version is %s. "
76 "Please upgrade your sfa and sfa-client packages."
77 % (MINIMUM_SFA_VERSION, sfa_version) )
81 def __init__(self, screens, parent=None):
82 QWidget.__init__(self, parent)
84 self.title = QLabel("", self)
85 scene=QGraphicsScene()
86 pixmap = QPixmap(locate_image_file('graphic-sfa64'))
87 logolabel=QLabel("",self)
88 logolabel.setPixmap(pixmap)
90 hlayout = QHBoxLayout()
91 hlayout.addWidget(logolabel)
92 hlayout.addWidget(self.title)
94 gotolabel=QLabel("Go to: ", self)
95 gotolabel.setAlignment(Qt.AlignRight)
96 hlayout.addWidget(gotolabel)
98 self.screenLabels = []
99 for screen in screens:
100 label = QLabel(screen.getLinkText(), self)
101 label.setAlignment(Qt.AlignRight)
102 self.screenLabels.append(label)
103 hlayout.addWidget(label)
105 self.setLayout(hlayout)
107 def setTitle(self, title):
108 self.title.setText(title)
111 class Status(QLabel):
112 def __init__(self, parent=None):
113 QLabel.__init__(self, "", parent)
114 self.setMaximumWidth(640)
115 self.sliceUpdateDate()
117 def set(self, msg, timeout):
120 QTimer.singleShot(timeout, self.reset)
122 def sliceUpdateDate(self):
123 rspec_file = config.getSliceRSpecFile()
124 if not os.path.exists(rspec_file):
127 creation_time = os.stat(rspec_file).st_ctime
128 last_update = time.ctime(creation_time)
129 self.set("Slice data last refreshed on %s" % last_update, timeout=None)
133 QTimer.singleShot(1500, self.sliceUpdateDate)
136 class MainWindow(QWidget):
137 def __init__(self, parent=None):
138 QWidget.__init__(self, parent)
142 # These are top-level windows and should be initialized with parent set
143 # to our parent. Otherwise, getting a segfault on exit in Ubuntu.
144 self.logWindow = LogWindow(parent)
145 self.rspecWindow = RSpecWindow(parent)
146 self.resourcesWindow = ResourcesWindow(parent)
147 self.trackerWindow = init_tracker(parent)
149 self.pix = QLabel(self)
151 screenClasses = load_screens(os.path.dirname(sface.screens.__file__))
152 self.screenWidgets = []
154 self.screens = QStackedWidget(self)
155 for screen in screenClasses:
156 # use a try/catch block to isolate the screen. Third-party plugins
159 screenWidget = screen(self)
160 self.screenWidgets.append(screenWidget)
161 self.screens.addWidget(screenWidget)
163 print "Exception while creating screen", screen.__name__
164 traceback.print_exc()
166 self.screens.addWidget(self.pix)
167 self.next_screen = None
169 self.nav = Nav(self.screenWidgets, self)
171 if self.screenWidgets:
172 self.nav.setTitle(self.screenWidgets[0].getTitleText())
174 self.status = Status(self)
175 self.tracker = QLabel("<a href='showtracker'>Xmlrpc</a>", self)
176 self.log = QLabel("<a href='showlog'>Log</a>", self)
177 self.rspec = QLabel("<a href='showlog'>RSpec</a>", self)
178 self.resources = QLabel("<a href='showlog'>Resources</a>", self)
180 hlayout = QHBoxLayout()
181 hlayout.addWidget(self.status)
183 hlayout.addWidget(self.tracker)
184 hlayout.addWidget(self.rspec)
185 hlayout.addWidget(self.resources)
186 hlayout.addWidget(self.log)
188 layout = QVBoxLayout()
189 layout.addWidget(self.nav)
190 layout.addWidget(self.screens)
191 layout.addLayout(hlayout)
192 self.setLayout(layout)
193 self.resize(800, 500)
195 for link in self.nav.screenLabels:
196 self.connect(link, SIGNAL('linkActivated(QString)'),
197 self.animateToScreen)
199 self.connect(self.tracker, SIGNAL('linkActivated(QString)'),
200 self.showTrackerWindow)
201 self.connect(self.log, SIGNAL('linkActivated(QString)'),
203 self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
204 self.showRSpecWindow)
205 self.connect(self.resources, SIGNAL('linkActivated(QString)'),
206 self.showResourcesWindow)
208 def redirectOutputToLog(self):
209 self.logWindow.redirectOutput()
211 def showTrackerWindow(self):
212 tracker = get_tracker()
214 tracker.resize(750, 580)
216 tracker.activateWindow()
218 def showLogWindow(self, link):
219 self.logWindow.show()
220 self.logWindow.resize(800, 200)
221 self.logWindow.raise_()
222 self.logWindow.activateWindow()
224 def showRSpecWindow(self, link):
225 self.rspecWindow.show()
226 self.rspecWindow.resize(500, 640)
227 self.rspecWindow.raise_()
228 self.rspecWindow.activateWindow()
230 def showResourcesWindow(self, link):
231 self.resourcesWindow.show()
232 self.resourcesWindow.resize(500, 640)
233 self.resourcesWindow.raise_()
234 self.resourcesWindow.activateWindow()
236 def animatePixmap(self, y):
239 def animateToScreen(self, link):
240 for screen in self.screenWidgets:
241 if link == screen.name:
242 self.next_screen = screen
244 curr_screen = self.screens.currentWidget()
246 if self.next_screen == curr_screen:
247 self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
250 # This is an optimization to have a smoother animation. We
251 # render the widget into a pixmap and animate that instead of
252 # moving the whole widget around.
253 pixmap = QPixmap(self.screens.size())
254 curr_screen.render(pixmap)
255 self.screens.setCurrentWidget(self.pix)
256 self.pix.setPixmap(pixmap)
258 timeLine = QTimeLine(500, self)
259 timeLine.setFrameRange(0, self.screens.height());
260 self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
261 self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
264 def toNextScreen(self):
265 self.screens.setCurrentWidget(self.next_screen)
266 self.nav.setTitle(self.next_screen.getTitleText())
268 def setStatus(self, msg, timeout):
269 self.status.set(msg, timeout)
271 def nodeSelectionChanged(self, hostname):
272 if self.rspecWindow.isVisible():
273 self.rspecWindow.showNode(hostname)
275 def closeEvent(self, event):
276 # give the screens an opportunity to veto the close
277 for screen in self.screenWidgets:
278 if not screen.canClose():
282 # give the screens an opportunity to close gracefully
283 for screen in self.screenWidgets:
284 screen.mainWindowClose()
286 QWidget.closeEvent(self, event)