在初始化Token时设置附加属性

Author: smallcode

chinaapi/open.py

@@ -25,12 +25,13 @@ def __init__(self, access_token=None, expires_in=None, refresh_token=None, **kwa
         """
         self.access_token = access_token
         self.expired_at = None
-        self.refresh_token = refresh_token
         self.expires_in = expires_in
-        self._data = kwargs
+        self.refresh_token = refresh_token
+        for key, value in kwargs.items():
+            setattr(self, key, value)
 
-    @classmethod
-    def _now(cls):
+    @staticmethod
+    def _now():
         return int(time.time())
 
     def _get_expires_in(self):
@@ -47,11 +48,6 @@ def _set_expires_in(self, expires_in):
     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):
-        if item in self._data:
-            return self._data[item]
-        raise AttributeError
-
 
 class App(object):
     def __init__(self, key='', secret='', redirect_uri=''):