3 from __future__ import with_statement
20 @contextlib.contextmanager
21 def flock(path, timeout=-1):
28 while timeSpent <= timeout or timeout == WAIT_FOREVER:
30 fd = os.open(path, os.O_CREAT | os.O_EXCL | os.O_RDWR)
34 if e.errno != errno.EEXIST:
39 assert acquired, "Failed to grab file-lock %s within timeout %d" % (path, timeout)
47 @contextlib.contextmanager
49 gtk.gdk.threads_enter()
53 gtk.gdk.threads_leave()
56 def find_parent_window(widget):
58 parent = widget.get_parent()
59 if isinstance(parent, gtk.Window):
66 Decorator that makes a generator-function into a function that will continue execution on next call
70 @functools.wraps(func)
71 def decorated_func(*args, **kwds):
73 a.append(func(*args, **kwds))
84 def asynchronous_gtk_message(original_func):
86 @note Idea came from http://www.aclevername.com/articles/python-webgui/
90 args, kwargs = allArgs
92 original_func(*args, **kwargs)
95 @functools.wraps(original_func)
96 def delayed_func(*args, **kwargs):
97 gobject.idle_add(execute, (args, kwargs))
102 def synchronous_gtk_message(original_func):
104 @note Idea came from http://www.aclevername.com/articles/python-webgui/
107 @functools.wraps(original_func)
108 def immediate_func(*args, **kwargs):
110 return original_func(*args, **kwargs)
112 return immediate_func
118 ... def grep_sink(pattern):
119 ... print "Looking for %s" % pattern
122 ... if pattern in line:
124 >>> g = grep_sink("python")
126 >>> g.send("Yeah but no but yeah but no")
127 >>> g.send("A series of tubes")
128 >>> g.send("python generators rock!")
129 python generators rock!
133 @functools.wraps(func)
134 def start(*args, **kwargs):
135 cr = func(*args, **kwargs)
143 def printer_sink(format = "%s"):
145 >>> pr = printer_sink("%r")
152 >>> p = printer_sink()
157 >>> # p.throw(RuntimeError, "Goodbye")
163 print format % (item, )
169 Good for uses like with cochain to pick up any slack
176 def comap(function, target):
178 >>> p = printer_sink()
179 >>> cm = comap(lambda x: x+1, p)
190 mappedItem = function(*item)
191 target.send(mappedItem)
192 except StandardError, e:
193 target.throw(e.__class__, e.message)
196 def _flush_queue(queue):
197 while not queue.empty():
202 def queue_sink(queue):
204 >>> q = Queue.Queue()
205 >>> qs = queue_sink(q)
208 >>> qs.throw(RuntimeError, "Goodbye")
211 >>> print [i for i in _flush_queue(q)]
212 [(None, 'Hello'), (None, 'World'), (<type 'exceptions.RuntimeError'>, 'Goodbye'), (None, 'Meh'), (<type 'exceptions.GeneratorExit'>, None)]
217 queue.put((None, item))
218 except StandardError, e:
219 queue.put((e.__class__, str(e)))
220 except GeneratorExit:
221 queue.put((GeneratorExit, None))
225 def decode_item(item, target):
229 elif item[0] is GeneratorExit:
233 target.throw(item[0], item[1])
237 def nonqueue_source(queue, target):
241 isDone = decode_item(item, target)
242 while not queue.empty():
246 def threaded_stage(target, thread_factory = threading.Thread):
247 messages = Queue.Queue()
249 run_source = functools.partial(nonqueue_source, messages, target)
250 thread = thread_factory(target=run_source)
251 thread.setDaemon(True)
254 # Sink running in current thread
255 return queue_sink(messages)
258 class LoginWindow(object):
260 def __init__(self, widgetTree):
262 @note Thread agnostic
264 self._dialog = widgetTree.get_widget("loginDialog")
265 self._parentWindow = widgetTree.get_widget("mainWindow")
266 self._serviceCombo = widgetTree.get_widget("serviceCombo")
267 self._usernameEntry = widgetTree.get_widget("usernameentry")
268 self._passwordEntry = widgetTree.get_widget("passwordentry")
270 self._serviceList = gtk.ListStore(gobject.TYPE_INT, gobject.TYPE_STRING)
271 self._serviceCombo.set_model(self._serviceList)
272 cell = gtk.CellRendererText()
273 self._serviceCombo.pack_start(cell, True)
274 self._serviceCombo.add_attribute(cell, 'text', 1)
275 self._serviceCombo.set_active(0)
278 "on_loginbutton_clicked": self._on_loginbutton_clicked,
279 "on_loginclose_clicked": self._on_loginclose_clicked,
281 widgetTree.signal_autoconnect(callbackMapping)
283 def request_credentials(self,
285 defaultCredentials = ("", "")
290 if parentWindow is None:
291 parentWindow = self._parentWindow
293 self._serviceCombo.hide()
294 self._serviceList.clear()
296 self._usernameEntry.set_text(defaultCredentials[0])
297 self._passwordEntry.set_text(defaultCredentials[1])
300 self._dialog.set_transient_for(parentWindow)
301 self._dialog.set_default_response(gtk.RESPONSE_OK)
302 response = self._dialog.run()
303 if response != gtk.RESPONSE_OK:
304 raise RuntimeError("Login Cancelled")
306 username = self._usernameEntry.get_text()
307 password = self._passwordEntry.get_text()
308 self._passwordEntry.set_text("")
312 return username, password
314 def request_credentials_from(self,
317 defaultCredentials = ("", "")
322 if parentWindow is None:
323 parentWindow = self._parentWindow
325 self._serviceList.clear()
326 for serviceIdserviceName in services:
327 self._serviceList.append(serviceIdserviceName)
328 self._serviceCombo.set_active(0)
329 self._serviceCombo.show()
331 self._usernameEntry.set_text(defaultCredentials[0])
332 self._passwordEntry.set_text(defaultCredentials[1])
335 self._dialog.set_transient_for(parentWindow)
336 self._dialog.set_default_response(gtk.RESPONSE_OK)
337 response = self._dialog.run()
338 if response != gtk.RESPONSE_OK:
339 raise RuntimeError("Login Cancelled")
341 username = self._usernameEntry.get_text()
342 password = self._passwordEntry.get_text()
346 itr = self._serviceCombo.get_active_iter()
347 serviceId = int(self._serviceList.get_value(itr, 0))
348 self._serviceList.clear()
349 return serviceId, username, password
351 def _on_loginbutton_clicked(self, *args):
352 self._dialog.response(gtk.RESPONSE_OK)
354 def _on_loginclose_clicked(self, *args):
355 self._dialog.response(gtk.RESPONSE_CANCEL)
358 def safecall(f, errorDisplay=None, default=None, exception=Exception):
360 Returns modified f. When the modified f is called and throws an
361 exception, the default value is returned
363 def _safecall(*args, **argv):
365 return f(*args,**argv)
367 if errorDisplay is not None:
368 errorDisplay.push_exception(e)
373 class ErrorDisplay(object):
375 def __init__(self, widgetTree):
376 super(ErrorDisplay, self).__init__()
377 self.__errorBox = widgetTree.get_widget("errorEventBox")
378 self.__errorDescription = widgetTree.get_widget("errorDescription")
379 self.__errorClose = widgetTree.get_widget("errorClose")
380 self.__parentBox = self.__errorBox.get_parent()
382 self.__errorBox.connect("button_release_event", self._on_close)
385 self.__parentBox.remove(self.__errorBox)
387 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_with_lock(self, exception = None, stacklevel=3):
399 self.push_exception(exception, stacklevel=stacklevel)
401 def push_exception(self, exception = None, stacklevel=2):
402 if exception is None:
403 userMessage = str(sys.exc_value)
404 warningMessage = str(traceback.format_exc())
406 userMessage = str(exception)
407 warningMessage = str(exception)
408 self.push_message(userMessage)
409 warnings.warn(warningMessage, stacklevel=stacklevel)
411 def pop_message(self):
412 if 0 < len(self.__messages):
413 self.__show_message(self.__messages[0])
414 del self.__messages[0]
416 self.__hide_message()
418 def _on_close(self, *args):
421 def __show_message(self, message):
422 self.__errorDescription.set_text(message)
423 self.__parentBox.pack_start(self.__errorBox, False, False)
424 self.__parentBox.reorder_child(self.__errorBox, 1)
426 def __hide_message(self):
427 self.__errorDescription.set_text("")
428 self.__parentBox.remove(self.__errorBox)
431 class DummyErrorDisplay(object):
434 super(DummyErrorDisplay, self).__init__()
438 def push_message_with_lock(self, message):
439 self.push_message(message)
441 def push_message(self, message):
442 if 0 < len(self.__messages):
443 self.__messages.append(message)
445 self.__show_message(message)
447 def push_exception(self, exception = None):
448 if exception is None:
449 warningMessage = traceback.format_exc()
451 warningMessage = exception
452 warnings.warn(warningMessage, stacklevel=3)
454 def pop_message(self):
455 if 0 < len(self.__messages):
456 self.__show_message(self.__messages[0])
457 del self.__messages[0]
459 def __show_message(self, message):
460 warnings.warn(message, stacklevel=2)
463 class MessageBox(gtk.MessageDialog):
465 def __init__(self, message):
467 gtk.MessageDialog.__init__(
470 gtk.DIALOG_MODAL|gtk.DIALOG_DESTROY_WITH_PARENT,
475 self.set_default_response(gtk.RESPONSE_OK)
476 self.connect('response', self._handle_clicked)
478 def _handle_clicked(self, *args):
482 class MessageBox2(gtk.MessageDialog):
484 def __init__(self, message):
486 gtk.MessageDialog.__init__(
489 gtk.DIALOG_DESTROY_WITH_PARENT,
494 self.set_default_response(gtk.RESPONSE_OK)
495 self.connect('response', self._handle_clicked)
497 def _handle_clicked(self, *args):
501 class PopupCalendar(object):
503 def __init__(self, parent, displayDate, title = ""):
504 self._displayDate = displayDate
506 self._calendar = gtk.Calendar()
507 self._calendar.select_month(self._displayDate.month, self._displayDate.year)
508 self._calendar.select_day(self._displayDate.day)
509 self._calendar.set_display_options(
510 gtk.CALENDAR_SHOW_HEADING |
511 gtk.CALENDAR_SHOW_DAY_NAMES |
512 gtk.CALENDAR_NO_MONTH_CHANGE |
515 self._calendar.connect("day-selected", self._on_day_selected)
517 self._popupWindow = gtk.Window()
518 self._popupWindow.set_title(title)
519 self._popupWindow.add(self._calendar)
520 self._popupWindow.set_transient_for(parent)
521 self._popupWindow.set_modal(True)
522 self._popupWindow.set_type_hint(gtk.gdk.WINDOW_TYPE_HINT_DIALOG)
523 self._popupWindow.set_skip_pager_hint(True)
524 self._popupWindow.set_skip_taskbar_hint(True)
527 self._popupWindow.show_all()
529 def _on_day_selected(self, *args):
531 self._calendar.select_month(self._displayDate.month, self._displayDate.year)
532 self._calendar.select_day(self._displayDate.day)
533 except StandardError, e:
534 warnings.warn(e.message)
537 class QuickAddView(object):
539 def __init__(self, widgetTree, errorDisplay, signalSink, prefix):
540 self._errorDisplay = errorDisplay
542 self._signalSink = signalSink
544 self._clipboard = gtk.clipboard_get()
546 self._taskNameEntry = widgetTree.get_widget(prefix+"-nameEntry")
547 self._addTaskButton = widgetTree.get_widget(prefix+"-addButton")
548 self._pasteTaskNameButton = widgetTree.get_widget(prefix+"-pasteNameButton")
549 self._clearTaskNameButton = widgetTree.get_widget(prefix+"-clearNameButton")
551 self._onAddClickedId = None
552 self._onAddReleasedId = None
553 self._addToEditTimerId = None
554 self._onClearId = None
555 self._onPasteId = None
557 def enable(self, manager):
558 self._manager = manager
560 self._onAddId = self._addTaskButton.connect("clicked", self._on_add)
561 self._onAddClickedId = self._addTaskButton.connect("pressed", self._on_add_pressed)
562 self._onAddReleasedId = self._addTaskButton.connect("released", self._on_add_released)
563 self._onPasteId = self._pasteTaskNameButton.connect("clicked", self._on_paste)
564 self._onClearId = self._clearTaskNameButton.connect("clicked", self._on_clear)
569 self._addTaskButton.disconnect(self._onAddId)
570 self._addTaskButton.disconnect(self._onAddClickedId)
571 self._addTaskButton.disconnect(self._onAddReleasedId)
572 self._pasteTaskNameButton.disconnect(self._onPasteId)
573 self._clearTaskNameButton.disconnect(self._onClearId)
575 def set_addability(self, addability):
576 self._addTaskButton.set_sensitive(addability)
578 def _on_add(self, *args):
580 name = self._taskNameEntry.get_text()
581 self._taskNameEntry.set_text("")
583 self._signalSink.stage.send(("add", name))
584 except StandardError, e:
585 self._errorDisplay.push_exception()
587 def _on_add_edit(self, *args):
589 name = self._taskNameEntry.get_text()
590 self._taskNameEntry.set_text("")
592 self._signalSink.stage.send(("add-edit", name))
593 except StandardError, e:
594 self._errorDisplay.push_exception()
596 def _on_add_pressed(self, widget):
598 self._addToEditTimerId = gobject.timeout_add(1000, self._on_add_edit)
599 except StandardError, e:
600 self._errorDisplay.push_exception()
602 def _on_add_released(self, widget):
604 if self._addToEditTimerId is not None:
605 gobject.source_remove(self._addToEditTimerId)
606 self._addToEditTimerId = None
607 except StandardError, e:
608 self._errorDisplay.push_exception()
610 def _on_paste(self, *args):
612 entry = self._taskNameEntry.get_text()
613 addedText = self._clipboard.wait_for_text()
616 self._taskNameEntry.set_text(entry)
617 except StandardError, e:
618 self._errorDisplay.push_exception()
620 def _on_clear(self, *args):
622 self._taskNameEntry.set_text("")
623 except StandardError, e:
624 self._errorDisplay.push_exception()
627 class TapOrHold(object):
629 def __init__(self, widget):
630 self._widget = widget
632 self._isPointerInside = True
633 self._holdTimeoutId = None
634 self._tapTimeoutId = None
643 self._bpeId = self._widget.connect("button-press-event", self._on_button_press)
644 self._breId = self._widget.connect("button-release-event", self._on_button_release)
645 self._eneId = self._widget.connect("enter-notify-event", self._on_enter)
646 self._lneId = self._widget.connect("leave-notify-event", self._on_leave)
649 self._widget.disconnect(self._bpeId)
650 self._widget.disconnect(self._breId)
651 self._widget.disconnect(self._eneId)
652 self._widget.disconnect(self._lneId)
654 def on_tap(self, taps):
657 def on_hold(self, taps):
660 def on_holding(self):
666 def _on_button_press(self, *args):
667 # Hack to handle weird notebook behavior
668 self._isPointerInside = True
671 if self._tapTimeoutId is not None:
672 gobject.source_remove(self._tapTimeoutId)
673 self._tapTimeoutId = None
676 if self._holdTimeoutId is None:
677 self._tapTimeoutId = None
680 self._holdTimeoutId = gobject.timeout_add(1000, self._on_hold_timeout)
684 def _on_button_release(self, *args):
685 assert self._tapTimeoutId is None
686 # Handle release after timeout if user hasn't double-clicked
687 self._tapTimeoutId = gobject.timeout_add(100, self._on_tap_timeout)
689 def _on_actual_press(self, *args):
690 if self._holdTimeoutId is not None:
691 gobject.source_remove(self._holdTimeoutId)
692 self._holdTimeoutId = None
694 if self._isPointerInside:
696 self.on_tap(self._taps)
698 self.on_hold(self._taps)
702 def _on_tap_timeout(self, *args):
703 self._tapTimeoutId = None
704 self._on_actual_press()
707 def _on_hold_timeout(self, *args):
708 self._holdTimeoutId = None
713 def _on_enter(self, *args):
714 self._isPointerInside = True
716 def _on_leave(self, *args):
717 self._isPointerInside = False
720 if __name__ == "__main__":
723 cal = PopupCalendar(None, datetime.datetime.now())
724 cal._popupWindow.connect("destroy", lambda w: gtk.main_quit())