Refactor and introduce GitHubAppInstallation class

swinton · swinton · commit becc715a0b86 · 2018-02-25T19:52:16.000-06:00
diff --git a/auth.py b/auth.py
@@ -9,6 +9,10 @@
 import requests
 
 
+class NoInstallationException(Exception):
+    pass
+
+
 class JWTAuth(requests.auth.AuthBase):
     def __init__(self, iss, key, expiration=10 * 60):
         self.iss = iss
@@ -35,25 +39,15 @@ def __call__(self, r):
         return r
 
 
-class GitHubApp():
+class GitHubRequest():
     session = None
-    private_key = None
 
     def __init__(self):
         self.domain = os.environ.get('GITHUB_API_DOMAIN', 'api.github.com')
         self.session = requests.Session()
-        self.session.auth = JWTAuth(
-            iss=os.environ['APP_ID'],
-            key=self.read_private_key())
         self.session.headers.update(dict(
             accept='application/vnd.github.machine-man-preview+json'))
 
-    def read_private_key(self):
-        if self.private_key is None:
-            with open(os.environ['PRIVATE_KEY_FILE']) as fp:
-                self.private_key = fp.read()
-        return self.private_key
-
     def _request(self, method, path):
         self.response = self.session.request(method, 'https://{}/{}'.format(self.domain, path))
         return self.response.json()
@@ -64,22 +58,77 @@ def _get(self, path):
     def _post(self, path):
         return self._request('POST', path)
 
+
+class GitHubApp(GitHubRequest):
+    private_key = None
+
+    def __init__(self):
+        super().__init__()
+        self.session.auth = JWTAuth(
+            iss=os.environ['APP_ID'],
+            key=self.read_private_key())
+
+    def read_private_key(self):
+        if self.private_key is None:
+            with open(os.environ['PRIVATE_KEY_FILE']) as fp:
+                self.private_key = fp.read()
+        return self.private_key
+
     def get_app(self):
         return self._get('app')
 
     def get_installations(self):
         return self._get('app/installations')
 
+    def get_installation(self, login):
+        installation = [installation for
+            installation in self.get_installations()
+                if installation['account']['login'] == login]
+
+        if len(installation) > 0:
+            return GitHubAppInstallation(self,
+                installation[0],
+                self.get_installation_access_token(installation[0]['id']))
+
+        else:
+            raise NoInstallationException(
+                'No installation found for "{}".'.format(login))
+
     def get_installation_access_token(self, installation_id):
         return self._post('installations/{}/access_tokens'
                 .format(installation_id))
 
+
+class TokenAuth(requests.auth.AuthBase):
+    def __init__(self, app, access_token):
+        self._app = app
+        self._access_token = access_token
+
+    def __call__(self, r):
+        r.headers['Authorization'] = 'token {}'.format(self._access_token['token'])
+        return r
+
+
+class GitHubAppInstallation(GitHubRequest):
+    def __init__(self, app, installation_dict, access_token):
+        super().__init__()
+
+        self._app = app
+        self._dict = installation_dict
+        self._access_token = access_token
+
+        for k in installation_dict:
+            setattr(self, k, installation_dict[k])
+
+        self.session.auth = TokenAuth(app, access_token)
+
+    def get_repositories(self):
+        return self._get('installation/repositories')
+
+
 if __name__ == '__main__':
     import pprint
 
     app = GitHubApp()
-    pprint.pprint(app.get_app())
-
-    installations = app.get_installations()
-    pprint.pprint(installations)
-    pprint.pprint([(installation['id'], app.get_installation_access_token(installation['id'])) for installation in installations])
+    installation = app.get_installation('swinton')
+    pprint.pprint(installation.get_repositories())
