Add new APIs for adding team members

1 files changed, 36 insertions, 0 deletions

diff --git a/arthur/apis/github/teams.py b/arthur/apis/github/teams.py
index a8775a3..489da90 100644
--- a/arthur/apis/github/teams.py
+++ b/arthur/apis/github/teams.py
@@ -1,3 +1,5 @@
+import aiohttp
+
 from arthur.config import CONFIG
 
 HEADERS = {
@@ -5,3 +7,37 @@ HEADERS = {
     "X-GitHub-Api-Version": "2022-11-28",
     "Authorization": f"Bearer {CONFIG.github_token.get_secret_value()}",
 }
+
+HTTP_404 = 404
+HTTP_403 = 403
+HTTP_422 = 422
+
+
+class GitHubError(Exception):
+    """Custom exception for GitHub API errors."""
+
+    def __init__(self, message: str):
+        super().__init__(message)
+
+
+async def add_staff_member(username: str) -> None:
+    """Add a user to the default GitHub team."""
+    async with aiohttp.ClientSession() as session:
+        endpoint = f"https://api.github.com/orgs/{CONFIG.github_org}/teams/{CONFIG.github_team}/memberships/{username}"
+        async with session.put(endpoint, headers=HEADERS) as response:
+            try:
+                response.raise_for_status()
+                return await response.json()
+            except aiohttp.ClientResponseError as e:
+                if e.status == HTTP_404:
+                    msg = f"Team or user not found: {e.message}"
+                    raise GitHubError(msg)
+                if e.status == HTTP_403:
+                    msg = f"Forbidden: {e.message}"
+                    raise GitHubError(msg)
+                if e.status == HTTP_422:
+                    msg = "Cannot add organisation as a team member"
+                    raise GitHubError(msg)
+
+                msg = f"Unexpected error: {e.message}"
+                raise GitHubError(msg)