generate.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

#!/usr/bin/env python
# --------------------------------------------------------------------
# generate.py
#
# Author: Lain Musgrove (lain.proliant@gmail.com)
# Date: Friday September 15, 2023
# --------------------------------------------------------------------

import hashlib

import requests
import getpass
from jinja2 import Template
from xeno.shell import Shell, check

# --------------------------------------------------------------------
NAME = "xeno"


# --------------------------------------------------------------------
def pypi_checksum(package_name, version):
    url = f"https://pypi.org/pypi/{package_name}/json"
    response = requests.get(url)
    data = response.json()
    download_url = data["releases"][version][0]["url"]
    response = requests.get(download_url)
    checksum = hashlib.sha256(response.content).hexdigest()
    return checksum


# --------------------------------------------------------------------
def pypi_latest_version(package_name):
    url = f"https://pypi.org/pypi/{package_name}/json"
    response = requests.get(url)
    data = response.json()
    return data["info"]["version"]


# --------------------------------------------------------------------
def load_template(file):
    with open(file, "r") as infile:
        return Template(infile.read())


# --------------------------------------------------------------------
def main():
    pkgbuild_template = load_template("./PKGBUILD.jinja")
    srcinfo_template = load_template("./SRCINFO.jinja")

    version = pypi_latest_version(NAME)

    data = {"version": version, "checksum": pypi_checksum(NAME, version)}

    with open("./.SRCINFO", "w") as outfile:
        outfile.write(srcinfo_template.render(**data))

    with open("./PKGBUILD", "w") as outfile:
        outfile.write(pkgbuild_template.render(**data))

    check("makepkg -fs")
    Shell().interact(f"git add .SRCINFO PKGBUILD generate.py && git commit -m 'Update v{version}'", check=True, version=version)

    if getpass.getuser() == "lainproliant":
        check("sgit lainproliant push")
    else:
        check("git push")


# --------------------------------------------------------------------
if __name__ == "__main__":
    main()