3 from __future__ import with_statement
17 @contextlib.contextmanager
19 gtk.gdk.threads_enter()
23 gtk.gdk.threads_leave()
26 def find_parent_window(widget):
28 parent = widget.get_parent()
29 if isinstance(parent, gtk.Window):
36 Decorator that makes a generator-function into a function that will continue execution on next call
40 @functools.wraps(func)
41 def decorated_func(*args, **kwds):
43 a.append(func(*args, **kwds))
54 def asynchronous_gtk_message(original_func):
56 @note Idea came from http://www.aclevername.com/articles/python-webgui/
60 args, kwargs = allArgs
62 original_func(*args, **kwargs)
65 @functools.wraps(original_func)
66 def delayed_func(*args, **kwargs):
67 gobject.idle_add(execute, (args, kwargs))
72 def synchronous_gtk_message(original_func):
74 @note Idea came from http://www.aclevername.com/articles/python-webgui/
77 @functools.wraps(original_func)
78 def immediate_func(*args, **kwargs):
80 return original_func(*args, **kwargs)
88 ... def grep_sink(pattern):
89 ... print "Looking for %s" % pattern
92 ... if pattern in line:
94 >>> g = grep_sink("python")
96 >>> g.send("Yeah but no but yeah but no")
97 >>> g.send("A series of tubes")
98 >>> g.send("python generators rock!")
99 python generators rock!
103 @functools.wraps(func)
104 def start(*args, **kwargs):
105 cr = func(*args, **kwargs)
115 Good for uses like with cochain to pick up any slack
122 def comap(function, target):
124 >>> p = printer_sink()
125 >>> cm = comap(lambda x: x+1, p)
132 >>> # cm.throw(RuntimeError, "Goodbye")
140 mappedItem = function(*item)
141 target.send(mappedItem)
142 except StandardError, e:
143 target.throw(e.__class__, e.message)
147 def queue_sink(queue):
149 >>> q = Queue.Queue()
150 >>> qs = queue_sink(q)
153 >>> qs.throw(RuntimeError, "Goodbye")
156 >>> print [i for i in _flush_queue(q)]
157 [(None, 'Hello'), (None, 'World'), (<type 'exceptions.RuntimeError'>, 'Goodbye'), (None, 'Meh'), (<type 'exceptions.GeneratorExit'>, None)]
162 queue.put((None, item))
163 except StandardError, e:
164 queue.put((e.__class__, e.message))
165 except GeneratorExit:
166 queue.put((GeneratorExit, None))
170 def decode_item(item, target):
174 elif item[0] is GeneratorExit:
178 target.throw(item[0], item[1])
182 def nonqueue_source(queue, target):
184 >>> q = Queue.Queue()
188 ... (GeneratorExit, None),
191 >>> qs = queue_source(q, printer_sink())
197 isDone = decode_item(item, target)
198 while not queue.empty():
202 def threaded_stage(target, thread_factory = threading.Thread):
203 messages = Queue.Queue()
205 run_source = functools.partial(nonqueue_source, messages, target)
206 thread = thread_factory(target=run_source)
207 thread.setDaemon(True)
210 # Sink running in current thread
211 return queue_sink(messages)
214 class LoginWindow(object):
216 def __init__(self, widgetTree):
218 @note Thread agnostic
220 self._dialog = widgetTree.get_widget("loginDialog")
221 self._parentWindow = widgetTree.get_widget("mainWindow")
222 self._serviceCombo = widgetTree.get_widget("serviceCombo")
223 self._usernameEntry = widgetTree.get_widget("usernameentry")
224 self._passwordEntry = widgetTree.get_widget("passwordentry")
226 self._serviceList = gtk.ListStore(gobject.TYPE_INT, gobject.TYPE_STRING)
227 self._serviceCombo.set_model(self._serviceList)
228 cell = gtk.CellRendererText()
229 self._serviceCombo.pack_start(cell, True)
230 self._serviceCombo.add_attribute(cell, 'text', 1)
231 self._serviceCombo.set_active(0)
234 "on_loginbutton_clicked": self._on_loginbutton_clicked,
235 "on_loginclose_clicked": self._on_loginclose_clicked,
237 widgetTree.signal_autoconnect(callbackMapping)
239 def request_credentials(self, parentWindow = None):
243 if parentWindow is None:
244 parentWindow = self._parentWindow
246 self._serviceCombo.hide()
247 self._serviceList.clear()
250 self._dialog.set_transient_for(parentWindow)
251 self._dialog.set_default_response(gtk.RESPONSE_OK)
252 response = self._dialog.run()
253 if response != gtk.RESPONSE_OK:
254 raise RuntimeError("Login Cancelled")
256 username = self._usernameEntry.get_text()
257 password = self._passwordEntry.get_text()
258 self._passwordEntry.set_text("")
262 return username, password
264 def request_credentials_from(self,
267 defaultCredentials = ("", "")
272 if parentWindow is None:
273 parentWindow = self._parentWindow
275 self._serviceList.clear()
276 for serviceIdserviceName in services:
277 self._serviceList.append(serviceIdserviceName)
278 self._serviceCombo.set_active(0)
279 self._serviceCombo.show()
281 self._usernameEntry.set_text(defaultCredentials[0])
282 self._passwordEntry.set_text(defaultCredentials[1])
285 self._dialog.set_transient_for(parentWindow)
286 self._dialog.set_default_response(gtk.RESPONSE_OK)
287 response = self._dialog.run()
288 if response != gtk.RESPONSE_OK:
289 raise RuntimeError("Login Cancelled")
291 username = self._usernameEntry.get_text()
292 password = self._passwordEntry.get_text()
296 itr = self._serviceCombo.get_active_iter()
297 serviceId = int(self._serviceList.get_value(itr, 0))
298 self._serviceList.clear()
299 return serviceId, username, password
301 def _on_loginbutton_clicked(self, *args):
302 self._dialog.response(gtk.RESPONSE_OK)
304 def _on_loginclose_clicked(self, *args):
305 self._dialog.response(gtk.RESPONSE_CANCEL)
308 def safecall(f, errorDisplay=None, default=None, exception=Exception):
310 Returns modified f. When the modified f is called and throws an
311 exception, the default value is returned
313 def _safecall(*args, **argv):
315 return f(*args,**argv)
317 if errorDisplay is not None:
318 errorDisplay.push_exception(e)
323 class ErrorDisplay(object):
325 def __init__(self, widgetTree):
326 super(ErrorDisplay, self).__init__()
327 self.__errorBox = widgetTree.get_widget("errorEventBox")
328 self.__errorDescription = widgetTree.get_widget("errorDescription")
329 self.__errorClose = widgetTree.get_widget("errorClose")
330 self.__parentBox = self.__errorBox.get_parent()
332 self.__errorBox.connect("button_release_event", self._on_close)
335 self.__parentBox.remove(self.__errorBox)
337 def push_message_with_lock(self, message):
339 self.push_message(message)
341 def push_message(self, message):
342 if 0 < len(self.__messages):
343 self.__messages.append(message)
345 self.__show_message(message)
347 def push_exception_with_lock(self, exception = None):
349 self.push_exception(exception)
351 def push_exception(self, exception = None):
352 if exception is None:
353 userMessage = str(sys.exc_value)
354 warningMessage = str(traceback.format_exc())
356 userMessage = str(exception)
357 warningMessage = str(exception)
358 self.push_message(userMessage)
359 warnings.warn(warningMessage, stacklevel=2)
361 def pop_message(self):
362 if 0 < len(self.__messages):
363 self.__show_message(self.__messages[0])
364 del self.__messages[0]
366 self.__hide_message()
368 def _on_close(self, *args):
371 def __show_message(self, message):
372 self.__errorDescription.set_text(message)
373 self.__parentBox.pack_start(self.__errorBox, False, False)
374 self.__parentBox.reorder_child(self.__errorBox, 1)
376 def __hide_message(self):
377 self.__errorDescription.set_text("")
378 self.__parentBox.remove(self.__errorBox)
381 class DummyErrorDisplay(object):
384 super(DummyErrorDisplay, self).__init__()
388 def push_message_with_lock(self, message):
389 self.push_message(message)
391 def push_message(self, message):
392 if 0 < len(self.__messages):
393 self.__messages.append(message)
395 self.__show_message(message)
397 def push_exception(self, exception = None):
398 if exception is None:
399 warningMessage = traceback.format_exc()
401 warningMessage = exception
402 warnings.warn(warningMessage, stacklevel=3)
404 def pop_message(self):
405 if 0 < len(self.__messages):
406 self.__show_message(self.__messages[0])
407 del self.__messages[0]
409 def __show_message(self, message):
410 warnings.warn(message, stacklevel=2)
413 class MessageBox(gtk.MessageDialog):
415 def __init__(self, message):
417 gtk.MessageDialog.__init__(
420 gtk.DIALOG_MODAL|gtk.DIALOG_DESTROY_WITH_PARENT,
425 self.set_default_response(gtk.RESPONSE_OK)
426 self.connect('response', self._handle_clicked)
428 def _handle_clicked(self, *args):
432 class MessageBox2(gtk.MessageDialog):
434 def __init__(self, message):
436 gtk.MessageDialog.__init__(
439 gtk.DIALOG_DESTROY_WITH_PARENT,
444 self.set_default_response(gtk.RESPONSE_OK)
445 self.connect('response', self._handle_clicked)
447 def _handle_clicked(self, *args):
451 class PopupCalendar(object):
453 def __init__(self, parent, displayDate, title = ""):
454 self._displayDate = displayDate
456 self._calendar = gtk.Calendar()
457 self._calendar.select_month(self._displayDate.month, self._displayDate.year)
458 self._calendar.select_day(self._displayDate.day)
459 self._calendar.set_display_options(
460 gtk.CALENDAR_SHOW_HEADING |
461 gtk.CALENDAR_SHOW_DAY_NAMES |
462 gtk.CALENDAR_NO_MONTH_CHANGE |
465 self._calendar.connect("day-selected", self._on_day_selected)
467 self._popupWindow = gtk.Window()
468 self._popupWindow.set_title(title)
469 self._popupWindow.add(self._calendar)
470 self._popupWindow.set_transient_for(parent)
471 self._popupWindow.set_modal(True)
472 self._popupWindow.set_type_hint(gtk.gdk.WINDOW_TYPE_HINT_DIALOG)
473 self._popupWindow.set_skip_pager_hint(True)
474 self._popupWindow.set_skip_taskbar_hint(True)
477 self._popupWindow.show_all()
479 def _on_day_selected(self, *args):
481 self._calendar.select_month(self._displayDate.month, self._displayDate.year)
482 self._calendar.select_day(self._displayDate.day)
483 except StandardError, e:
484 warnings.warn(e.message)
487 class QuickAddView(object):
489 def __init__(self, widgetTree, errorDisplay, signalSink, prefix):
490 self._errorDisplay = errorDisplay
492 self._signalSink = signalSink
494 self._clipboard = gtk.clipboard_get()
496 self._taskNameEntry = widgetTree.get_widget(prefix+"-nameEntry")
497 self._addTaskButton = widgetTree.get_widget(prefix+"-addButton")
498 self._pasteTaskNameButton = widgetTree.get_widget(prefix+"-pasteNameButton")
499 self._clearTaskNameButton = widgetTree.get_widget(prefix+"-clearNameButton")
501 self._onAddClickedId = None
502 self._onAddReleasedId = None
503 self._addToEditTimerId = None
504 self._onClearId = None
505 self._onPasteId = None
507 def enable(self, manager):
508 self._manager = manager
510 self._onAddId = self._addTaskButton.connect("clicked", self._on_add)
511 self._onAddClickedId = self._addTaskButton.connect("pressed", self._on_add_pressed)
512 self._onAddReleasedId = self._addTaskButton.connect("released", self._on_add_released)
513 self._onPasteId = self._pasteTaskNameButton.connect("clicked", self._on_paste)
514 self._onClearId = self._clearTaskNameButton.connect("clicked", self._on_clear)
519 self._addTaskButton.disconnect(self._onAddId)
520 self._addTaskButton.disconnect(self._onAddClickedId)
521 self._addTaskButton.disconnect(self._onAddReleasedId)
522 self._pasteTaskNameButton.disconnect(self._onPasteId)
523 self._clearTaskNameButton.disconnect(self._onClearId)
525 def set_addability(self, addability):
526 self._addTaskButton.set_sensitive(addability)
528 def _on_add(self, *args):
530 name = self._taskNameEntry.get_text()
531 self._taskNameEntry.set_text("")
533 self._signalSink.stage.send(("add", name))
534 except StandardError, e:
535 self._errorDisplay.push_exception()
537 def _on_add_edit(self, *args):
539 name = self._taskNameEntry.get_text()
540 self._taskNameEntry.set_text("")
542 self._signalSink.stage.send(("add-edit", name))
543 except StandardError, e:
544 self._errorDisplay.push_exception()
546 def _on_add_pressed(self, widget):
548 self._addToEditTimerId = gobject.timeout_add(1000, self._on_add_edit)
549 except StandardError, e:
550 self._errorDisplay.push_exception()
552 def _on_add_released(self, widget):
554 if self._addToEditTimerId is not None:
555 gobject.source_remove(self._addToEditTimerId)
556 self._addToEditTimerId = None
557 except StandardError, e:
558 self._errorDisplay.push_exception()
560 def _on_paste(self, *args):
562 entry = self._taskNameEntry.get_text()
563 addedText = self._clipboard.wait_for_text()
566 self._taskNameEntry.set_text(entry)
567 except StandardError, e:
568 self._errorDisplay.push_exception()
570 def _on_clear(self, *args):
572 self._taskNameEntry.set_text("")
573 except StandardError, e:
574 self._errorDisplay.push_exception()
577 class TapOrHold(object):
579 def __init__(self, widget):
580 self._widget = widget
582 self._isPointerInside = True
583 self._holdTimeoutId = None
584 self._tapTimeoutId = None
593 self._bpeId = self._widget.connect("button-press-event", self._on_button_press)
594 self._breId = self._widget.connect("button-release-event", self._on_button_release)
595 self._eneId = self._widget.connect("enter-notify-event", self._on_enter)
596 self._lneId = self._widget.connect("leave-notify-event", self._on_leave)
599 self._widget.disconnect(self._bpeId)
600 self._widget.disconnect(self._breId)
601 self._widget.disconnect(self._eneId)
602 self._widget.disconnect(self._lneId)
604 def on_tap(self, taps):
607 def on_hold(self, taps):
610 def on_holding(self):
616 def _on_button_press(self, *args):
617 # Hack to handle weird notebook behavior
618 self._isPointerInside = True
620 if self._tapTimeoutId is not None:
621 gobject.source_remove(self._tapTimeoutId)
622 self._tapTimeoutId = None
625 if self._holdTimeoutId is None:
626 self._tapTimeoutId = None
629 self._holdTimeoutId = gobject.timeout_add(1000, self._on_hold_timeout)
633 def _on_button_release(self, *args):
634 assert self._tapTimeoutId is None
635 # Handle release after timeout if user hasn't double-clicked
636 self._tapTimeoutId = gobject.timeout_add(100, self._on_tap_timeout)
638 def _on_actual_press(self, *args):
639 if self._holdTimeoutId is not None:
640 gobject.source_remove(self._holdTimeoutId)
641 self._holdTimeoutId = None
643 if self._isPointerInside:
645 self.on_tap(self._taps)
647 self.on_hold(self._taps)
651 def _on_tap_timeout(self, *args):
652 self._tapTimeoutId = None
653 self._on_actual_press()
656 def _on_hold_timeout(self, *args):
657 self._holdTimeoutId = None
662 def _on_enter(self, *args):
663 self._isPointerInside = True
665 def _on_leave(self, *args):
666 self._isPointerInside = False
669 if __name__ == "__main__":
672 cal = PopupCalendar(None, datetime.datetime.now())
673 cal._popupWindow.connect("destroy", lambda w: gtk.main_quit())