git.maemo.org Git - theonering/blob - src/aliasing.py

   1 import logging
   2
   3 import telepathy
   4
   5 import gtk_toolbox
   6 import handle
   7
   8
   9 USER_ALIAS_ACCOUNT = "account"
  10 USER_ALIAS_CALLBACK = "callback"
  11
  12 USER_ALIAS = USER_ALIAS_ACCOUNT
  13
  14
  15 _moduleLogger = logging.getLogger('aliasing')
  16
  17
  18 def make_pretty(phonenumber):
  19         """
  20         Function to take a phone number and return the pretty version
  21         pretty numbers:
  22                 if phonenumber begins with 0:
  23                         ...-(...)-...-....
  24                 if phonenumber begins with 1: ( for gizmo callback numbers )
  25                         1 (...)-...-....
  26                 if phonenumber is 13 digits:
  27                         (...)-...-....
  28                 if phonenumber is 10 digits:
  29                         ...-....
  30         >>> make_pretty("12")
  31         '12'
  32         >>> make_pretty("1234567")
  33         '123-4567'
  34         >>> make_pretty("2345678901")
  35         '(234)-567-8901'
  36         >>> make_pretty("12345678901")
  37         '1 (234)-567-8901'
  38         >>> make_pretty("01234567890")
  39         '+012-(345)-678-90'
  40         """
  41         if phonenumber is None or phonenumber is "":
  42                 return ""
  43
  44         phonenumber = handle.strip_number(phonenumber)
  45
  46         if len(phonenumber) < 3:
  47                 return phonenumber
  48
  49         if phonenumber[0] == "0":
  50                 prettynumber = ""
  51                 prettynumber += "+%s" % phonenumber[0:3]
  52                 if 3 < len(phonenumber):
  53                         prettynumber += "-(%s)" % phonenumber[3:6]
  54                         if 6 < len(phonenumber):
  55                                 prettynumber += "-%s" % phonenumber[6:9]
  56                                 if 9 < len(phonenumber):
  57                                         prettynumber += "-%s" % phonenumber[9:]
  58                 return prettynumber
  59         elif len(phonenumber) <= 7:
  60                 prettynumber = "%s-%s" % (phonenumber[0:3], phonenumber[3:])
  61         elif len(phonenumber) > 8 and phonenumber[0] == "1":
  62                 prettynumber = "1 (%s)-%s-%s" % (phonenumber[1:4], phonenumber[4:7], phonenumber[7:])
  63         elif len(phonenumber) > 7:
  64                 prettynumber = "(%s)-%s-%s" % (phonenumber[0:3], phonenumber[3:6], phonenumber[6:])
  65         return prettynumber
  66
  67
  68 class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
  69
  70         def __init__(self):
  71                 telepathy.server.ConnectionInterfaceAliasing.__init__(self)
  72
  73         @property
  74         def session(self):
  75                 """
  76                 @abstract
  77                 """
  78                 raise NotImplementedError("Abstract property called")
  79
  80         def handle(self, handleType, handleId):
  81                 """
  82                 @abstract
  83                 """
  84                 raise NotImplementedError("Abstract function called")
  85
  86         @gtk_toolbox.log_exception(_moduleLogger)
  87         def GetAliasFlags(self):
  88                 return 0
  89
  90         @gtk_toolbox.log_exception(_moduleLogger)
  91         def RequestAliases(self, contactHandleIds):
  92                 _moduleLogger.debug("Called RequestAliases")
  93                 return [self._get_alias(handleId) for handleId in contactHandleIds]
  94
  95         @gtk_toolbox.log_exception(_moduleLogger)
  96         def GetAliases(self, contactHandleIds):
  97                 _moduleLogger.debug("Called GetAliases")
  98
  99                 idToAlias = dict(
 100                         (handleId, self._get_alias(handleId))
 101                         for handleId in contactHandleIds
 102                 )
 103                 return idToAlias
 104
 105         @gtk_toolbox.log_exception(_moduleLogger)
 106         def SetAliases(self, aliases):
 107                 _moduleLogger.debug("Called SetAliases")
 108                 if USER_ALIAS == USER_ALIAS_ACCOUNT:
 109                         raise telepathy.errors.PermissionDenied("No user customizable aliases")
 110                 elif USER_ALIAS != USER_ALIAS_CALLBACK:
 111                         raise RuntimeError("Invalid alias type: %r" % USER_ALIAS)
 112
 113                 # first validate that no other handle types are included
 114                 userHandleAndAlias = None
 115                 for handleId, alias in aliases.iteritems():
 116                         h = self.handle(telepathy.HANDLE_TYPE_CONTACT, handleId)
 117                         if not isinstance(h, handle.ConnectionHandle):
 118                                 raise telepathy.errors.PermissionDenied("No user customizable aliases")
 119                         userHandleAndAlias = h, alias
 120                 if userHandleAndAlias is None:
 121                         _moduleLogger.debug("No user handle")
 122                         return
 123
 124                 # Update callback
 125                 uglyNumber = handle.strip_number(userHandleAndAlias[1])
 126                 self.session.backend.set_callback_number(uglyNumber)
 127
 128                 # Inform of change
 129                 changedAliases = (userHandleAndAlias, )
 130                 self.AliasesChanged(changedAliases)
 131
 132         def _get_alias(self, handleId):
 133                 h = self.handle(telepathy.HANDLE_TYPE_CONTACT, handleId)
 134                 if isinstance(h, handle.ConnectionHandle):
 135                         if USER_ALIAS == USER_ALIAS_CALLBACK:
 136                                 aliasNumber = self.session.backend.get_callback_number()
 137                         elif USER_ALIAS == USER_ALIAS_ACCOUNT:
 138                                 aliasNumber = self.session.backend.get_account_number()
 139                         else:
 140                                 raise RuntimeError("Invalid alias type: %r" % USER_ALIAS)
 141                         userAlias = make_pretty(aliasNumber)
 142                         return userAlias
 143                 else:
 144                         contactId = h.contactID
 145                         if contactId:
 146                                 contactAlias = self.session.addressbook.get_contact_name(contactId)
 147                         else:
 148                                 contactAlias = make_pretty(h.phoneNumber)
 149                         return contactAlias