6 from PyQt4.QtCore import *
7 from PyQt4.QtGui import *
10 from sface.config import config
11 from sface.logwindow import LogWindow
12 from sface.rspecwindow import RSpecWindow, ResourcesWindow
13 from sface.screens.sfascreen import SfaScreen
14 from sface.xmlrpcwindow import get_tracker, init_tracker
16 # depending on the platform..
17 # could probably use Qt's resource system but looks overkill for just one file...
18 def locate_image_file (filename):
19 for dir in [ '/usr/share', '/Applications/sface.app/Contents/Resources/sface' ] :
20 for suffix in ['png','jpg']:
21 attempt=os.path.join(dir,'images',"%s.%s"%(filename,suffix))
22 if os.path.isfile(attempt) : return attempt
23 return os.path.join('/could/not/locate/image/file',filename)
25 def load_screens(dirname):
27 for fn in os.listdir(dirname):
28 if not fn.endswith(".py"):
30 modname = fn.rsplit(".py",1)[0]
31 if modname == "sfascreen":
32 # ignore this, it's the base class, not a screen
34 modnames.append(modname)
36 # we want the stock screens to show up in a specific order. plugins can
37 # show up in any order afterward.
39 sort_order = ["mainscreen", "userscreen", "configscreen", "helpscreen"]
41 for modname in sort_order:
42 if modname in modnames:
43 modnames.remove(modname)
44 modnames.insert(0,modname)
46 # import each module and find whatever class(es) is descendant from
47 # SfaScreen within the module. Might be a better idea to just define a
48 # screens=[] variable in each module
51 for modname in modnames:
53 mod = __import__("sface.screens." + modname, fromlist=["modname"])
55 print "Exception while importing screen", modname
59 for object in dir(mod):
60 object = getattr(mod, object)
61 if hasattr(object, "__bases__") and sface.screens.sfascreen.SfaScreen in object.__bases__:
62 screens.append(object)
67 def __init__(self, screens, parent=None):
68 QWidget.__init__(self, parent)
70 self.title = QLabel("", self)
71 scene=QGraphicsScene()
72 pixmap = QPixmap(locate_image_file('graphic-sfa64'))
73 logolabel=QLabel("",self)
74 logolabel.setPixmap(pixmap)
76 hlayout = QHBoxLayout()
77 hlayout.addWidget(logolabel)
78 hlayout.addWidget(self.title)
80 gotolabel=QLabel("Go to: ", self)
81 gotolabel.setAlignment(Qt.AlignRight)
82 hlayout.addWidget(gotolabel)
84 self.screenLabels = []
85 for screen in screens:
86 label = QLabel(screen.getLinkText(), self)
87 label.setAlignment(Qt.AlignRight)
88 self.screenLabels.append(label)
89 hlayout.addWidget(label)
91 self.setLayout(hlayout)
93 def setTitle(self, title):
94 self.title.setText(title)
98 def __init__(self, parent=None):
99 QLabel.__init__(self, "", parent)
100 self.setMaximumWidth(640)
101 self.sliceUpdateDate()
103 def set(self, msg, timeout):
106 QTimer.singleShot(timeout, self.reset)
108 def sliceUpdateDate(self):
109 rspec_file = config.getSliceRSpecFile()
110 if not os.path.exists(rspec_file):
113 creation_time = os.stat(rspec_file).st_ctime
114 last_update = time.ctime(creation_time)
115 self.set("Slice data last refreshed on %s" % last_update, timeout=None)
119 QTimer.singleShot(1500, self.sliceUpdateDate)
122 class MainWindow(QWidget):
123 def __init__(self, parent=None):
124 QWidget.__init__(self, parent)
126 # These are top-level windows and should be initialized with parent set
127 # to our parent. Otherwise, getting a segfault on exit in Ubuntu.
128 self.logWindow = LogWindow(parent)
129 self.rspecWindow = RSpecWindow(parent)
130 self.resourcesWindow = ResourcesWindow(parent)
131 self.trackerWindow = init_tracker(parent)
133 self.pix = QLabel(self)
135 screenClasses = load_screens(os.path.dirname(sface.screens.__file__))
136 self.screenWidgets = []
138 self.screens = QStackedWidget(self)
139 for screen in screenClasses:
140 # use a try/catch block to isolate the screen. Third-party plugins
143 screenWidget = screen(self)
144 self.screenWidgets.append(screenWidget)
145 self.screens.addWidget(screenWidget)
147 print "Exception while creating screen", screen.__name__
148 traceback.print_exc()
150 self.screens.addWidget(self.pix)
151 self.next_screen = None
153 self.nav = Nav(self.screenWidgets, self)
155 if self.screenWidgets:
156 self.nav.setTitle(self.screenWidgets[0].getTitleText())
158 self.status = Status(self)
159 self.tracker = QLabel("<a href='showtracker'>Xmlrpc</a>", self)
160 self.log = QLabel("<a href='showlog'>Log</a>", self)
161 self.rspec = QLabel("<a href='showlog'>RSpec</a>", self)
162 self.resources = QLabel("<a href='showlog'>Resources</a>", self)
164 hlayout = QHBoxLayout()
165 hlayout.addWidget(self.status)
167 hlayout.addWidget(self.tracker)
168 hlayout.addWidget(self.rspec)
169 hlayout.addWidget(self.resources)
170 hlayout.addWidget(self.log)
172 layout = QVBoxLayout()
173 layout.addWidget(self.nav)
174 layout.addWidget(self.screens)
175 layout.addLayout(hlayout)
176 self.setLayout(layout)
177 self.resize(800, 500)
179 for link in self.nav.screenLabels:
180 self.connect(link, SIGNAL('linkActivated(QString)'),
181 self.animateToScreen)
183 self.connect(self.tracker, SIGNAL('linkActivated(QString)'),
184 self.showTrackerWindow)
185 self.connect(self.log, SIGNAL('linkActivated(QString)'),
187 self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
188 self.showRSpecWindow)
189 self.connect(self.resources, SIGNAL('linkActivated(QString)'),
190 self.showResourcesWindow)
192 def redirectOutputToLog(self):
193 self.logWindow.redirectOutput()
195 def showTrackerWindow(self):
196 tracker = get_tracker()
198 tracker.resize(500, 640)
200 tracker.activateWindow()
202 def showLogWindow(self, link):
203 self.logWindow.show()
204 self.logWindow.resize(800, 200)
205 self.logWindow.raise_()
206 self.logWindow.activateWindow()
208 def showRSpecWindow(self, link):
209 self.rspecWindow.show()
210 self.rspecWindow.resize(500, 640)
211 self.rspecWindow.raise_()
212 self.rspecWindow.activateWindow()
214 def showResourcesWindow(self, link):
215 self.resourcesWindow.show()
216 self.resourcesWindow.resize(500, 640)
217 self.resourcesWindow.raise_()
218 self.resourcesWindow.activateWindow()
220 def animatePixmap(self, y):
223 def animateToScreen(self, link):
224 for screen in self.screenWidgets:
225 if link == screen.name:
226 self.next_screen = screen
228 curr_screen = self.screens.currentWidget()
230 if self.next_screen == curr_screen:
231 self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
234 # This is an optimization to have a smoother animation. We
235 # render the widget into a pixmap and animate that instead of
236 # moving the whole widget around.
237 pixmap = QPixmap(self.screens.size())
238 curr_screen.render(pixmap)
239 self.screens.setCurrentWidget(self.pix)
240 self.pix.setPixmap(pixmap)
242 timeLine = QTimeLine(500, self)
243 timeLine.setFrameRange(0, self.screens.height());
244 self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
245 self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
248 def toNextScreen(self):
249 self.screens.setCurrentWidget(self.next_screen)
250 self.nav.setTitle(self.next_screen.getTitleText())
252 def setStatus(self, msg, timeout):
253 self.status.set(msg, timeout)
255 def nodeSelectionChanged(self, hostname):
256 if self.rspecWindow.isVisible():
257 self.rspecWindow.showNode(hostname)
259 def closeEvent(self, event):
260 # give the screens an opportunity to veto the close
261 for screen in self.screenWidgets:
262 if not screen.canClose():
266 # give the screens an opportunity to close gracefully
267 for screen in self.screenWidgets:
268 screen.mainWindowClose()
270 QWidget.closeEvent(self, event)