From 21e1f9986a7fc9e5cf548a3804f954bc94460fa8 Mon Sep 17 00:00:00 2001 From: Marek Marecki Date: Sun, 25 Aug 2013 01:16:04 +0200 Subject: [PATCH] Nicer error handling in Connection.getUserData() --- diaspy/connection.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/diaspy/connection.py b/diaspy/connection.py index 4c60783..001e215 100644 --- a/diaspy/connection.py +++ b/diaspy/connection.py @@ -154,10 +154,9 @@ class Connection(): :returns: dict -- json formatted user info. """ request = self.get('bookmarklet') - try: - userdata = json.loads(self._userinfo_regex.search(request.text).group(1)) - except AttributeError: - raise errors.DiaspyError('cannot find user data') + userdata = self._userinfo_regex.search(request.text) + if userdata is None: raise errors.DiaspyError('cannot find user data') + userdata = json.loads(userdata.group(1)) return userdata def _fetchtoken(self): -- 2.25.1