Sven Luthi
c32161b8a2
modified: netbox_dns_bridge/dns_server.py modified: netbox_dns_bridge/management/commands/dns-export-zone.py modified: netbox_dns_bridge/management/commands/dns-export-zones.py modified: netbox_dns_bridge/management/commands/dns-settings.py modified: netbox_dns_bridge/management/commands/dns-transfer-endpoint.py modified: netbox_dns_bridge/migrations/0001_initial.py modified: netbox_dns_bridge/models.py modified: netbox_dns_bridge/request_handler.py modified: netbox_dns_bridge/signals.py modified: netbox_dns_bridge/utils.py Black code formating corrected
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
from django.core.management.base import BaseCommand, CommandError
|
|
from netbox_dns_bridge.models import IntegerKeyValueSetting
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Get or set integer key-value settings"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"action", choices=["get", "set", "list"], help="Action to perform"
|
|
)
|
|
parser.add_argument("key", nargs="?", help="Setting key")
|
|
parser.add_argument(
|
|
"value", nargs="?", type=int, help="Value to set (required for 'set')"
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
action = options["action"]
|
|
key = options.get("key")
|
|
value = options.get("value")
|
|
|
|
if action == "list":
|
|
settings = IntegerKeyValueSetting.objects.all()
|
|
if not settings.exists():
|
|
self.stdout.write("No settings found.")
|
|
return
|
|
for setting in settings:
|
|
self.stdout.write(str(setting))
|
|
|
|
elif action == "get":
|
|
if not key:
|
|
raise CommandError("A key is required for GET")
|
|
try:
|
|
setting = IntegerKeyValueSetting.objects.get(key=key)
|
|
self.stdout.write(f"{setting.key}: {setting.value}")
|
|
except IntegerKeyValueSetting.DoesNotExist:
|
|
raise CommandError(f"No setting was found for {key}")
|
|
|
|
elif action == "set":
|
|
if not key:
|
|
raise CommandError("A setting is required for SET command")
|
|
if value is None:
|
|
raise CommandError("A value is required for SET command")
|
|
try:
|
|
setting = IntegerKeyValueSetting.objects.get(key=key)
|
|
old_value = setting.value
|
|
setting.value = value
|
|
setting.save()
|
|
self.stdout.write(f"Updated {key} from {old_value} to {value}")
|
|
except IntegerKeyValueSetting.DoesNotExist:
|
|
raise CommandError(f"No setting found for {key}")
|