toolkit/mangle/quick-bump.py
Neil Hanlon 7557ff5987
Some checks failed
Build empanada images for imagefactory / buildx (push) Failing after 5s
Build empanada container images for lorax / buildx (push) Successful in 1s
feat: support looping over packages; formatting
2024-07-09 18:00:48 -04:00

189 lines
5.7 KiB
Python
Executable File

#!/usr/bin/env python3
#
# quick-bump.py - Quickly bumps a release version for a rebuild. This is mostly
# used in cases of */src/* package rebuilds like for SIG's. It
# should be rare that the base distribution needs this.
import os
import subprocess
import sys
import argparse
import shutil
parser = argparse.ArgumentParser()
parser.add_argument(
"--pkg", help="Package name to bump (can be comma delimited list)", required=True
)
parser.add_argument("--branch", help="Package branch", required=True)
parser.add_argument("--sig", help="Name of SIG that holds this package")
parser.add_argument(
"--peridot-import",
help="Tell peridot to import",
required=False,
action="store_true",
)
parser.add_argument(
"--peridot-endpoint",
help="Peridot API Endpoint (PERIDOT_ENDPOINT)",
required=False,
default="peridot-api.build.resf.org",
)
parser.add_argument(
"--peridot-hdr-endpoint",
help="Peridot HDR Endpoint (PERIDOT_HDR_ENDPOINT)",
required=False,
default="hdr.build.resf.org",
)
parser.add_argument(
"--peridot-project-id",
help="Peridot project ID (PERIDOT_PROJECT_ID)",
required=False,
)
parser.add_argument(
"--peridot-client-id", help="Peridot client ID (PERIDOT_CLIENT_ID)", required=False
)
parser.add_argument(
"--peridot-client-secret",
help="Peridot client secret (PERIDOT_CLIENT_SECRET)",
required=False,
)
parser.add_argument(
"--dry", help="Do a dry bump for testing", required=False, action="store_true"
)
parser.add_argument(
"--rightmost",
help="see rpmdev-bumpspec; noop with --string",
required=False,
action="store_true",
)
parser.add_argument(
"--string", help="see rpmdev-bumpspec; trumps --rightmost", required=False
)
parser.add_argument("--git-user", default="Release Engineering")
parser.add_argument("--git-email", default="releng@rockylinux.org")
parser.add_argument(
"--url",
default="ssh://git@git.rockylinux.org:22220/staging/src/{pkg}.git",
help="override default URL",
)
parser.add_argument(
"--change-user",
help="Sets the user for the rpm changelog (first last <email>)",
default="Release Engineering <releng@rockylinux.org>",
)
parser.add_argument(
"--change-comment",
help="Sets the comment that will appear in the changelog and commit message",
default="Release tag bump for rebuild (https://sig-core.rocky.page/rebuild/)",
)
args = parser.parse_args(args=None if sys.argv[1:] else ["--help"])
user = args.change_user
comment = args.change_comment
if args.peridot_import:
peridot_api_endpoint = os.environ.get(
"PERIDOT_ENDPOINT") or args.peridot_endpoint
peridot_hdr_endpoint = (
os.environ.get("PERIDOT_HDR_ENDPOINT") or args.peridot_hdr_endpoint
)
peridot_project_id = os.environ.get(
"PERIDOT_PROJECT_ID") or args.peridot_project_id
peridot_client_id = os.environ.get(
"PERIDOT_CLIENT_ID") or args.peridot_client_id
peridot_client_secret = (
os.environ.get("PERIDOT_CLIENT_SECRET") or args.peridot_client_secret
)
print("Peridot import not supported yet")
sys.exit(1)
if os.geteuid() == 0:
print("DO NOT RUN AS ROOT")
sys.exit(1)
if args.sig:
default_url = "ssh://git@git.rockylinux.org:22220/sig/{sig}/src/{pkg}.git"
if args.url:
default_url = getattr(args, "url")
rightmost = "--rightmost" if args.rightmost else None
string = f"--string={args.string}" if args.string else ""
extra_args = [arg for arg in (rightmost, string) if arg]
# functions
workdir = "/var/tmp"
environment = os.environ
pkgs = args.pkg.split(",")
def runcmd(cmd, action, package, env, pwd=workdir):
"""
Runs a command using subprocess and returns 0 or 1
"""
try:
subprocess.check_call(cmd, env=env, cwd=pwd)
except subprocess.CalledProcessError as err:
sys.stderr.write("%s failed %s: %s\n" % (package, action, err))
return 1
return 0
for pkg in pkgs:
joined_dir = os.path.join(workdir, pkg)
spec_dir = os.path.join(workdir, pkg, "SPECS")
checkout_url = default_url.format(pkg=pkg, sig=args.sig)
print(f"Checking out {checkout_url}")
gitcmd = ["git", "clone", checkout_url,
"--branch", args.branch, joined_dir]
if runcmd(gitcmd, "git clone", pkg, environment):
continue
files = os.listdir(spec_dir)
spec = ""
for file in files:
if file.endswith(".spec"):
spec = os.path.join(spec_dir, file)
break
if not spec:
sys.stderr.write("Failed to find a spec for %s\n" % pkg)
continue
print("Bumping release of %s (%s)" % (spec, extra_args))
bumprel = ["rpmdev-bumpspec", "-D", "-u",
user, "-c", comment, spec, *extra_args]
bumprel_old = ["rpmdev-bumpspec", "-u",
user, "-c", comment, spec, *extra_args]
if runcmd(bumprel, "rpmdev-bumpspec", pkg, environment):
print("Potentially old bumpspec version. Trying again.")
if runcmd(bumprel_old, "rpmdev-bumpspec", pkg, environment):
continue
print("Setting git user and email for this operation")
git_name = ["git", "config", "user.name", args.git_user]
git_mail = ["git", "config", "user.email", args.git_email]
if runcmd(git_name, "git_name", pkg, environment, pwd=joined_dir):
continue
if runcmd(git_mail, "git_mail", pkg, environment, pwd=joined_dir):
continue
print("Committing changes")
commit = ["git", "commit", "-asm", comment, "--allow-empty"]
if runcmd(commit, "commit", pkg, environment, pwd=joined_dir):
continue
if not args.dry:
push = ["git", "push"]
print("Pushing changes for %s" % pkg)
if runcmd(push, "push", pkg, environment, pwd=joined_dir):
continue
shutil.rmtree(joined_dir)