2024-03-02 17:42:26 -08:00

119 lines
3.2 KiB
Python

#!/usr/bin/python3
"""Add packages to a repository.
Functions:
==========
* adds(cachedir, pkgs)
* repo_add(db, pkgs, opts=None)
"""
import argparse
import logging
import sys
import packaging_scripts.pacman as pacman
import packaging_scripts.pkgfiles as pkgfiles
import packaging_scripts.repos as repos
from pathlib import Path
# ========== Constants ==========
DB_EXT = "db.tar.xz"
LOGFORMAT = "==> %(levelname)s %(message)s"
# ----- Exit codes -----
E_NOFILESERR = 1
E_REPO_ADDERR = 2
# ========== Logging setup ==========
console_formatter = logging.Formatter(LOGFORMAT)
syslog = logging.getLogger("packaging_scripts")
syslog.setLevel(logging.DEBUG)
stdout_handler = logging.StreamHandler(sys.stdout)
stdout_handler.setLevel(logging.INFO)
stdout_handler.setFormatter(console_formatter)
stdout_handler.addFilter(lambda record: record.levelno <= logging.INFO)
stderr_handler = logging.StreamHandler(sys.stderr)
stderr_handler.setLevel(logging.WARNING)
stderr_handler.setFormatter(console_formatter)
syslog.addHandler(stdout_handler)
syslog.addHandler(stderr_handler)
# ========== Main Script ==========
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"-c", "--cachedir", help="alternative directory to use for moving package"
)
parser.add_argument(
"-d",
"--db-filename",
help="alternative filename for database without extension",
)
parser.add_argument(
"--files-only",
action="store_true",
help="only add the built files to the repository",
)
parser.add_argument(
"-s",
"--sign",
dest="opts",
action="append_const",
const="--sign",
help="sign repository file",
)
parser.add_argument(
"-v", "--verbose", action="store_true", help="increase script verbosity"
)
parser.add_argument(
"repository",
choices=pacman.list_configured_repos(),
help="the repository to operate on",
metavar="repo",
)
parser.add_argument("packages", default=None, nargs="*", help="packages to add")
args = parser.parse_args()
repo = args.repository
opts = [] if args.opts is None else args.opts
cachedir = (
Path(args.cachedir)
if args.cachedir
else Path("/var") / "cache" / "pacman" / repo
)
db = (
cachedir / f"{args.db_filename}.{DB_EXT}"
if args.db_filename
else cachedir / f"{repo}.{DB_EXT}"
)
if args.verbose:
stdout_handler.setLevel(logging.DEBUG)
if args.packages:
pkg_tarballs = pkgfiles.filter(args.packages)
sigfiles = pkgfiles.filter(args.packages, signatures_only=True)
else:
pkg_tarballs = list(pkgfiles.get())
sigfiles = list(pkgfiles.get(signatures_only=True))
if not pkg_tarballs:
syslog.critical("No package tarballs have been found, exiting")
exit(E_NOFILESERR)
if not args.files_only:
try:
repos.db_modify("add", db, *opts, *pkg_tarballs)
except repos.RepoAddError as e:
syslog.error(e)
exit(E_REPO_ADDERR)
for pkgfile in (*pkg_tarballs, *sigfiles):
pkgfiles.add(pkgfile, cachedir)