cleanups

Kenneth Reitz · Kenneth Reitz · commit 5e0c27007ba8 · 2011-07-24T11:28:12.000-04:00
diff --git a/github3/models.py b/github3/models.py
@@ -17,6 +17,7 @@ class BaseResource(object):
     _bools = []
     _map = {}
     _writeable = []
+    _modified = []
 
 
     def __init__(self):
@@ -47,10 +48,11 @@ def new_from_dict(cls, d, gh=None):
 
         return to_python(
             obj=cls(), in_dict=d,
-            str_keys = cls._strings,
+            str_keys = cls._strs,
             int_keys = cls._ints,
-            date_keys = cls._datetimes,
-            bool_keys = cls._booleans,
+            date_keys = cls._dates,
+            bool_keys = cls._bools,
+            object_map = cls._map,
             _gh = gh
         )
 
@@ -61,6 +63,17 @@ def setattr(self, k, v):
         # TODO: when writable key changed,
         pass
 
+class Plan(BaseResource):
+    """Github Plan object model."""
+
+    _strs = ['name']
+    _ints = ['space', 'collaborators', 'private_repos']
+
+    def __repr__(self):
+        return '<plan {0}>'.format(str(self.name))
+
+
+
 class User(BaseResource):
     """Github User object model."""
 
@@ -69,31 +82,43 @@ class User(BaseResource):
         'email', 'bio', 'html_url']
 
     _ints = ['id', 'public_repos', 'public_gists', 'followers', 'following']
-    _datetimes = ['created_at',]
-    _booleans = ['hireable', ]
+    _dates = ['created_at',]
+    _bools = ['hireable', ]
     _map = {}
     _writeable = ['name', 'email', 'blog', 'company', 'location', 'hireable', 'bio']
 
+    def __repr__(self):
+        return '<user {0}>'.format(self.login)
 
-    def __init__(self):
-        super(User, self).__init__()
+
+class User(BaseResource):
+    """Github User object model."""
+
+    _strings = [
+        'login','gravatar_url', 'url', 'name', 'company', 'blog', 'location',
+        'email', 'bio', 'html_url']
+
+    _ints = ['id', 'public_repos', 'public_gists', 'followers', 'following', 'total_private_repos', 'owned_private_repos', 'private_gists', 'disk_usage', 'collaborators']
+    _dates = ['created_at',]
+    _bools = ['hireable', ]
+    _map = {'plan': Plan}
+    _writeable = ['name', 'email', 'blog', 'company', 'location', 'hireable', 'bio']
 
     def __repr__(self):
         return '<user {0}>'.format(self.login)
 
+    # def _update(self):
+    #     """Update the User."""
 
-    def _update(self):
-        """Update the User."""
-
-        args = to_api(
-            dict(
-                favorite=self.favorite,
-                archive=self.archive,
-                read_percent=self.read_percent,
-            ),
-            int_keys=('favorite', 'archive')
-        )
+    #     args = to_api(
+    #         dict(
+    #             favorite=self.favorite,
+    #             archive=self.archive,
+    #             read_percent=self.read_percent,
+    #         ),
+    #         int_keys=('favorite', 'archive')
+    #     )
 
-        r = self._rdd._post_resource(('bookmarks', self.id), **args)
+    #     r = self._rdd._post_resource(('bookmarks', self.id), **args)
 
-        return r
+    #     return r
