prime/tools/upload_progress.py

#!/usr/bin/env python3

###
# Uploads progress information to https://github.com/decompals/frogress.
#
# Usage:
#   python3 tools/upload_progress.py -b https://progress.decomp.club/ -p [project] -v [version] build/[version]/progress.json
#
# If changes are made, please submit a PR to
# https://github.com/encounter/dtk-template
###

import argparse
import json
import os
import requests
import subprocess
import sys


def get_git_commit_timestamp() -> int:
    return int(
        subprocess.check_output(["git", "show", "-s", "--format=%ct"])
        .decode("ascii")
        .rstrip()
    )


def get_git_commit_sha() -> str:
    return subprocess.check_output(["git", "rev-parse", "HEAD"]).decode("ascii").strip()


def generate_url(args: argparse.Namespace) -> str:
    url_components = [args.base_url.rstrip("/"), "data"]

    for arg in [args.project, args.version]:
        if arg != "":
            url_components.append(arg)

    return str.join("/", url_components) + "/"


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Upload progress information.")
    parser.add_argument("-b", "--base_url", help="API base URL", required=True)
    parser.add_argument("-a", "--api_key", help="API key (env var PROGRESS_API_KEY)")
    parser.add_argument("-p", "--project", help="Project slug", required=True)
    parser.add_argument("-v", "--version", help="Version slug", required=True)
    parser.add_argument("input", help="Progress JSON input")

    args = parser.parse_args()
    api_key = args.api_key or os.environ.get("PROGRESS_API_KEY")
    if not api_key:
        raise KeyError("API key required")
    url = generate_url(args)

    entries = []
    with open(args.input, "r") as f:
        data = json.load(f)
        entries.append(
            {
                "timestamp": get_git_commit_timestamp(),
                "git_hash": get_git_commit_sha(),
                "categories": data,
            }
        )

    print("Publishing entry to", url)
    json.dump(entries[0], sys.stdout, indent=4)
    print()
    r = requests.post(
        url,
        json={
            "api_key": api_key,
            "entries": entries,
        },
    )
    r.raise_for_status()
    print("Done!")