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

   1 import urllib, hashlib, xmlrpclib
   2 import org.maemo.hermes.engine.service
   3
   4 class Service(org.maemo.hermes.engine.service.Service):
   5     """Gravatar backend for Hermes.
   6
   7        Copyright (c) Fredrik Wendt <fredrik@wendt.se> 2010.
   8        Released under the Artistic Licence."""
   9
  10     _image_url_base = "http://www.gravatar.com/avatar.php?"
  11
  12     # -----------------------------------------------------------------------
  13     def __init__(self, api_email, api_key):
  14         """Initialise the Gravatar service
  15
  16         api_email is the email address to use when talking to the backend.
  17         api_key is the "secrect" key used when talking to the backend
  18         """
  19         self._api_email = api_email
  20         self._api_key = api_key
  21         if self._api_key is None or self._api_email is None:
  22             raise Exception('No Gravatar application keys found. Installation error.')
  23
  24         self._address_to_contact = {}
  25         self._hash_to_address = {}
  26         self._hash_has_gravatar = {}
  27         self._empty_cache = True
  28
  29         self._friends_by_contact = {}
  30         self._contacts_by_friend = {}
  31
  32         self._api_url = 'https://secure.gravatar.com/xmlrpc?user=' + hashlib.md5(self._api_email).hexdigest()
  33
  34
  35     # -----------------------------------------------------------------------
  36     def get_name(self):
  37         return "Gravatar"
  38
  39
  40     # -----------------------------------------------------------------------
  41     def pre_process_contact(self, contact):
  42         """Extracts addresses from the contact."""
  43         for address in contact.get_emails():
  44             self._address_to_contact[address] = contact
  45
  46
  47     # -----------------------------------------------------------------------
  48     def process_contact(self, contact):
  49         """On first call (with a contact missing a photo), go get data from Gravatar's servers."""
  50
  51         if not self._has_photo(contact):
  52             for address in contact.get_emails():
  53                 hash = self._get_hash_for_address(address)
  54                 if (self._hash_has_gravatar.has_key(hash) and self._hash_has_gravatar[hash]):
  55                     friend = self._create_friend(contact.get_name())
  56                     friend.set_photo_url(self._get_url_for_email_hash(hash))
  57                     self._register_match(contact, friend)
  58
  59
  60     # -----------------------------------------------------------------------
  61     def process_friends(self):
  62         self._lookup_addresses()
  63
  64
  65     # -----------------------------------------------------------------------
  66     def get_friends(self):
  67         return self._contacts_by_friend.keys()
  68
  69
  70     def get_contacts_with_match(self):
  71         """Returns a dict with Contact objects as keys and Friend objects as values"""
  72         return self._friends_by_contact
  73
  74
  75     def get_unmatched_friends(self):
  76         """Will always return None - Gravatar only reacts on e-mail address input."""
  77         return None
  78
  79
  80
  81
  82     # -----------------------------------------------------------------------
  83     def _register_match(self, contact, friend):
  84         self._friends_by_contact[contact] = friend
  85         self._contacts_by_friend[friend] = contact
  86
  87     # -----------------------------------------------------------------------
  88     # FIXME
  89     def _has_photo(self, contact):
  90         return False
  91
  92
  93     # -----------------------------------------------------------------------
  94
  95     def _lookup_addresses(self):
  96         """Constructs hashes for address_to_contact, makes call to the Gravatar.com service and updates
  97         self._hash_has_gravatar"""
  98
  99         addresses = self._address_to_contact.keys()
 100         if len(addresses) == 0:
 101             self._set_hash_information({})
 102         else:
 103             args = { "apikey" : self._api_key}
 104             hashes = self._construct_hashes(addresses)
 105             args["hashes"] = list(hashes)
 106             url = self._api_url
 107             self._set_hash_information(self._get_hash_info_from_server(args, url))
 108
 109
 110     # -----------------------------------------------------------------------
 111     def _get_hash_info_from_server(self, args, url):
 112         """Makes the actual XML-RPC call - override this for testing"""
 113
 114         service = xmlrpclib.ServerProxy(url)
 115         return service.grav.exists(args)
 116
 117
 118     # -----------------------------------------------------------------------
 119     def _set_hash_information(self, hash_info):
 120         self._hash_has_gravatar = hash_info
 121         self._empty_cache = False
 122
 123
 124     # -----------------------------------------------------------------------
 125     def _get_url_for_email_hash(self, hash, default="404", size="128"):
 126         """Assembles the URL to a gravatar, based on a hash of an e-mail address"""
 127
 128         return self._image_url_base + urllib.urlencode({'gravatar_id':hash, 'd':default, 'size':size})
 129
 130
 131     # -----------------------------------------------------------------------
 132     def _construct_hashes(self, addresses):
 133         """Creates hashes for all addresses specified, returning a set of hashes"""
 134
 135         result = set()
 136         for address in addresses:
 137             hash = self._get_hash_for_address(address)
 138             self._hash_to_address[hash] = address
 139             result.add(hash)
 140
 141         return result
 142
 143     # -----------------------------------------------------------------------
 144     def _get_hash_for_address(self, address):
 145         return hashlib.md5(address.lower()).hexdigest()