X-Git-Url: http://git.maemo.org/git/?p=python-purple;a=blobdiff_plain;f=account.pyx;h=0588ea31be11c12651e2d7a35872e4a3e895b347;hp=d896dd05f558a1bdb5e4a1bd419637ca2fa670d5;hb=60e2bde48e3f6f4f451f1fae7fc0122b6d1e9f72;hpb=fa83e5e05e62fa5c6707312d375b97195143e6b7

diff --git a/account.pyx b/account.pyx
index d896dd0..0588ea3 100644
--- a/account.pyx
+++ b/account.pyx
@@ -22,24 +22,29 @@ cimport purple
 cdef class Account:
     """
     Account class
-    @param core
     @param username
-    @param protocol_id
+    @param protocol Protocol class instance
+    @param core Purple class instance
     """
 
-    def __init__(self, core, username, protocol_id):
-        self.__core = core
+    cdef object __username
+    cdef object __protocol
+    cdef object __core
+    cdef object __exists
+
+    def __init__(self, username, protocol, core):
         self.__username = username
-        self.__protocol = Protocol(self, protocol_id)
+        self.__protocol = protocol
+        self.__core = core
 
-        if self._get_structure() == NULL:
-            self.__exists = False
-        else:
+        if protocol.exists and self._get_structure() != NULL:
             self.__exists = True
+        else:
+            self.__exists = False
 
     cdef account.PurpleAccount *_get_structure(self):
-        return account.purple_accounts_find(self.username, \
-                self.protocol_id)
+        return account.purple_accounts_find(self.__username, \
+                self.__protocol.id)
 
     def __is_connected(self):
         if self.__exists:
@@ -84,18 +89,9 @@ cdef class Account:
             return self.__username
     username = property(__get_username)
 
-    def __get_protocol_id(self):
-        cdef char *protocol_id = NULL
-        if self.__exists:
-            protocol_id = <char *> account.purple_account_get_protocol_id( \
-                    self._get_structure())
-            if protocol_id:
-                return protocol_id
-            else:
-                return None
-        else:
-            return self.protocol_id
-    protocol_id = property(__get_protocol_id)
+    def __get_protocol(self):
+        return self.__protocol
+    protocol = property(__get_protocol)
 
     def __get_password(self):
         cdef char *password = NULL
@@ -145,7 +141,7 @@ cdef class Account:
     def __get_enabled(self):
         if self.__exists:
             return account.purple_account_get_enabled(self._get_structure(), \
-                    self.core.ui_name)
+                    self.__core.ui_name)
         else:
             return None
     enabled = property(__get_enabled)
@@ -164,15 +160,17 @@ cdef class Account:
         else:
             return False
 
-    def set_protocol_id(self, protocol_id):
+    def set_protocol(self, protocol):
         """
-        Sets the account's protocol ID.
+        Sets the account's protocol.
 
-        @param protocol_id The protocol ID
+        @param protocol A Protocol class instance
         @return True if successful, False if account doesn't exists
         """
-        if self.__exists:
-            self.__protocol._set_protocol_id(protocol_id)
+        if protocol.exists and self.__exists:
+            account.purple_account_set_protocol_id(self._get_structure(), \
+                        protocol.id)
+            self.__protocol = protocol
             return True
         else:
             return False
@@ -243,7 +241,7 @@ cdef class Account:
         """
         if self.__exists:
             account.purple_account_set_enabled(self._get_structure(), \
-                    self.core.ui_name, bool(value))
+                    self.__core.ui_name, bool(value))
             return True
         else:
             return False
@@ -257,7 +255,7 @@ cdef class Account:
         if self.__exists:
             return False
         else:
-            account.purple_account_new(self.username, self.protocol_id)
+            account.purple_account_new(self.__username, self.__protocol.id)
             self.__exists = True
             return True