Peter van Arkel
1c8cbd96ca
Prefer --schedule-expression for scripted schedule updates while keeping --cron as a compatibility alias. Clean up management command help, errors, and output so operator-facing text talks about hosts, global config, and Django backup configuration instead of model names or old SQL-backed pobsync wording.
66 lines
3.0 KiB
Python
66 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
from pobsync.config.retention import parse_retention
|
|
from pobsync.util import sanitize_host
|
|
from pobsync_backend.models import GlobalConfig, HostConfig
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Create or update a host backup configuration."
|
|
|
|
def add_arguments(self, parser) -> None:
|
|
parser.add_argument("host")
|
|
parser.add_argument("--address", required=True)
|
|
parser.add_argument("--ssh-user", default="")
|
|
parser.add_argument("--ssh-port", type=int, default=None)
|
|
parser.add_argument("--source-root", default="")
|
|
parser.add_argument("--include", action="append", default=[])
|
|
parser.add_argument("--exclude-add", action="append", default=[])
|
|
parser.add_argument("--exclude-replace", action="append", default=None)
|
|
parser.add_argument("--rsync-extra-arg", action="append", default=[])
|
|
parser.add_argument("--retention", default=None)
|
|
parser.add_argument("--disabled", action="store_true")
|
|
parser.add_argument("--force", action="store_true", help="Update existing host")
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
host = sanitize_host(options["host"])
|
|
if HostConfig.objects.filter(host=host).exists() and not options["force"]:
|
|
raise CommandError(f"Host {host!r} already exists; use --force to update")
|
|
|
|
retention = self._retention(options["retention"])
|
|
defaults = {
|
|
"address": options["address"],
|
|
"enabled": not options["disabled"],
|
|
"ssh_user": options["ssh_user"],
|
|
"ssh_port": options["ssh_port"],
|
|
"source_root": options["source_root"],
|
|
"includes": list(options["include"]),
|
|
"excludes_add": [] if options["exclude_replace"] is not None else list(options["exclude_add"]),
|
|
"excludes_replace": options["exclude_replace"],
|
|
"rsync_extra_args": list(options["rsync_extra_arg"]),
|
|
"retention_daily": retention["daily"],
|
|
"retention_weekly": retention["weekly"],
|
|
"retention_monthly": retention["monthly"],
|
|
"retention_yearly": retention["yearly"],
|
|
}
|
|
_obj, created = HostConfig.objects.update_or_create(host=host, defaults=defaults)
|
|
action = "Created" if created else "Updated"
|
|
self.stdout.write(self.style.SUCCESS(f"{action} host {host!r}."))
|
|
|
|
def _retention(self, value: str | None) -> dict[str, int]:
|
|
if value:
|
|
return parse_retention(value)
|
|
global_config = GlobalConfig.objects.filter(name="default").first()
|
|
if global_config is None:
|
|
return {"daily": 14, "weekly": 8, "monthly": 12, "yearly": 0}
|
|
return {
|
|
"daily": global_config.retention_daily,
|
|
"weekly": global_config.retention_weekly,
|
|
"monthly": global_config.retention_monthly,
|
|
"yearly": global_config.retention_yearly,
|
|
}
|