Moved scripts to packaging_scripts.bin

packaging_scripts/bin/addpkg

@@ -0,0 +1,122 @@
+#!/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 ==========
+def 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)
+
+
+if __name__ == "__main__":
+    main()