3 from __future__ import with_statement
20 _moduleLogger = logging.getLogger("gtk_toolbox")
23 def get_screen_orientation():
24 width, height = gtk.gdk.get_default_root_window().get_size()
26 return gtk.ORIENTATION_VERTICAL
28 return gtk.ORIENTATION_HORIZONTAL
31 def orientation_change_connect(handler, *args):
33 @param handler(orientation, *args) -> None(?)
35 initialScreenOrientation = get_screen_orientation()
36 orientationAndArgs = list(itertools.chain((initialScreenOrientation, ), args))
38 def _on_screen_size_changed(screen):
39 newScreenOrientation = get_screen_orientation()
40 if newScreenOrientation != orientationAndArgs[0]:
41 orientationAndArgs[0] = newScreenOrientation
42 handler(*orientationAndArgs)
44 rootScreen = gtk.gdk.get_default_root_window()
45 return gtk.connect(rootScreen, "size-changed", _on_screen_size_changed)
48 @contextlib.contextmanager
49 def flock(path, timeout=-1):
56 while timeSpent <= timeout or timeout == WAIT_FOREVER:
58 fd = os.open(path, os.O_CREAT | os.O_EXCL | os.O_RDWR)
62 if e.errno != errno.EEXIST:
67 assert acquired, "Failed to grab file-lock %s within timeout %d" % (path, timeout)
75 @contextlib.contextmanager
77 gtk.gdk.threads_enter()
81 gtk.gdk.threads_leave()
84 def find_parent_window(widget):
86 parent = widget.get_parent()
87 if isinstance(parent, gtk.Window):
94 Decorator that makes a generator-function into a function that will continue execution on next call
98 @functools.wraps(func)
99 def decorated_func(*args, **kwds):
101 a.append(func(*args, **kwds))
105 except StopIteration:
109 return decorated_func
112 def asynchronous_gtk_message(original_func):
114 @note Idea came from http://www.aclevername.com/articles/python-webgui/
117 def execute(allArgs):
118 args, kwargs = allArgs
120 original_func(*args, **kwargs)
123 @functools.wraps(original_func)
124 def delayed_func(*args, **kwargs):
125 gobject.idle_add(execute, (args, kwargs))
130 def synchronous_gtk_message(original_func):
132 @note Idea came from http://www.aclevername.com/articles/python-webgui/
135 @functools.wraps(original_func)
136 def immediate_func(*args, **kwargs):
138 return original_func(*args, **kwargs)
140 return immediate_func
146 ... def grep_sink(pattern):
147 ... print "Looking for %s" % pattern
150 ... if pattern in line:
152 >>> g = grep_sink("python")
154 >>> g.send("Yeah but no but yeah but no")
155 >>> g.send("A series of tubes")
156 >>> g.send("python generators rock!")
157 python generators rock!
161 @functools.wraps(func)
162 def start(*args, **kwargs):
163 cr = func(*args, **kwargs)
171 def printer_sink(format = "%s"):
173 >>> pr = printer_sink("%r")
180 >>> p = printer_sink()
185 >>> # p.throw(RuntimeError, "Goodbye")
191 print format % (item, )
197 Good for uses like with cochain to pick up any slack
204 def comap(function, target):
206 >>> p = printer_sink()
207 >>> cm = comap(lambda x: x+1, p)
218 mappedItem = function(*item)
219 target.send(mappedItem)
221 _moduleLogger.exception("Forwarding exception!")
222 target.throw(e.__class__, str(e))
225 def _flush_queue(queue):
226 while not queue.empty():
231 def queue_sink(queue):
233 >>> q = Queue.Queue()
234 >>> qs = queue_sink(q)
237 >>> qs.throw(RuntimeError, "Goodbye")
240 >>> print [i for i in _flush_queue(q)]
241 [(None, 'Hello'), (None, 'World'), (<type 'exceptions.RuntimeError'>, 'Goodbye'), (None, 'Meh'), (<type 'exceptions.GeneratorExit'>, None)]
246 queue.put((None, item))
248 queue.put((e.__class__, str(e)))
249 except GeneratorExit:
250 queue.put((GeneratorExit, None))
254 def decode_item(item, target):
258 elif item[0] is GeneratorExit:
262 target.throw(item[0], item[1])
266 def nonqueue_source(queue, target):
270 isDone = decode_item(item, target)
271 while not queue.empty():
275 def threaded_stage(target, thread_factory = threading.Thread):
276 messages = Queue.Queue()
278 run_source = functools.partial(nonqueue_source, messages, target)
279 thread = thread_factory(target=run_source)
280 thread.setDaemon(True)
283 # Sink running in current thread
284 return queue_sink(messages)
287 def log_exception(logger):
289 def log_exception_decorator(func):
291 @functools.wraps(func)
292 def wrapper(*args, **kwds):
294 return func(*args, **kwds)
296 logger.exception(func.__name__)
300 return log_exception_decorator
303 class LoginWindow(object):
305 def __init__(self, widgetTree):
307 @note Thread agnostic
309 self._dialog = widgetTree.get_widget("loginDialog")
310 self._parentWindow = widgetTree.get_widget("mainWindow")
311 self._serviceCombo = widgetTree.get_widget("serviceCombo")
312 self._usernameEntry = widgetTree.get_widget("usernameentry")
313 self._passwordEntry = widgetTree.get_widget("passwordentry")
315 self._serviceList = gtk.ListStore(gobject.TYPE_INT, gobject.TYPE_STRING)
316 self._serviceCombo.set_model(self._serviceList)
317 cell = gtk.CellRendererText()
318 self._serviceCombo.pack_start(cell, True)
319 self._serviceCombo.add_attribute(cell, 'text', 1)
320 self._serviceCombo.set_active(0)
322 widgetTree.get_widget("loginbutton").connect("clicked", self._on_loginbutton_clicked)
323 widgetTree.get_widget("logins_close_button").connect("clicked", self._on_loginclose_clicked)
325 def request_credentials(self,
327 defaultCredentials = ("", "")
332 if parentWindow is None:
333 parentWindow = self._parentWindow
335 self._serviceCombo.hide()
336 self._serviceList.clear()
338 self._usernameEntry.set_text(defaultCredentials[0])
339 self._passwordEntry.set_text(defaultCredentials[1])
342 self._dialog.set_transient_for(parentWindow)
343 self._dialog.set_default_response(gtk.RESPONSE_OK)
344 response = self._dialog.run()
345 if response != gtk.RESPONSE_OK:
346 raise RuntimeError("Login Cancelled")
348 username = self._usernameEntry.get_text()
349 password = self._passwordEntry.get_text()
350 self._passwordEntry.set_text("")
354 return username, password
356 def request_credentials_from(self,
359 defaultCredentials = ("", "")
364 if parentWindow is None:
365 parentWindow = self._parentWindow
367 self._serviceList.clear()
368 for serviceIdserviceName in services:
369 self._serviceList.append(serviceIdserviceName)
370 self._serviceCombo.set_active(0)
371 self._serviceCombo.show()
373 self._usernameEntry.set_text(defaultCredentials[0])
374 self._passwordEntry.set_text(defaultCredentials[1])
377 self._dialog.set_transient_for(parentWindow)
378 self._dialog.set_default_response(gtk.RESPONSE_OK)
379 response = self._dialog.run()
380 if response != gtk.RESPONSE_OK:
381 raise RuntimeError("Login Cancelled")
383 username = self._usernameEntry.get_text()
384 password = self._passwordEntry.get_text()
388 itr = self._serviceCombo.get_active_iter()
389 serviceId = int(self._serviceList.get_value(itr, 0))
390 self._serviceList.clear()
391 return serviceId, username, password
393 def _on_loginbutton_clicked(self, *args):
394 self._dialog.response(gtk.RESPONSE_OK)
396 def _on_loginclose_clicked(self, *args):
397 self._dialog.response(gtk.RESPONSE_CANCEL)
400 def safecall(f, errorDisplay=None, default=None, exception=Exception):
402 Returns modified f. When the modified f is called and throws an
403 exception, the default value is returned
405 def _safecall(*args, **argv):
407 return f(*args,**argv)
409 if errorDisplay is not None:
410 errorDisplay.push_exception(e)
415 class ErrorDisplay(object):
417 def __init__(self, errorBox, errorDescription, errorClose):
418 super(ErrorDisplay, self).__init__()
419 self.__errorBox = errorBox
420 self.__errorDescription = errorDescription
421 self.__errorClose = errorClose
422 self.__parentBox = self.__errorBox.get_parent()
424 self.__errorBox.connect("button_release_event", self._on_close)
427 self.__parentBox.remove(self.__errorBox)
429 def push_message_with_lock(self, message):
431 self.push_message(message)
433 def push_message(self, message):
434 self.__messages.append(message)
435 if 1 == len(self.__messages):
436 self.__show_message(message)
438 def push_exception_with_lock(self):
440 self.push_exception()
442 def push_exception(self):
443 userMessage = str(sys.exc_info()[1])
444 self.push_message(userMessage)
445 _moduleLogger.exception(userMessage)
447 def pop_message(self):
448 del self.__messages[0]
449 if 0 == len(self.__messages):
450 self.__hide_message()
452 self.__errorDescription.set_text(self.__messages[0])
454 def _on_close(self, *args):
457 def __show_message(self, message):
458 self.__errorDescription.set_text(message)
459 self.__parentBox.pack_start(self.__errorBox, False, False)
460 self.__parentBox.reorder_child(self.__errorBox, 1)
462 def __hide_message(self):
463 self.__errorDescription.set_text("")
464 self.__parentBox.remove(self.__errorBox)
467 class DummyErrorDisplay(object):
470 super(DummyErrorDisplay, self).__init__()
474 def push_message_with_lock(self, message):
475 self.push_message(message)
477 def push_message(self, message):
478 if 0 < len(self.__messages):
479 self.__messages.append(message)
481 self.__show_message(message)
483 def push_exception(self, exception = None):
484 userMessage = str(sys.exc_value)
485 _moduleLogger.exception(userMessage)
487 def pop_message(self):
488 if 0 < len(self.__messages):
489 self.__show_message(self.__messages[0])
490 del self.__messages[0]
492 def __show_message(self, message):
493 _moduleLogger.debug(message)
496 class MessageBox(gtk.MessageDialog):
498 def __init__(self, message):
500 gtk.MessageDialog.__init__(
503 gtk.DIALOG_MODAL|gtk.DIALOG_DESTROY_WITH_PARENT,
508 self.set_default_response(gtk.RESPONSE_OK)
509 self.connect('response', self._handle_clicked)
511 def _handle_clicked(self, *args):
515 class MessageBox2(gtk.MessageDialog):
517 def __init__(self, message):
519 gtk.MessageDialog.__init__(
522 gtk.DIALOG_DESTROY_WITH_PARENT,
527 self.set_default_response(gtk.RESPONSE_OK)
528 self.connect('response', self._handle_clicked)
530 def _handle_clicked(self, *args):
534 class PopupCalendar(object):
536 def __init__(self, parent, displayDate, title = ""):
537 self._displayDate = displayDate
539 self._calendar = gtk.Calendar()
540 self._calendar.select_month(self._displayDate.month, self._displayDate.year)
541 self._calendar.select_day(self._displayDate.day)
542 self._calendar.set_display_options(
543 gtk.CALENDAR_SHOW_HEADING |
544 gtk.CALENDAR_SHOW_DAY_NAMES |
545 gtk.CALENDAR_NO_MONTH_CHANGE |
548 self._calendar.connect("day-selected", self._on_day_selected)
550 self._popupWindow = gtk.Window()
551 self._popupWindow.set_title(title)
552 self._popupWindow.add(self._calendar)
553 self._popupWindow.set_transient_for(parent)
554 self._popupWindow.set_modal(True)
555 self._popupWindow.set_type_hint(gtk.gdk.WINDOW_TYPE_HINT_DIALOG)
556 self._popupWindow.set_skip_pager_hint(True)
557 self._popupWindow.set_skip_taskbar_hint(True)
560 self._popupWindow.show_all()
562 def _on_day_selected(self, *args):
564 self._calendar.select_month(self._displayDate.month, self._displayDate.year)
565 self._calendar.select_day(self._displayDate.day)
567 _moduleLogger.exception(e)
570 class QuickAddView(object):
572 def __init__(self, widgetTree, errorDisplay, signalSink, prefix):
573 self._errorDisplay = errorDisplay
575 self._signalSink = signalSink
577 self._clipboard = gtk.clipboard_get()
579 self._taskNameEntry = widgetTree.get_widget(prefix+"-nameEntry")
580 self._addTaskButton = widgetTree.get_widget(prefix+"-addButton")
581 self._pasteTaskNameButton = widgetTree.get_widget(prefix+"-pasteNameButton")
582 self._clearTaskNameButton = widgetTree.get_widget(prefix+"-clearNameButton")
584 self._onAddClickedId = None
585 self._onAddReleasedId = None
586 self._addToEditTimerId = None
587 self._onClearId = None
588 self._onPasteId = None
590 def enable(self, manager):
591 self._manager = manager
593 self._onAddId = self._addTaskButton.connect("clicked", self._on_add)
594 self._onAddClickedId = self._addTaskButton.connect("pressed", self._on_add_pressed)
595 self._onAddReleasedId = self._addTaskButton.connect("released", self._on_add_released)
596 self._onPasteId = self._pasteTaskNameButton.connect("clicked", self._on_paste)
597 self._onClearId = self._clearTaskNameButton.connect("clicked", self._on_clear)
602 self._addTaskButton.disconnect(self._onAddId)
603 self._addTaskButton.disconnect(self._onAddClickedId)
604 self._addTaskButton.disconnect(self._onAddReleasedId)
605 self._pasteTaskNameButton.disconnect(self._onPasteId)
606 self._clearTaskNameButton.disconnect(self._onClearId)
608 def set_addability(self, addability):
609 self._addTaskButton.set_sensitive(addability)
611 def _on_add(self, *args):
613 name = self._taskNameEntry.get_text()
614 self._taskNameEntry.set_text("")
616 self._signalSink.stage.send(("add", name))
618 self._errorDisplay.push_exception()
620 def _on_add_edit(self, *args):
622 name = self._taskNameEntry.get_text()
623 self._taskNameEntry.set_text("")
625 self._signalSink.stage.send(("add-edit", name))
627 self._errorDisplay.push_exception()
629 def _on_add_pressed(self, widget):
631 self._addToEditTimerId = gobject.timeout_add(1000, self._on_add_edit)
633 self._errorDisplay.push_exception()
635 def _on_add_released(self, widget):
637 if self._addToEditTimerId is not None:
638 gobject.source_remove(self._addToEditTimerId)
639 self._addToEditTimerId = None
641 self._errorDisplay.push_exception()
643 def _on_paste(self, *args):
645 entry = self._taskNameEntry.get_text()
646 addedText = self._clipboard.wait_for_text()
649 self._taskNameEntry.set_text(entry)
651 self._errorDisplay.push_exception()
653 def _on_clear(self, *args):
655 self._taskNameEntry.set_text("")
657 self._errorDisplay.push_exception()
660 class TapOrHold(object):
662 def __init__(self, widget):
663 self._widget = widget
665 self._isPointerInside = True
666 self._holdTimeoutId = None
667 self._tapTimeoutId = None
676 self._bpeId = self._widget.connect("button-press-event", self._on_button_press)
677 self._breId = self._widget.connect("button-release-event", self._on_button_release)
678 self._eneId = self._widget.connect("enter-notify-event", self._on_enter)
679 self._lneId = self._widget.connect("leave-notify-event", self._on_leave)
682 self._widget.disconnect(self._bpeId)
683 self._widget.disconnect(self._breId)
684 self._widget.disconnect(self._eneId)
685 self._widget.disconnect(self._lneId)
687 def on_tap(self, taps):
690 def on_hold(self, taps):
693 def on_holding(self):
699 def _on_button_press(self, *args):
700 # Hack to handle weird notebook behavior
701 self._isPointerInside = True
704 if self._tapTimeoutId is not None:
705 gobject.source_remove(self._tapTimeoutId)
706 self._tapTimeoutId = None
709 if self._holdTimeoutId is None:
710 self._tapTimeoutId = None
713 self._holdTimeoutId = gobject.timeout_add(1000, self._on_hold_timeout)
717 def _on_button_release(self, *args):
718 assert self._tapTimeoutId is None
719 # Handle release after timeout if user hasn't double-clicked
720 self._tapTimeoutId = gobject.timeout_add(100, self._on_tap_timeout)
722 def _on_actual_press(self, *args):
723 if self._holdTimeoutId is not None:
724 gobject.source_remove(self._holdTimeoutId)
725 self._holdTimeoutId = None
727 if self._isPointerInside:
729 self.on_tap(self._taps)
731 self.on_hold(self._taps)
735 def _on_tap_timeout(self, *args):
736 self._tapTimeoutId = None
737 self._on_actual_press()
740 def _on_hold_timeout(self, *args):
741 self._holdTimeoutId = None
746 def _on_enter(self, *args):
747 self._isPointerInside = True
749 def _on_leave(self, *args):
750 self._isPointerInside = False
753 if __name__ == "__main__":
756 cal = PopupCalendar(None, datetime.datetime.now())
757 cal._popupWindow.connect("destroy", lambda w: gtk.main_quit())