get_latest_artifact.py

import requests
from dateutil import parser
import os
import zipfile
from pprint import pprint
import logging
logging.basicConfig(level=logging.DEBUG)
repo=f"https://api.github.com/repos/{os.environ['INPUT_REPO']}/actions/artifacts"

print(f"retrieving artifacts from repo {repo}")

artifacts = requests.get(repo).json().get('artifacts')

if not artifacts:
  print("Did not find a Dracon DB, this is not fatal as the enricher will create it, exiting")
  exit(0)

latest = parser.isoparse(artifacts[-1]['updated_at'])
artifact_url = None
for art in artifacts:
    if art["expired"] == "true" or "dracon_enrichment_db" not in art["name"]:
        continue
    d =parser.isoparse(art['updated_at'])
    if d > latest:
        latest = d
        artifact_url = art["archive_download_url"]

if not artifact_url:
    print("Did not find a Dracon DB, this is not fatal as the enricher will create it, exiting")
    exit(0)


token =  os.environ["INPUT_GH_ACCESS_TOKEN"]
if len(token) == 0:
    pprint("empty token")
resp = requests.get(artifact_url, stream=True,headers={"Authorization" :"token %s"%token})
if resp.status_code == 200:
    pprint("success! Downloaded latest enrichment database")
elif resp.status_code != 200:
    print(f"Error receiving files for artifact {artifact_url}  status code was {resp.status_code}, error message is:")
    pprint(requests.get(artifact_url).json())
    raise ValueError

with open("/tmp/latest.zip", "wb") as fl:
    for chunk in resp.iter_content():
        fl.write(chunk)
try:
    with zipfile.ZipFile("/tmp/latest.zip") as zip:
        print("Extracting files to %s"%os.environ["INPUT_OUTPUT_DIR"])
        zip.extractall(os.environ["INPUT_OUTPUT_DIR"])

    print("Files extracted, dir contents are:")
    pprint(os.listdir(os.environ["INPUT_OUTPUT_DIR"]))
except zipfile.BadZipFile as bzf:
    with open("/tmp/latest.zip") as l:
        pprint(l.read())