140 lines
4.3 KiB
Python
140 lines
4.3 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding=utf-8
|
|
from pyalpm import Handle
|
|
from sys import argv, exit, stdout
|
|
from os import listdir
|
|
from os.path import exists, basename, join
|
|
from logging import warning, info, basicConfig, INFO
|
|
from requests import get, post, put, delete, RequestException, Response
|
|
from argparse import ArgumentParser
|
|
|
|
default_server = "https://repo-updater.classfun.cn"
|
|
|
|
|
|
def FormatSize(size: int) -> str:
|
|
units = ['Bytes', 'KB', 'MB', 'GB', 'TB', 'GB']
|
|
unit = units[0]
|
|
for unit in units:
|
|
if size < 1024:
|
|
break
|
|
size /= 1024
|
|
return "{:.2f} {}".format(size, unit)
|
|
|
|
|
|
def GetReasonText(res: Response, default: str = "Unknown") -> str:
|
|
if "text/plain" not in res.headers["Content-Type"]:
|
|
return default
|
|
length = res.headers["Content-Length"]
|
|
if length is None or int(length) >= 128:
|
|
return default
|
|
return res.text.strip()
|
|
|
|
|
|
def UploadFile(url: str, file: str, code: int = 201):
|
|
with open(file, "rb") as f:
|
|
res = put(url, f.read())
|
|
text = GetReasonText(res)
|
|
if res.status_code != code:
|
|
raise RequestException(
|
|
"upload %s status not %d: %d (%s)" %
|
|
(file, code, res.status_code, text)
|
|
)
|
|
|
|
|
|
def UploadPackage(
|
|
pkg: str,
|
|
sign: str = None,
|
|
arch: str = None,
|
|
server: str = default_server,
|
|
handle: Handle = None
|
|
):
|
|
if sign is None:
|
|
sign = pkg + ".sig"
|
|
if handle is None:
|
|
handle = Handle("/", "/var/lib/pacman")
|
|
if not exists(pkg):
|
|
raise FileNotFoundError("Target package file %s not found" % pkg)
|
|
if not exists(sign):
|
|
raise FileNotFoundError("Target signature file %s not found" % pkg)
|
|
if pkg + ".sig" != sign:
|
|
warning("signature filename mismatch with package file name")
|
|
alpm = handle.load_pkg(pkg)
|
|
if not alpm:
|
|
raise IOError("Open package %s failed" % pkg)
|
|
info("package name: %s" % alpm.name)
|
|
info("package version: %s" % alpm.version)
|
|
info("package architecture: %s" % alpm.arch)
|
|
info("package packager: %s" % alpm.packager)
|
|
info("package size: %s" % FormatSize(alpm.size))
|
|
info("package installed size: %s" % FormatSize(alpm.isize))
|
|
info("package url: %s" % alpm.url)
|
|
name = "%s-%s-%s" % (alpm.name, alpm.version, alpm.arch)
|
|
res = get("%s/api/info" % server)
|
|
if arch is None:
|
|
arch = alpm.arch
|
|
if arch == "any":
|
|
raise ValueError("Unable to detect target architecture")
|
|
if res.status_code != 200:
|
|
raise RequestException("status not 200: %d" % res.status_code)
|
|
base = res.json()
|
|
pkg_ext = next((i for i in base["pkg_exts"] if pkg.endswith(i)), None)
|
|
sign_ext = next((i for i in base["sign_exts"] if sign.endswith(i)), None)
|
|
if pkg_ext is None:
|
|
raise ValueError("Unknown package type")
|
|
if sign_ext is None:
|
|
raise ValueError("Unknown signature type")
|
|
if arch not in base["arch"]:
|
|
raise ValueError("Target architecture not found")
|
|
pkg_name = name + pkg_ext
|
|
sign_name = name + sign_ext
|
|
if pkg_name != basename(pkg):
|
|
warning("package filename mismatch with metadata")
|
|
info("upload as %s" % pkg_name)
|
|
try:
|
|
UploadFile("%s/%s" % (server, sign_name), sign)
|
|
UploadFile("%s/%s" % (server, pkg_name), pkg)
|
|
res = post(
|
|
"%s/api/update" % server,
|
|
json={"arch": arch, "target": pkg_name},
|
|
headers={"Content-Type": "application/json"}
|
|
)
|
|
text = GetReasonText(res)
|
|
if res.status_code != 200:
|
|
raise RequestException(
|
|
"update status not 201: %d (%s)" %
|
|
(res.status_code, text)
|
|
)
|
|
info("upload done")
|
|
finally:
|
|
delete("%s/%s" % (server, pkg_name))
|
|
delete("%s/%s" % (server, sign_name))
|
|
|
|
|
|
def main(args: list) -> int:
|
|
prs = ArgumentParser("Renegade Project Arch Linux Repo Uploader")
|
|
prs.add_argument("-a", "--arch", help="Target repo architecture", required=False)
|
|
prs.add_argument("-d", "--dir", help="Package folder", required=False)
|
|
prs.add_argument("-p", "--pkg", help="Package tarball file", required=False)
|
|
prs.add_argument("-s", "--sign", help="Package signature file", required=False)
|
|
prs.add_argument("-u", "--url", help="Updater Server URL", required=False, default=default_server)
|
|
ps = prs.parse_args(args[1:])
|
|
basicConfig(level=INFO, stream=stdout)
|
|
cnt = 0
|
|
if ps.pkg:
|
|
UploadPackage(ps.pkg, ps.sign, ps.arch, ps.url)
|
|
cnt += 1
|
|
elif ps.dir:
|
|
exts = [".pkg.tar.gz", ".pkg.tar.xz", ".pkg.tar.zst"]
|
|
for f in listdir(ps.dir):
|
|
full = join(ps.dir, f)
|
|
if any(f.endswith(ext) for ext in exts):
|
|
UploadPackage(full, None, ps.arch, ps.url)
|
|
cnt += 1
|
|
if cnt <= 0:
|
|
raise Exception("no any package found")
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
exit(main(argv))
|