adding organization model and relevant helpers

garbas · garbas · commit 27fec35b1fa1 · 2011-08-24T05:19:28.000+02:00
diff --git a/github3/api.py b/github3/api.py
@@ -156,6 +156,9 @@ def get_repo(self, username, reponame):
         """Get the authenticated user."""
         return self._get_resource(('repos', username, reponame), Repo)
 
+    def get_org(self, login):
+        """Get organization."""
+        return self._get_resource(('orgs', login), Org)
 
 
 class ResponseError(Exception):
diff --git a/github3/models.py b/github3/models.py
@@ -91,7 +91,10 @@ def __repr__(self):
         return '<user {0}>'.format(self.login)
 
     def repos(self, limit=None):
-         return self._gh._get_resources(('users', self.login, 'repos'), Repo)
+        return self._gh._get_resources(('users', self.login, 'repos'), Repo)
+
+    def orgs(self):
+        return self._gh._get_resources(('users', self.login, 'orgs'), Org)
 
 
 class CurrentUser(User):
@@ -110,6 +113,30 @@ def __repr__(self):
     def repos(self, limit=None):
          return self._gh._get_resources(('user', 'repos'), Repo)
 
+    def orgs(self):
+        return self._gh._get_resources(('user', 'orgs'), Org)
+
+
+class Org(BaseResource):
+    """Github Organization object model."""
+
+    _strs = [
+        'login', 'url', 'avatar_url', 'name', 'company', 'blog', 'location', 'email'
+        'html_url', 'type']
+    _ints = [
+        'id', 'public_repos', 'public_gists', 'followers', 'following',
+        'total_private_repos', 'owned_private_repos', 'private_gists', 'disk_usage',
+        'collaborators']
+    _dates = ['created_at']
+    _map = {'plan': Plan}
+    _writable = ['billing_email', 'blog', 'company', 'email', 'location', 'name']
+
+    def __repr__(self):
+        return '<org {0}>'.format(self.login)
+
+    def repos(self, limit=None):
+         return self._gh._get_resources(('orgs', self.login, 'repos'), Repo)
+
 
 class Repo(BaseResource):
     _strs = [
