45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
|
||
|
|
from pobsync_backend.self_check import collect_self_checks, summarize_self_checks
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = "Run pobsync runtime self checks for native installs and updates."
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument(
|
||
|
|
"--fail-on-warning",
|
||
|
|
action="store_true",
|
||
|
|
help="Exit with an error when warnings are present.",
|
||
|
|
)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
checks = collect_self_checks()
|
||
|
|
summary = summarize_self_checks(checks)
|
||
|
|
|
||
|
|
for check in checks:
|
||
|
|
line = f"[{check.status.upper()}] {check.name}: {check.message}"
|
||
|
|
if check.detail:
|
||
|
|
line = f"{line} ({check.detail})"
|
||
|
|
if check.status == "failed":
|
||
|
|
self.stderr.write(line)
|
||
|
|
elif check.status == "warning":
|
||
|
|
self.stderr.write(line)
|
||
|
|
else:
|
||
|
|
self.stdout.write(line)
|
||
|
|
|
||
|
|
self.stdout.write(
|
||
|
|
"Summary: "
|
||
|
|
f"{summary['ok']} ok, "
|
||
|
|
f"{summary['warning']} warning(s), "
|
||
|
|
f"{summary['failed']} failed, "
|
||
|
|
f"{summary['skipped']} skipped"
|
||
|
|
)
|
||
|
|
|
||
|
|
if summary["failed"]:
|
||
|
|
raise CommandError("pobsync install self check failed.")
|
||
|
|
if options["fail_on_warning"] and summary["warning"]:
|
||
|
|
raise CommandError("pobsync install self check reported warnings.")
|