allow to run single command for multiple architectures at the same time

This commit is contained in:
Evgenii Alekseev 2021-03-15 04:57:10 +03:00
parent 2cef540cc0
commit a288986450
4 changed files with 46 additions and 35 deletions

View File

@ -23,7 +23,7 @@ optdepends=('aws-cli: sync to s3'
source=("https://github.com/arcan1s/ahriman/releases/download/$pkgver/$pkgname-$pkgver-src.tar.xz"
'ahriman.sysusers'
'ahriman.tmpfiles')
sha512sums=('21a1a216af25aeb580f5f2f574e9eed6eb79b74a1038c40a2e1594f18ca0afcd8a64358d11494b87a9a9ea25d5f6be27c39591b124a6ae2d7d4e78fbe44fdd1d'
sha512sums=('1af6c022bb1cf3bb55cb3ba426673ee73f98bd1a4de2191033a2d09bd73c6afb10c2a14ac3c70daa6ecbf7a62385178005f5276ee6a9942dc295d7c49d7ed412'
'13718afec2c6786a18f0b223ef8e58dccf0688bca4cdbe203f14071f5031ed20120eb0ce38b52c76cfd6e8b6581a9c9eaa2743eb11abbaca637451a84c33f075'
'55b20f6da3d66e7bbf2add5d95a3b60632df121717d25a993e56e737d14f51fe063eb6f1b38bd81cc32e05db01c0c1d80aaa720c45cde87f238d8b46cdb8cbc4')
backup=('etc/ahriman.ini'

View File

@ -19,6 +19,8 @@
#
import argparse
from multiprocessing import Pool
import ahriman.version as version
from ahriman.application.application import Application
@ -26,64 +28,81 @@ from ahriman.application.lock import Lock
from ahriman.core.configuration import Configuration
def add(args: argparse.Namespace, config: Configuration) -> None:
def _call(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
additional function to wrap all calls for multiprocessing library
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
with Lock(args.lock, architecture, args.force, config):
args.fn(args, architecture, config)
def add(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
add packages callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
Application.from_args(args, config).add(args.package, args.without_dependencies)
Application(architecture, config).add(args.package, args.without_dependencies)
def rebuild(args: argparse.Namespace, config: Configuration) -> None:
def rebuild(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
world rebuild callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
app = Application.from_args(args, config)
app = Application(architecture, config)
packages = app.repository.packages()
app.update(packages)
def remove(args: argparse.Namespace, config: Configuration) -> None:
def remove(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
remove packages callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
Application.from_args(args, config).remove(args.package)
Application(architecture, config).remove(args.package)
def report(args: argparse.Namespace, config: Configuration) -> None:
def report(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
generate report callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
Application.from_args(args, config).report(args.target)
Application(architecture, config).report(args.target)
def sync(args: argparse.Namespace, config: Configuration) -> None:
def sync(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
sync to remote server callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
Application.from_args(args, config).sync(args.target)
Application(architecture, config).sync(args.target)
def update(args: argparse.Namespace, config: Configuration) -> None:
def update(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
update packages callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
# typing workaround
def log_fn(line: str) -> None:
return print(line) if args.dry_run else app.logger.info(line)
app = Application.from_args(args, config)
app = Application(architecture, config)
packages = app.get_updates(args.package, args.no_aur, args.no_manual, args.no_vcs, log_fn)
if args.dry_run:
return
@ -91,20 +110,21 @@ def update(args: argparse.Namespace, config: Configuration) -> None:
app.update(packages)
def web(args: argparse.Namespace, config: Configuration) -> None:
def web(args: argparse.Namespace, architecture: str, config: Configuration) -> None:
'''
web server callback
:param args: command line args
:param architecture: repository architecture
:param config: configuration instance
'''
from ahriman.web.web import run_server, setup_service
app = setup_service(args.architecture, config)
run_server(app, args.architecture)
app = setup_service(architecture, config)
run_server(app, architecture)
if __name__ == '__main__':
parser = argparse.ArgumentParser(prog='ahriman', description='ArcHlinux ReposItory MANager')
parser.add_argument('-a', '--architecture', help='target architecture', required=True)
parser.add_argument('-a', '--architecture', help='target architectures', action='append')
parser.add_argument('-c', '--config', help='configuration path', default='/etc/ahriman.ini')
parser.add_argument('--force', help='force run, remove file lock', action='store_true')
parser.add_argument('--lock', help='lock file', default='/tmp/ahriman.lock')
@ -118,7 +138,8 @@ if __name__ == '__main__':
check_parser = subparsers.add_parser('check', description='check for updates')
check_parser.add_argument('package', help='filter check by packages', nargs='*')
check_parser.set_defaults(fn=update, no_aur=False, no_manual=True, no_vcs=False, dry_run=True)
check_parser.add_argument('--no-vcs', help='do not check VCS packages', action='store_true')
check_parser.set_defaults(fn=update, no_aur=False, no_manual=True, dry_run=True)
rebuild_parser = subparsers.add_parser('rebuild', description='rebuild whole repository')
rebuild_parser.set_defaults(fn=rebuild)
@ -153,5 +174,5 @@ if __name__ == '__main__':
exit(1)
config = Configuration.from_path(args.config)
with Lock(args.lock, args.architecture, args.force, config):
args.fn(args, config)
with Pool(len(args.architecture)) as pool:
pool.starmap(_call, [(args, architecture, config) for architecture in args.architecture])

View File

@ -19,7 +19,6 @@
#
from __future__ import annotations
import argparse
import logging
import os
import shutil
@ -53,16 +52,6 @@ class Application:
self.architecture = architecture
self.repository = Repository(architecture, config)
@classmethod
def from_args(cls: Type[Application], args: argparse.Namespace, config: Configuration) -> Application:
'''
constructor which has to be used to build instance from command line args
:param args: command line args
:param config: configuration instance
:return: application instance
'''
return cls(args.architecture, config)
def _known_packages(self) -> Set[str]:
'''
load packages from repository and pacman repositories

View File

@ -25,7 +25,7 @@ from dataclasses import dataclass
@dataclass
class RepositoryPaths:
'''
repository paths holder
repository paths holder. For the most operations with paths you want to use this object
:ivar root: repository root (i.e. ahriman home)
:ivar architecture: repository architecture
'''
@ -38,6 +38,7 @@ class RepositoryPaths:
'''
:return: directory for devtools chroot
'''
# for the chroot directory devtools will create own tree and we don't have to specify architecture here
return os.path.join(self.root, 'chroot')
@property
@ -45,14 +46,14 @@ class RepositoryPaths:
'''
:return: directory for manual updates (i.e. from add command)
'''
return os.path.join(self.root, 'manual')
return os.path.join(self.root, 'manual', self.architecture)
@property
def packages(self) -> str:
'''
:return: directory for built packages
'''
return os.path.join(self.root, 'packages')
return os.path.join(self.root, 'packages', self.architecture)
@property
def repository(self) -> str:
@ -66,7 +67,7 @@ class RepositoryPaths:
'''
:return: directory for downloaded PKGBUILDs for current build
'''
return os.path.join(self.root, 'sources')
return os.path.join(self.root, 'sources', self.architecture)
def create_tree(self) -> None:
'''