82 lines
2.8 KiB
Plaintext
82 lines
2.8 KiB
Plaintext
|
#!/usr/bin/env python
|
||
|
|
||
|
import argparse, os, re, sys, requests
|
||
|
|
||
|
USERNAME = os.environ["BINTRAY_USERNAME"]
|
||
|
API_KEY = os.environ["BINTRAY_API_KEY"]
|
||
|
API_BASE = os.environ.get("BINTRAY_URL", "https://api.bintray.com/content")
|
||
|
ORGANIZATION = os.environ.get("BINTRAY_ORGANIZATION")
|
||
|
ALLOWED_ARCH = ["all", "arm", "i686", "aarch64", "x86_64"]
|
||
|
|
||
|
VERSION_RE = re.compile("^\d+(\.\d+)+([^\-]+)?\-([\da-z]+)$")
|
||
|
VERIFY_FORMAT = """
|
||
|
Does this look correct?
|
||
|
- Name: {a.name}
|
||
|
- Version: {a.version}
|
||
|
- Architecture: {a.architecture}
|
||
|
- Distribution: {a.distribution}
|
||
|
- Component: {a.component}
|
||
|
- BinTray Repo: {a.org}/{a.repo}
|
||
|
|
||
|
(Please enter y/n): """
|
||
|
|
||
|
|
||
|
def main():
|
||
|
parser = argparse.ArgumentParser(description="Upload BinTray package.")
|
||
|
parser.add_argument("package")
|
||
|
parser.add_argument("--name", "-n")
|
||
|
parser.add_argument("--no-confirm", action='store_true', default=False)
|
||
|
parser.add_argument("--repo", "-r", default="main")
|
||
|
parser.add_argument("--org", "-o", default=ORGANIZATION)
|
||
|
parser.add_argument("--component", "-c", default="main")
|
||
|
parser.add_argument("--distribution", "-d", default="stable")
|
||
|
|
||
|
args = parser.parse_args()
|
||
|
basename = os.path.basename(args.package)
|
||
|
|
||
|
if not os.path.isfile(args.package):
|
||
|
raise ValueError("File {0} is not valid / does not exist.".format(args.package))
|
||
|
|
||
|
if not args.org:
|
||
|
raise ValueError("Organization is missing.")
|
||
|
|
||
|
# Determine name, version and architecture from file name:
|
||
|
base, _ = os.path.splitext(basename)
|
||
|
name, version, arch = base.split("_")
|
||
|
if arch not in ALLOWED_ARCH: raise ValueError("Architecture {0} not discoverable".format(arch))
|
||
|
if not VERSION_RE.match(version): raise ValueError("Version not discoverable ({0})".format(version))
|
||
|
|
||
|
args.architecture = arch
|
||
|
args.version = version
|
||
|
args.name = args.name or name
|
||
|
|
||
|
url = "{url}/{a.org}/{a.repo}/{a.name}/{a.version}/pool/{a.component}" \
|
||
|
"/{a.name[0]}/{basename}?publish=1".format(
|
||
|
url=API_BASE, a=args, basename=basename)
|
||
|
|
||
|
if not args.no_confirm:
|
||
|
sys.stdout.write(VERIFY_FORMAT.format(a=args))
|
||
|
if not sys.stdin.readline().rstrip() == "y":
|
||
|
print("\nNot submitting package.")
|
||
|
return
|
||
|
|
||
|
print("Submitting package...")
|
||
|
|
||
|
parameters = {"publish": "1"}
|
||
|
headers = {
|
||
|
"X-Bintray-Debian-Distribution": args.distribution,
|
||
|
"X-Bintray-Debian-Architecture": args.architecture,
|
||
|
"X-Bintray-Debian-Component": args.component
|
||
|
}
|
||
|
|
||
|
with open(args.package, "rb") as package_fp:
|
||
|
response = requests.put(url, auth=(USERNAME, API_KEY), params=parameters, headers=headers, data=package_fp)
|
||
|
|
||
|
if response.status_code != 201:
|
||
|
raise Exception("Failed to submit package: {0}\n{1}".format(response.status_code, response.text))
|
||
|
|
||
|
print("Submitted successfully.")
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|