add ability to remove package from status page

This commit is contained in:
2021-04-02 01:26:46 +03:00
parent 6470b13511
commit 3e8e54857f
3 changed files with 20 additions and 2 deletions

View File

@ -226,6 +226,7 @@ def _set_status_update_parser(root: SubParserAction) -> argparse.ArgumentParser:
nargs="*")
parser.add_argument("--status", help="new status", choices=[value.value for value in BuildStatusEnum],
default="success")
parser.add_argument("--remove", help="remove package status page", action="store_true")
parser.set_defaults(handler=handlers.StatusUpdate, lock=None, no_report=True, unsafe=True)
return parser

View File

@ -19,7 +19,7 @@
#
import argparse
from typing import Type
from typing import Callable, Type
from ahriman.application.application import Application
from ahriman.application.handlers.handler import Handler
@ -42,10 +42,11 @@ class StatusUpdate(Handler):
"""
client = Application(architecture, configuration).repository.reporter
status = BuildStatusEnum(args.status)
callback: Callable[[str], None] = lambda p: client.remove(p) if args.remove else client.update(p, status)
if args.package:
# update packages statuses
for package in args.package:
client.update(package, status)
callback(package)
else:
# update service status
client.update_self(status)