Small changes in connection

[diaspy.git] / diaspy / connection.py

diff --git a/diaspy/connection.py b/diaspy/connection.py
index 3c1a0be1a489cc603c366c86f8f0e0e78f568ecd..8654b068ede0c4c2acc4e7884fbe8ec3c404592e 100644 (file)
--- a/diaspy/connection.py
+++ b/diaspy/connection.py
@@ -19,8 +19,7 @@ class TokenError(Exception):
 
 
 class Connection():
-    """Object representing connection with the server.
-    It is pushed around internally and is considered private.
+    """Object representing connection with the pod.
     """
     _token_regex = re.compile(r'content="(.*?)"\s+name="csrf-token')
     _userinfo_regex = re.compile(r'window.current_user_attributes = ({.*})')
@@ -157,7 +156,10 @@ class Connection():
         :returns: dict -- json formatted user info.
         """
         request = self.get('bookmarklet')
-        userdata = json.loads(self._userinfo_regex.search(request.text).group(1))
+        try:
+            userdata = json.loads(self._userinfo_regex.search(request.text).group(1))
+        except AttributeError:
+            raise errors.DiaspyError('cannot find user data')
         return userdata
 
     def _fetchtoken(self):