A small assert because I'm annoyed with leaking states

[theonering] / src / aliasing.py

diff --git a/src/aliasing.py b/src/aliasing.py
index bf73e06..c22f6bb 100644 (file)
--- a/src/aliasing.py
+++ b/src/aliasing.py
@@ -3,15 +3,10 @@ import logging
 import telepathy
 
 import gtk_toolbox
+import util.misc as util_misc
 import handle
 
 
-USER_ALIAS_ACCOUNT = "account"
-USER_ALIAS_CALLBACK = "callback"
-
-USER_ALIAS = USER_ALIAS_ACCOUNT
-
-
 _moduleLogger = logging.getLogger('aliasing')
 
 
@@ -41,7 +36,7 @@ def make_pretty(phonenumber):
        if phonenumber is None or phonenumber is "":
                return ""
 
-       phonenumber = handle.strip_number(phonenumber)
+       phonenumber = util_misc.strip_number(phonenumber)
 
        if len(phonenumber) < 3:
                return phonenumber
@@ -67,6 +62,9 @@ def make_pretty(phonenumber):
 
 class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
 
+       USER_ALIAS_ACCOUNT = "account"
+       USER_ALIAS_CALLBACK = "callback"
+
        def __init__(self):
                telepathy.server.ConnectionInterfaceAliasing.__init__(self)
 
@@ -77,6 +75,13 @@ class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
                """
                raise NotImplementedError("Abstract property called")
 
+       @property
+       def userAliasType(self):
+               """
+               @abstract
+               """
+               raise NotImplementedError("Abstract property called")
+
        def handle(self, handleType, handleId):
                """
                @abstract
@@ -105,10 +110,10 @@ class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
        @gtk_toolbox.log_exception(_moduleLogger)
        def SetAliases(self, aliases):
                _moduleLogger.debug("Called SetAliases")
-               if USER_ALIAS == USER_ALIAS_ACCOUNT:
+               if self.userAliasType == self.USER_ALIAS_ACCOUNT:
                        raise telepathy.errors.PermissionDenied("No user customizable aliases")
-               elif USER_ALIAS != USER_ALIAS_CALLBACK:
-                       raise RuntimeError("Invalid alias type: %r" % USER_ALIAS)
+               elif self.userAliasType != self.USER_ALIAS_CALLBACK:
+                       raise RuntimeError("Invalid alias type: %r" % self.userAliasType)
 
                # first validate that no other handle types are included
                userHandleAndAlias = None
@@ -122,7 +127,7 @@ class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
                        return
 
                # Update callback
-               uglyNumber = handle.strip_number(userHandleAndAlias[1])
+               uglyNumber = util_misc.strip_number(userHandleAndAlias[1])
                self.session.backend.set_callback_number(uglyNumber)
 
                # Inform of change
@@ -132,12 +137,12 @@ class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
        def _get_alias(self, handleId):
                h = self.handle(telepathy.HANDLE_TYPE_CONTACT, handleId)
                if isinstance(h, handle.ConnectionHandle):
-                       if USER_ALIAS == USER_ALIAS_CALLBACK:
+                       if self.userAliasType == self.USER_ALIAS_CALLBACK:
                                aliasNumber = self.session.backend.get_callback_number()
-                       elif USER_ALIAS == USER_ALIAS_ACCOUNT:
+                       elif self.userAliasType == self.USER_ALIAS_ACCOUNT:
                                aliasNumber = self.session.backend.get_account_number()
                        else:
-                               raise RuntimeError("Invalid alias type: %r" % USER_ALIAS)
+                               raise RuntimeError("Invalid alias type: %r" % self.userAliasType)
                        userAlias = make_pretty(aliasNumber)
                        return userAlias
                else: