Attempt transition to GitHub.

monitoring/setup.py

@@ -1,9 +1,10 @@
+import io
 import os
 import tempfile
-import urllib.request
 import zipfile
 from pathlib import Path
 
+import requests
 from google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
@@ -41,16 +42,27 @@ def get_tmpdir():
     return Path(tempfile.gettempdir())
 
 
-def create_and_get_week_dir() -> Path:
-    dropbox_url = os.environ["dropbox_url"]
+def create_and_get_week_dir(week: str = "23-weeks") -> Path:
+    username = os.environ["github_username"]
+    repo = os.environ["github_repo"]
+    ref = os.environ["github_ref"]
+    pat = os.environ["github_pat"]
 
-    tmpdir = get_tmpdir()
-    zip_path = tmpdir / "org.zip"
-    urllib.request.urlretrieve(dropbox_url, zip_path)
+    url = f"https://api.github.com/repos/{username}/{repo}/zipball/{ref}"
 
-    weeks_dir = tmpdir / "23-weeks"
+    headers = {
+        "Accept": "application/vnd.github+json",
+        "Authorization": f"Bearer {pat}",
+        "X-GitHub-Api-Version": "2022-11-28",
+    }
 
-    with zipfile.ZipFile(zip_path, "r") as zip_ref:
-        zip_ref.extractall(weeks_dir)
+    response = requests.get(url, headers=headers)
 
+    tmpdir = get_tmpdir() / "org"
+    tmpdir.mkdir(exist_ok=True)
+
+    z = zipfile.ZipFile(io.BytesIO(response.content))
+    z.extractall(tmpdir)
+
+    weeks_dir = tmpdir / z.filelist[0].filename / week
     return weeks_dir

requirements.txt

@@ -6,3 +6,4 @@ matplotlib
 google.auth
 google-api-python-client
 pydantic==1.10.12
+requests