fix:__getattr__

Author: smallcode

chinaapi/open.py

@@ -48,7 +48,9 @@ def is_expires(self):
         return not self.access_token or (self.expired_at is not None and self._now() > self.expired_at)
 
     def __getattr__(self, item):
-        return self._data[item]
+        if item in self._data:
+            return self._data[item]
+        raise AttributeError
 
 
 class App(object):
@@ -136,7 +138,9 @@ def try_request():
         return try_request()
 
     def __getattr__(self, attr):
-        return ClientWrapper(self, attr)
+        if not attr.startswith('__'):
+            return ClientWrapper(self, attr)
+        raise AttributeError
 
 
 class OAuthBase(Request):

tests/test_open.py

@@ -187,5 +187,5 @@ def test_get_attr(self):
 
     def test_get_not_exist_attr(self):
         token = Token()
-        with self.assertRaises(KeyError):
+        with self.assertRaises(AttributeError):
             _ = token.not_exist_attr