3 # DialCentral - Front end for Google's Grand Central service.
4 # Copyright (C) 2008 Mark Bergman bergman AT merctech DOT com
6 # This library is free software; you can redistribute it and/or
7 # modify it under the terms of the GNU Lesser General Public
8 # License as published by the Free Software Foundation; either
9 # version 2.1 of the License, or (at your option) any later version.
11 # This library is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 # Lesser General Public License for more details.
16 # You should have received a copy of the GNU Lesser General Public
17 # License along with this library; if not, write to the Free Software
18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
28 def make_ugly(prettynumber):
30 function to take a phone number and strip out all non-numeric
33 >>> make_ugly("+012-(345)-678-90")
37 uglynumber = re.sub('\D', '', prettynumber)
41 def make_pretty(phonenumber):
43 Function to take a phone number and return the pretty version
45 if phonenumber begins with 0:
47 if phonenumber begins with 1: ( for gizmo callback numbers )
49 if phonenumber is 13 digits:
51 if phonenumber is 10 digits:
55 >>> make_pretty("1234567")
57 >>> make_pretty("2345678901")
59 >>> make_pretty("12345678901")
61 >>> make_pretty("01234567890")
64 if phonenumber is None or phonenumber is "":
67 if len(phonenumber) < 3:
70 if phonenumber[0] == "0":
72 prettynumber += "+%s" % phonenumber[0:3]
73 if 3 < len(phonenumber):
74 prettynumber += "-(%s)" % phonenumber[3:6]
75 if 6 < len(phonenumber):
76 prettynumber += "-%s" % phonenumber[6:9]
77 if 9 < len(phonenumber):
78 prettynumber += "-%s" % phonenumber[9:]
80 elif len(phonenumber) <= 7:
81 prettynumber = "%s-%s" % (phonenumber[0:3], phonenumber[3:])
82 elif len(phonenumber) > 8 and phonenumber[0] == "1":
83 prettynumber = "1 (%s)-%s-%s" % (phonenumber[1:4], phonenumber[4:7], phonenumber[7:])
84 elif len(phonenumber) > 7:
85 prettynumber = "(%s)-%s-%s" % (phonenumber[0:3], phonenumber[3:6], phonenumber[6:])
91 Decorator that makes a generator-function into a function that will continue execution on next call
95 def decorated_func(*args, **kwds):
97 a.append(func(*args, **kwds))
101 except StopIteration:
105 decorated_func.__name__ = func.__name__
106 decorated_func.__doc__ = func.__doc__
107 decorated_func.__dict__.update(func.__dict__)
109 return decorated_func
112 class DummyAddressBook(object):
114 Minimal example of both an addressbook factory and an addressbook
117 def clear_caches(self):
120 def get_addressbooks(self):
122 @returns Iterable of (Address Book Factory, Book Id, Book Name)
124 yield self, "", "None"
126 def open_addressbook(self, bookId):
130 def contact_source_short_name(contactId):
140 @returns Iterable of (contact id, contact name)
145 def get_contact_details(contactId):
147 @returns Iterable of (Phone Type, Phone Number)
152 class MergedAddressBook(object):
154 Merger of all addressbooks
157 def __init__(self, addressbookFactories, sorter = None):
158 self.__addressbookFactories = addressbookFactories
159 self.__addressbooks = None
160 self.__sort_contacts = sorter if sorter is not None else self.null_sorter
162 def clear_caches(self):
163 self.__addressbooks = None
164 for factory in self.__addressbookFactories:
165 factory.clear_caches()
167 def get_addressbooks(self):
169 @returns Iterable of (Address Book Factory, Book Id, Book Name)
173 def open_addressbook(self, bookId):
176 def contact_source_short_name(self, contactId):
177 if self.__addressbooks is None:
179 bookIndex, originalId = contactId.split("-", 1)
180 return self.__addressbooks[int(bookIndex)].contact_source_short_name(originalId)
184 return "All Contacts"
186 def get_contacts(self):
188 @returns Iterable of (contact id, contact name)
190 if self.__addressbooks is None:
191 self.__addressbooks = list(
192 factory.open_addressbook(id)
193 for factory in self.__addressbookFactories
194 for (f, id, name) in factory.get_addressbooks()
197 ("-".join([str(bookIndex), contactId]), contactName)
198 for (bookIndex, addressbook) in enumerate(self.__addressbooks)
199 for (contactId, contactName) in addressbook.get_contacts()
201 sortedContacts = self.__sort_contacts(contacts)
202 return sortedContacts
204 def get_contact_details(self, contactId):
206 @returns Iterable of (Phone Type, Phone Number)
208 if self.__addressbooks is None:
210 bookIndex, originalId = contactId.split("-", 1)
211 return self.__addressbooks[int(bookIndex)].get_contact_details(originalId)
214 def null_sorter(contacts):
216 Good for speed/low memory
221 def basic_firtname_sorter(contacts):
223 Expects names in "First Last" format
226 (contactName.rsplit(" ", 1)[0], (contactId, contactName))
227 for (contactId, contactName) in contacts
229 contactsWithKey.sort()
230 return (contactData for (lastName, contactData) in contactsWithKey)
233 def basic_lastname_sorter(contacts):
235 Expects names in "First Last" format
238 (contactName.rsplit(" ", 1)[-1], (contactId, contactName))
239 for (contactId, contactName) in contacts
241 contactsWithKey.sort()
242 return (contactData for (lastName, contactData) in contactsWithKey)
245 def reversed_firtname_sorter(contacts):
247 Expects names in "Last, First" format
250 (contactName.split(", ", 1)[-1], (contactId, contactName))
251 for (contactId, contactName) in contacts
253 contactsWithKey.sort()
254 return (contactData for (lastName, contactData) in contactsWithKey)
257 def reversed_lastname_sorter(contacts):
259 Expects names in "Last, First" format
262 (contactName.split(", ", 1)[0], (contactId, contactName))
263 for (contactId, contactName) in contacts
265 contactsWithKey.sort()
266 return (contactData for (lastName, contactData) in contactsWithKey)
269 def guess_firstname(name):
271 return name.split(", ", 1)[-1]
273 return name.rsplit(" ", 1)[0]
276 def guess_lastname(name):
278 return name.split(", ", 1)[0]
280 return name.rsplit(" ", 1)[-1]
283 def advanced_firtname_sorter(cls, contacts):
285 (cls.guess_firstname(contactName), (contactId, contactName))
286 for (contactId, contactName) in contacts
288 contactsWithKey.sort()
289 return (contactData for (lastName, contactData) in contactsWithKey)
292 def advanced_lastname_sorter(cls, contacts):
294 (cls.guess_lastname(contactName), (contactId, contactName))
295 for (contactId, contactName) in contacts
297 contactsWithKey.sort()
298 return (contactData for (lastName, contactData) in contactsWithKey)
301 class PhoneTypeSelector(object):
303 def __init__(self, widgetTree, gcBackend):
304 self._gcBackend = gcBackend
305 self._widgetTree = widgetTree
306 self._dialog = self._widgetTree.get_widget("phonetype_dialog")
308 self._selectButton = self._widgetTree.get_widget("select_button")
309 self._selectButton.connect("clicked", self._on_phonetype_select)
311 self._cancelButton = self._widgetTree.get_widget("cancel_button")
312 self._cancelButton.connect("clicked", self._on_phonetype_cancel)
314 self._typemodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
315 self._typeviewselection = None
317 typeview = self._widgetTree.get_widget("phonetypes")
318 typeview.connect("row-activated", self._on_phonetype_select)
319 typeview.set_model(self._typemodel)
320 textrenderer = gtk.CellRendererText()
322 # Add the column to the treeview
323 column = gtk.TreeViewColumn("Phone Numbers", textrenderer, text=1)
324 column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
326 typeview.append_column(column)
328 self._typeviewselection = typeview.get_selection()
329 self._typeviewselection.set_mode(gtk.SELECTION_SINGLE)
331 def run(self, contactDetails):
332 self._typemodel.clear()
334 for phoneType, phoneNumber in contactDetails:
335 self._typemodel.append((phoneNumber, "%s - %s" % (make_pretty(phoneNumber), phoneType)))
337 userResponse = self._dialog.run()
339 if userResponse == gtk.RESPONSE_OK:
340 model, itr = self._typeviewselection.get_selected()
342 phoneNumber = self._typemodel.get_value(itr, 0)
346 self._typeviewselection.unselect_all()
350 def _on_phonetype_select(self, *args):
351 self._dialog.response(gtk.RESPONSE_OK)
353 def _on_phonetype_cancel(self, *args):
354 self._dialog.response(gtk.RESPONSE_CANCEL)
357 class Dialpad(object):
359 def __init__(self, widgetTree):
360 self._numberdisplay = widgetTree.get_widget("numberdisplay")
361 self._dialButton = widgetTree.get_widget("dial")
362 self._phonenumber = ""
363 self._prettynumber = ""
364 self._clearall_id = None
367 "on_dial_clicked": self._on_dial_clicked,
368 "on_digit_clicked": self._on_digit_clicked,
369 "on_clear_number": self._on_clear_number,
370 "on_back_clicked": self._on_backspace,
371 "on_back_pressed": self._on_back_pressed,
372 "on_back_released": self._on_back_released,
374 widgetTree.signal_autoconnect(callbackMapping)
377 self._dialButton.grab_focus()
382 def dial(self, number):
384 @note Actual dial function is patched in later
386 raise NotImplementedError
388 def get_number(self):
389 return self._phonenumber
391 def set_number(self, number):
393 Set the callback phonenumber
395 self._phonenumber = make_ugly(number)
396 self._prettynumber = make_pretty(self._phonenumber)
397 self._numberdisplay.set_label("<span size='30000' weight='bold'>%s</span>" % (self._prettynumber))
402 def _on_dial_clicked(self, widget):
403 self.dial(self.get_number())
405 def _on_clear_number(self, *args):
408 def _on_digit_clicked(self, widget):
409 self.set_number(self._phonenumber + widget.get_name()[-1])
411 def _on_backspace(self, widget):
412 self.set_number(self._phonenumber[:-1])
414 def _on_clearall(self):
418 def _on_back_pressed(self, widget):
419 self._clearall_id = gobject.timeout_add(1000, self._on_clearall)
421 def _on_back_released(self, widget):
422 if self._clearall_id is not None:
423 gobject.source_remove(self._clearall_id)
424 self._clearall_id = None
427 class AccountInfo(object):
429 def __init__(self, widgetTree, backend = None):
430 self._backend = backend
432 self._callbackList = gtk.ListStore(gobject.TYPE_STRING)
433 self._accountViewNumberDisplay = widgetTree.get_widget("gcnumber_display")
434 self._callbackCombo = widgetTree.get_widget("callbackcombo")
437 assert self._backend.is_authed()
438 self.set_account_number("")
439 self._callbackList.clear()
441 self._callbackCombo.get_child().connect("changed", self._on_callbackentry_changed)
444 self._callbackCombo.get_child().disconnect("changed", self._on_callbackentry_changed)
446 self._callbackList.clear()
448 def get_selected_callback_number(self):
449 return make_ugly(self._callbackCombo.get_child().get_text())
451 def set_account_number(self, number):
453 Displays current account number
455 self._accountViewNumberDisplay.set_label("<span size='23000' weight='bold'>%s</span>" % (number))
458 self.populate_callback_combo()
459 self.set_account_number(self._backend.get_account_number())
462 self._callbackCombo.get_child().set_text("")
463 self.set_account_number("")
465 def populate_callback_combo(self):
466 self._callbackList.clear()
467 for number, description in self._backend.get_callback_numbers().iteritems():
468 self._callbackList.append((make_pretty(number),))
470 self._callbackCombo.set_model(self._callbackList)
471 self._callbackCombo.set_text_column(0)
472 self._callbackCombo.get_child().set_text(make_pretty(self._backend.get_callback_number()))
474 def _on_callbackentry_changed(self, *args):
476 @todo Potential blocking on web access, maybe we should defer this or put up a dialog?
478 text = self.get_selected_callback_number()
479 if not self._backend.is_valid_syntax(text):
480 warnings.warn("%s is not a valid callback number" % text, UserWarning, 2)
481 elif text == self._backend.get_callback_number():
482 warnings.warn("Callback number already is %s" % self._backend.get_callback_number(), UserWarning, 2)
484 self._backend.set_callback_number(text)
487 class RecentCallsView(object):
489 def __init__(self, widgetTree, backend = None):
490 self._backend = backend
491 self._recenttime = 0.0
492 self._recentmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
493 self._recentview = widgetTree.get_widget("recentview")
494 self._recentviewselection = None
495 textrenderer = gtk.CellRendererText()
496 self._recentviewColumn = gtk.TreeViewColumn("Calls", textrenderer, text=1)
497 self._recentviewColumn.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
500 assert self._backend.is_authed()
501 self._recentview.set_model(self._recentmodel)
503 self._recentview.append_column(self._recentviewColumn)
504 self._recentviewselection = self._recentview.get_selection()
505 self._recentviewselection.set_mode(gtk.SELECTION_SINGLE)
507 self._recentview.connect("row-activated", self._on_recentview_row_activated)
510 self._recentview.disconnect("row-activated", self._on_recentview_row_activated)
511 self._recentview.remove_column(self._recentviewColumn)
512 self._recentview.set_model(None)
514 def number_selected(self, number):
516 @note Actual dial function is patched in later
518 raise NotImplementedError
521 if (time.time() - self._recenttime) < 300:
523 backgroundPopulate = threading.Thread(target=self._idly_populate_recentview)
524 backgroundPopulate.setDaemon(True)
525 backgroundPopulate.start()
528 self._recenttime = 0.0
529 self._recentmodel.clear()
531 def _idly_populate_recentview(self):
532 self._recenttime = time.time()
533 self._recentmodel.clear()
535 for personsName, phoneNumber, date, action in self._backend.get_recent():
536 description = "%s on %s from/to %s - %s" % (action.capitalize(), date, personsName, phoneNumber)
537 item = (phoneNumber, description)
538 gtk.gdk.threads_enter()
540 self._recentmodel.append(item)
542 gtk.gdk.threads_leave()
546 def _on_recentview_row_activated(self, treeview, path, view_column):
547 model, itr = self._recentviewselection.get_selected()
551 self.number_selected(self._recentmodel.get_value(itr, 0))
552 self._recentviewselection.unselect_all()
555 class ContactsView(object):
557 def __init__(self, widgetTree, backend = None):
558 self._backend = backend
560 self._addressBook = None
561 self._addressBookFactories = [DummyAddressBook()]
563 self._booksList = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
564 self._booksSelectionBox = widgetTree.get_widget("addressbook_combo")
566 self._contactstime = 0.0
567 self._contactsmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
568 self._contactsviewselection = None
569 self._contactsview = widgetTree.get_widget("contactsview")
571 self._contactColumn = gtk.TreeViewColumn("Contact")
572 displayContactSource = True
573 if displayContactSource:
574 textrenderer = gtk.CellRendererText()
575 self._contactColumn.pack_start(textrenderer, expand=False)
576 self._contactColumn.add_attribute(textrenderer, 'text', 0)
577 textrenderer = gtk.CellRendererText()
578 self._contactColumn.pack_start(textrenderer, expand=True)
579 self._contactColumn.add_attribute(textrenderer, 'text', 1)
580 textrenderer = gtk.CellRendererText()
581 self._contactColumn.pack_start(textrenderer, expand=True)
582 self._contactColumn.add_attribute(textrenderer, 'text', 4)
583 self._contactColumn.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
584 self._contactColumn.set_sort_column_id(1)
585 self._contactColumn.set_visible(True)
587 self._phoneTypeSelector = PhoneTypeSelector(widgetTree, self._backend)
590 assert self._backend.is_authed()
592 self._contactsview.set_model(self._contactsmodel)
593 self._contactsview.append_column(self._contactColumn)
594 self._contactsviewselection = self._contactsview.get_selection()
595 self._contactsviewselection.set_mode(gtk.SELECTION_SINGLE)
597 self._booksList.clear()
598 for (factoryId, bookId), (factoryName, bookName) in self.get_addressbooks():
599 if factoryName and bookName:
600 entryName = "%s: %s" % (factoryName, bookName)
602 entryName = factoryName
606 entryName = "Bad name (%d)" % factoryId
607 row = (str(factoryId), bookId, entryName)
608 self._booksList.append(row)
610 self._booksSelectionBox.set_model(self._booksList)
611 cell = gtk.CellRendererText()
612 self._booksSelectionBox.pack_start(cell, True)
613 self._booksSelectionBox.add_attribute(cell, 'text', 2)
614 self._booksSelectionBox.set_active(0)
616 self._contactsview.connect("row-activated", self._on_contactsview_row_activated)
617 self._booksSelectionBox.connect("changed", self._on_addressbook_combo_changed)
620 self._booksSelectionBox.set_model(None)
621 self._contactsview.set_model(None)
622 self._contactsview.remove_column(self._contactColumn)
623 self._contactsview.disconnect("row-activated", self._on_contactsview_row_activated)
624 self._booksSelectionBox.disconnect("changed", self._on_addressbook_combo_changed)
626 def number_selected(self, number):
628 @note Actual dial function is patched in later
630 raise NotImplementedError
632 def get_addressbooks(self):
634 @returns Iterable of ((Factory Id, Book Id), (Factory Name, Book Name))
636 for i, factory in enumerate(self._addressBookFactories):
637 for bookFactory, bookId, bookName in factory.get_addressbooks():
638 yield (i, bookId), (factory.factory_name(), bookName)
640 def open_addressbook(self, bookFactoryId, bookId):
641 self._addressBook = self._addressBookFactories[bookFactoryId].open_addressbook(bookId)
642 self._contactstime = 0
643 backgroundPopulate = threading.Thread(target=self._idly_populate_contactsview)
644 backgroundPopulate.setDaemon(True)
645 backgroundPopulate.start()
648 if (time.time() - self._contactstime) < 300:
650 backgroundPopulate = threading.Thread(target=self._idly_populate_contactsview)
651 backgroundPopulate.setDaemon(True)
652 backgroundPopulate.start()
655 self._contactstime = 0.0
656 self._contactsmodel.clear()
658 def clear_caches(self):
659 for factory in self._addressBookFactories:
660 factory.clear_caches()
661 self._addressBook.clear_caches()
663 def append(self, book):
664 self._addressBookFactories.append(book)
666 def extend(self, books):
667 self._addressBookFactories.extend(books)
669 def _idly_populate_contactsview(self):
670 #@todo Add a lock so only one code path can be in here at a time
673 # completely disable updating the treeview while we populate the data
674 self._contactsview.freeze_child_notify()
675 self._contactsview.set_model(None)
677 addressBook = self._addressBook
678 for contactId, contactName in addressBook.get_contacts():
679 contactType = (addressBook.contact_source_short_name(contactId),)
680 self._contactsmodel.append(contactType + (contactName, "", contactId) + ("",))
682 # restart the treeview data rendering
683 self._contactsview.set_model(self._contactsmodel)
684 self._contactsview.thaw_child_notify()
687 def _on_addressbook_combo_changed(self, *args, **kwds):
688 itr = self._booksSelectionBox.get_active_iter()
691 factoryId = int(self._booksList.get_value(itr, 0))
692 bookId = self._booksList.get_value(itr, 1)
693 self.open_addressbook(factoryId, bookId)
695 def _on_contactsview_row_activated(self, treeview, path, view_column):
696 model, itr = self._contactsviewselection.get_selected()
700 contactId = self._contactsmodel.get_value(itr, 3)
701 contactDetails = self._addressBook.get_contact_details(contactId)
702 contactDetails = [phoneNumber for phoneNumber in contactDetails]
704 if len(contactDetails) == 0:
706 elif len(contactDetails) == 1:
707 phoneNumber = contactDetails[0][1]
709 phoneNumber = self._phoneTypeSelector.run(contactDetails)
711 if 0 < len(phoneNumber):
712 self.number_selected(phoneNumber)
714 self._contactsviewselection.unselect_all()