vcs.maemo.org Git - hermes/blob - package/src/org/maemo/hermes/engine/hermes.py

   1 from org.maemo.hermes.engine.friend import Friend
   2 from org.maemo.hermes.engine.contact import Contact
   3 import evolution
   4
   5 class Hermes:
   6     """Encapsulate the process of syncing online friends' information with the
   7        Evolution contacts' database.
   8
   9        Copyright (c) Andrew Flegg <andrew@bleb.org> 2010.
  10        Released under the Artistic Licence."""
  11
  12
  13     # -----------------------------------------------------------------------
  14     def __init__(self, services, gui_progress=None):
  15         """Constructor. Passed a list of services, and a
  16            method which will be invoked with three arguments:
  17                 str    Name of current step
  18                 int    Current position
  19                 int    Maximum value of position."""
  20
  21
  22         # -- These fields are currently part of the API...
  23         #
  24         self.updated = []
  25         self.matched = []
  26         self.unmatched = []
  27         self.friends = {}
  28         self.address_book = None
  29
  30         # -- Other initialisation...
  31         #
  32         self._services = services
  33         self._progress = gui_progress or (lambda msg, i, j: None)
  34
  35
  36     # -----------------------------------------------------------------------
  37     def run(self, overwrite_existing_fields=False):
  38         self._progress("Reading contacts...", 1, 10000)
  39
  40         contacts = []
  41         self.address_book = self._get_address_book()
  42         for econtact in self.address_book.get_all_contacts():
  43             contacts.append(self._create_contact_wrapper(econtact))
  44
  45         # work out progress bar info
  46         total_contacts = len(contacts) * len(self._services)
  47         total_ticks    = 6 * total_contacts # Number of distinct loops below
  48
  49         # warm up
  50         current_tick = 1
  51         for service in self._services:
  52             print "pre-process:", service.get_id()
  53             for contact in contacts:
  54                 self._progress("Pre-processing contacts...", current_tick, total_ticks)
  55                 current_tick += 1
  56                 service.pre_process_contact(contact)
  57
  58         # fetch data
  59         for service in self._services:
  60             print "process_friends:", service.get_id()
  61             self._progress("Reading friends...", current_tick, total_ticks)
  62             current_tick += len(contacts)
  63             service.process_friends()
  64
  65         # combine results into one friend
  66         for contact in contacts:
  67             result = Friend()
  68             for service in self._services:
  69                 print "process_contact:", service.get_id()
  70                 self._progress("Processing contacts...", current_tick, total_ticks)
  71                 current_tick += 1
  72                 friend = service.process_contact(contact)
  73                 if friend:
  74                     contact.add_mapping(service.get_id())
  75                     result.update_from_friend(friend)
  76
  77             if result.get_name() is not None:
  78                 self.update_contact(contact, result, overwrite_existing_fields)
  79             else:
  80                 self.unmatched.append(contact)
  81
  82         # give services a chance to create new contacts
  83         for service in self._services:
  84             print "create_contacts:", service.get_id()
  85             to_create = service.get_friends_to_create_contacts_for()
  86             tick_increment = len(contacts) / (len(to_create) or 1)
  87             print tick_increment, to_create
  88             for friend in to_create:
  89                 friend.set_source(service.get_id())
  90                 self._progress("Creating contacts...", current_tick, total_ticks)
  91                 current_tick += tick_increment
  92                 self.create_contact_from_friend(friend)
  93
  94         # finalisation
  95         for service in self._services:
  96             print "finalize:", service.get_id()
  97             self._progress("Finalising...", current_tick, total_ticks)
  98             current_tick += len(contacts)
  99             service.finalise(self.updated, overwrite_existing_fields)
 100             for friend in service.get_unmatched_friends():
 101                 friend.set_source(service.get_id())
 102                 key = unicode(friend.get_name()).encode('trans') + "_" + service.get_id()
 103                 self.friends[key] = friend
 104
 105         # commit changes
 106         tick_increment = total_contacts / (len(self.updated) or 1)
 107         print tick_increment, self.updated
 108         for contact in self.updated:
 109             print "committing changes to:", contact.get_name(), contact
 110             self._progress("Saving changes...", current_tick, total_ticks)
 111             current_tick += tick_increment
 112             self.address_book.commit_contact(contact.get_econtact())
 113
 114         self._progress('Finished', 1, -1)
 115
 116
 117     # -----------------------------------------------------------------------
 118     def update_contact(self, contact, friend, resync=False, commit=False):
 119         """Update the given contact with information from the given friend."""
 120
 121         print "updating contact ", contact, " with friend ", friend
 122         if friend.update_contact(contact, resync):
 123             self.updated.append(contact)
 124             if commit:
 125                 self.address_book.commit_contact(contact.get_econtact())
 126
 127         self.matched.append(contact)
 128         if friend.get_source() is not None:
 129             contact.add_mapping(friend.get_source())
 130
 131
 132     # -----------------------------------------------------------------------
 133     def create_contact_from_friend(self, friend):
 134         econtact = self._create_empty_contact(friend)
 135         contact = self._create_contact_wrapper(econtact)
 136
 137         self.address_book.add_contact(contact.get_econtact())
 138         self.update_contact(contact, friend)
 139
 140         print "Created [%s]" % (contact.get_name())
 141
 142
 143     # -----------------------------------------------------------------------
 144     def _get_address_book(self):
 145         return evolution.ebook.open_addressbook('default')
 146
 147     # -----------------------------------------------------------------------
 148     def _create_empty_contact(self, friend):
 149         econtact = evolution.ebook.EContact()
 150         econtact.props.given_name = friend['first_name']
 151         econtact.props.family_name = friend['last_name']
 152         econtact.props.full_name = friend.get_name()
 153         return econtact
 154
 155     # -----------------------------------------------------------------------
 156     def _create_contact_wrapper(self, econtact):
 157         return Contact(self.address_book, econtact)