initial commit

main.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python3
+from links import urls
+from typing import List, Dict
+import json
+import asyncio
+import aiohttp
+import time
+from tqdm import tqdm
+
+from parse import parse
+from key import token
+
+MAX_RETRY: int = 5
+
+async def getInsights(repoURL: str, attempt: int = 0) -> Dict[str, str|None] | None:
+  owner: str = repoURL.split("/")[3]
+  repo: str = repoURL.split("/")[4]
+
+  # print(f"Querying {repo}")
+
+  query: str = f"https://api.github.com/repos/{owner}/{repo}/stats/contributors"
+  headers: Dict[str, str] = {
+    "Accept": "application/vnd.github+json",
+    "Authorization": f"Bearer {token}",
+    "X-GitHub-Api-Version": "2022-11-28" 
+  }
+
+  async with aiohttp.ClientSession() as session:
+    async with session.get(query, headers=headers, timeout=5) as resp:
+      # print(f"   {urls.index(repoURL) + 1}/{len(urls)} - {resp.status}")
+      if resp.status > 400:
+        # print(f"An Error Occured Querying {repo}")
+        return None
+      if resp.status == 202 and attempt <= MAX_RETRY:
+        # print(f"{repo} Was queried, but no data was received")
+        time.sleep(10)
+        attempt += 1
+        team = await getInsights(repoURL, attempt)
+        return team
+      elif attempt > MAX_RETRY: return None
+
+      team: Dict[str, str|List] = {
+        "repo": repoURL,
+        "team": repo,
+        "contributors": []
+      }
+      
+      data: List[Dict[str, any]] = await resp.json()
+
+      for contributor in data:
+        newContributor: Dict[str, str|int] = {
+          "author": contributor["author"]["login"],
+          "commits": contributor["total"],
+          "added": 0,
+          "deleted": 0
+        }
+
+        for week in contributor["weeks"]:
+          newContributor["added"] += week["a"]
+          newContributor["deleted"] += week["d"]
+
+        team["contributors"].append(newContributor)
+
+  return team
+
+async def main() -> None:
+  teamsOriginal: List[Dict] = []
+  with open("teams.json", "r") as file:
+    teamsOriginal = json.load(file)
+
+  teams: List[Dict] = []
+
+  for i in tqdm(range(len(urls)),
+                desc="Reading Insights",
+                ascii=False, ncols=75):
+    url = urls[i]
+    team: Dict[str, str|List] = await getInsights(url)
+    if team is not None:
+      teams.append(team)
+    time.sleep(5)
+
+  print(f"Completed {len(teams)}/{len(urls)}")
+
+  # Parse gets Campus, Set, and Team ID
+  teams = parse(teams)
+
+  for originalTeam in teamsOriginal:
+    originalTeamID = originalTeam["id"]
+    exists: bool = False
+    for team in teams:
+      if team["id"] == originalTeamID:
+        exists = True
+        break
+
+    if not exists:
+      teams.append(originalTeam)
+
+  with open("teams.json", "w") as file:
+    json.dump(teams, file, indent=2)
+
+if __name__ == "__main__":
+  asyncio.run(main())