adding organization model and relevant helpers

1 files changed, 28 insertions, 1 deletions

diff --git a/github3/models.py b/github3/models.py
index d2709b6..1cdf239 100644
--- a/github3/models.py
+++ b/github3/models.py
@@ -91,7 +91,10 @@ class User(BaseResource):
         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 @@ class CurrentUser(User):
     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 = [