Friend now has _source field again :)

[hermes] / package / src / org / maemo / hermes / engine / friend.py

diff --git a/package/src/org/maemo/hermes/engine/friend.py b/package/src/org/maemo/hermes/engine/friend.py
index 736fae2..4278541 100644 (file)
--- a/package/src/org/maemo/hermes/engine/friend.py
+++ b/package/src/org/maemo/hermes/engine/friend.py
@@ -1,25 +1,59 @@
 class Friend():
+    """Encapsulate the data from a remote service.
     
-    def __init__(self, name, source=None):
+       Copyright (c) Fredrik Wendt <fredrik@wendt.se> 2010.
+       Released under the Artistic Licence."""
+
+    
+    def __init__(self, name=None, source=None, props=None):
         """ source is source service, such as LinkedIn """
-        self._attributes = { "fn": name }
+        self._attributes = {};
+        if name: self._set('fn', name)
         self._multi_attributes = {}
-        self._source = source 
+        self._source = source
+        if props:
+            for key in props:
+                self._set(key, props[key])
         
+    def __getitem__(self, key):
+        return self._safe_get(key)
+                              
     def __unicode__(self):
         return self.__repr__()
     
     def __repr__(self):
-        return "Friend %s" % self._attributes['fn']
+        return "Friend %s" % self.get_name()
     
     # public accessors -----------------
     
-    def get_name(self):
-        return self._attributes['fn']
-    
     def add_url(self, url):
         self._add('url', url)
         
+    def get_birthday_date(self):
+        return self._safe_get('bday')
+    
+    def get_contact(self):
+        return self._safe_get('contact')
+    
+    def get_name(self):
+        return self._safe_get('fn')
+    
+    def get_source(self):
+        return self._source
+    
+    def get_nickname(self):
+        return self._safe_get("nickname")
+    
+    def get_urls(self):
+        try: return self._multi_attributes['url'] 
+        except: return []
+        
+    def get_photo_url(self):
+        return self._safe_get('photo-url')
+    
+    def has_birthday_date(self):
+        return self._has('bday')
+    
     def is_empty(self):
         for a in self._attributes:
             return False
@@ -27,9 +61,6 @@ class Friend():
             return False
         return True
     
-    def has_birthday_date(self):
-        return self._has('bday')
-    
     def set_name(self, name):
         self._set('fn', name)
     
@@ -39,10 +70,13 @@ class Friend():
     def set_birthday_date(self, date):
         self._set('bday', date)
         
+    def set_contact(self, contact):
+        self._set('contact', contact)
+        
     def set_photo_url(self, url):
         self._set('photo-url', url)
     
-    def update(self, other_friend):
+    def update_from_friend(self, other_friend, overwrite=False):
         """
         Overwrites any attributes in this friend, with attributes from other_friend
         """
@@ -73,15 +107,6 @@ class Friend():
 
     # private helpers -----------------------
     #
-    def _if_defined(self, key, callback):
-        if self._attributes.has_key(key):
-            callback(self._attributes[key])
-    
-    def _set(self, key, value):
-        if value is not None:
-#            print "%s SET %s to %s" % (self, key, value)
-            self._attributes[key] = value
-    
     def _add(self, key, value):
         if value is not None:
             if not self._multi_attributes.has_key(key):
@@ -89,12 +114,26 @@ class Friend():
 #            print "%s ADD %s to %s" % (self, key, value)
             self._multi_attributes[key].append(value)
     
-    def _has(self, key):
-        return self._attributes.has_key(key) or self._multi_attributes.has_key(key)
-    
     def _contains(self, key, value):
         if self._attributes.has_key(key):
             return value == self._attributes[key]
         if self._multi_attributes.has_key(key):
             return value in self._multi_attributes[key]
         return False
+    
+    def _if_defined(self, key, callback):
+        if self._attributes.has_key(key):
+            callback(self._attributes[key])
+    
+    def _has(self, key):
+        return self._attributes.has_key(key) or self._multi_attributes.has_key(key)
+    
+    def _safe_get(self, key):
+        try: return self._attributes[key]
+        except: return None
+        
+    def _set(self, key, value):
+        if value is not None:
+#            print "%s SET %s to %s" % (self, key, value)
+            self._attributes[key] = value
+