137 lines
3.8 KiB
Python
137 lines
3.8 KiB
Python
#!/usr/bin/python3
|
|
""" Add packages to a repository.
|
|
|
|
Functions:
|
|
==========
|
|
add_pkgfiles(cachedir, pkgs)
|
|
repo_add(db, pkgs, opts=None)
|
|
"""
|
|
|
|
import argparse
|
|
import logging
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
|
|
import packaging_scripts.pacmanconf as pacmanconf
|
|
import packaging_scripts.pkgfiles as pkgfiles
|
|
|
|
# ========== Constants ==========
|
|
REPO_ADD_CMD = '/usr/bin/repo-add'
|
|
DB_EXT = 'db.tar.xz'
|
|
|
|
# ========== Exit codes ==========
|
|
E_REPO_ADDERR = 2
|
|
|
|
# ========== Logging setup ==========
|
|
console_formatter = logging.Formatter('==> %(levelname)s %(message)s')
|
|
syslog = logging.getLogger(__name__)
|
|
syslog.setLevel(logging.DEBUG)
|
|
|
|
stdout_handler = logging.StreamHandler(sys.stdout)
|
|
stdout_handler.setLevel(logging.INFO)
|
|
stdout_handler.setFormatter(console_formatter)
|
|
|
|
stderr_handler = logging.StreamHandler()
|
|
stderr_handler.setLevel(logging.ERROR)
|
|
stderr_handler.setFormatter(console_formatter)
|
|
|
|
syslog.addHandler(stdout_handler)
|
|
syslog.addHandler(stderr_handler)
|
|
|
|
|
|
# ========== Functions ==========
|
|
def add_pkgfile(pkg, cachedir):
|
|
"""Add package file to the repository directory.
|
|
|
|
:param pkg: path of package to add
|
|
:type pkg: str, bytes, or path-like object
|
|
:param cachedir: cache directory to move package to
|
|
:type cachedir: str, bytes, or path-like object
|
|
"""
|
|
syslog.info(f"Adding {pkg} to {cachedir}")
|
|
shutil.move(pkg, cachedir)
|
|
|
|
|
|
def repo_add(db, pkgs, opts=None):
|
|
"""Run repo-add.
|
|
|
|
:param repo: the repository to remove from
|
|
:type repo: str
|
|
:param pkgs: the names of the packages to remove
|
|
:type pkgs: any iterable
|
|
:param opts: extra options to pass to repo-add
|
|
:type opts: list
|
|
:raises: subprocess.CalledProcessError if repo-add failed
|
|
"""
|
|
syslog.info('Adding packages to database')
|
|
syslog.debug(f"Options: {opts}")
|
|
syslog.debug(f"Packages: {pkgs}")
|
|
|
|
cmd = [REPO_ADD_CMD]
|
|
|
|
if opts is not None:
|
|
cmd.extend(opts)
|
|
|
|
cmd.append(db)
|
|
cmd.extend(pkgs)
|
|
syslog.debug(f"Final repo-add command: {cmd}")
|
|
|
|
add_process = subprocess.run(cmd,
|
|
check=True,
|
|
capture_output=True,
|
|
text=True)
|
|
|
|
syslog.debug(add_process.stdout)
|
|
if add_process.stderr:
|
|
syslog.error(add_process.stderr)
|
|
|
|
syslog.info('Finished adding packages to database')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
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=pacmanconf.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
|
|
pkgs = args.packages
|
|
opts = args.opts
|
|
|
|
cachedir = f"/var/cache/pacman/{repo}"
|
|
# this assumes that the db file for the repo
|
|
# has the same name as that repo
|
|
db = f"{cachedir}/{repo}.{DB_EXT}"
|
|
|
|
if args.verbose:
|
|
stdout_handler.setLevel(logging.DEBUG)
|
|
|
|
if pkgs:
|
|
pkglist = pkgs
|
|
else:
|
|
pkglist = pkgfiles.get_pkgfiles() +\
|
|
pkgfiles.get_pkgfiles(signatures_only=True)
|
|
|
|
try:
|
|
repo_add(db, pkglist, opts)
|
|
except subprocess.CalledProcessError as e:
|
|
syslog.error(e)
|
|
exit(E_REPO_ADDERR)
|
|
|
|
map(lambda pkg: add_pkgfile(pkg, cachedir), pkglist)
|