first commit based on psycopg2 2.9 version

scripts/build/download_packages_github.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python
+"""Download packages from github actions artifacts
+"""
+
+import io
+import os
+import sys
+import logging
+import datetime as dt
+from pathlib import Path
+from zipfile import ZipFile
+
+import requests
+
+logger = logging.getLogger()
+logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(message)s")
+
+REPOS = "psycopg/psycopg2"
+WORKFLOW_NAME = "Build packages"
+
+
+class ScriptError(Exception):
+    """Controlled exception raised by the script."""
+
+
+def main():
+    try:
+        token = os.environ["GITHUB_TOKEN"]
+    except KeyError:
+        raise ScriptError("please set a GITHUB_TOKEN to download artifacts")
+
+    s = requests.Session()
+    s.headers["Accept"] = "application/vnd.github.v3+json"
+    s.headers["Authorization"] = f"token {token}"
+
+    logger.info("looking for recent runs")
+    resp = s.get(f"https://api.github.com/repos/{REPOS}/actions/runs?per_page=10")
+    resp.raise_for_status()
+    for run in resp.json()["workflow_runs"]:
+        if run["name"] == WORKFLOW_NAME:
+            break
+    else:
+        raise ScriptError(f"couldn't find {WORKFLOW_NAME!r} in recent runs")
+
+    if run["status"] != "completed":
+        raise ScriptError(f"run #{run['run_number']} is in status {run['status']}")
+
+    updated_at = dt.datetime.fromisoformat(run["updated_at"].replace("Z", "+00:00"))
+    now = dt.datetime.now(dt.timezone.utc)
+    age = now - updated_at
+    logger.info(f"found run #{run['run_number']} updated {pretty_interval(age)} ago")
+    if age > dt.timedelta(hours=6):
+        logger.warning("maybe it's a bit old?")
+
+    logger.info(f"looking for run #{run['run_number']} artifacts")
+    resp = s.get(f"{run['url']}/artifacts")
+    resp.raise_for_status()
+    artifacts = resp.json()["artifacts"]
+
+    dest = Path("packages")
+    if not dest.exists():
+        logger.info(f"creating dir {dest}")
+        dest.mkdir()
+
+    for artifact in artifacts:
+        logger.info(f"downloading {artifact['name']} archive")
+        zip_url = artifact["archive_download_url"]
+        resp = s.get(zip_url)
+        with ZipFile(io.BytesIO(resp.content)) as zf:
+            logger.info("extracting archive content")
+            zf.extractall(dest)
+
+    logger.info(f"now you can run: 'twine upload -s {dest}/*'")
+
+
+def pretty_interval(td):
+    secs = td.total_seconds()
+    mins, secs = divmod(secs, 60)
+    hours, mins = divmod(mins, 60)
+    days, hours = divmod(hours, 24)
+    if days:
+        return f"{int(days)} days, {int(hours)} hours, {int(mins)} minutes"
+    elif hours:
+        return f"{int(hours)} hours, {int(mins)} minutes"
+    else:
+        return f"{int(mins)} minutes"
+
+
+if __name__ == "__main__":
+    try:
+        sys.exit(main())
+
+    except ScriptError as e:
+        logger.error("%s", e)
+        sys.exit(1)
+
+    except KeyboardInterrupt:
+        logger.info("user interrupt")
+        sys.exit(1)