Source code for tcms.core.management.commands.set_domain
from django.conf import settings
from django.contrib.sites.models import Site
from django.core.management.base import BaseCommand
[docs]class Command(BaseCommand):
help = (
"Sets the domain of Kiwi TCMS instance. "
"If no arguments given returns current domain."
)
[docs] def add_arguments(self, parser):
parser.add_argument(
"domain",
nargs="?",
default=None,
help="The domain of Kiwi TCMS instance",
)
[docs] def handle(self, *args, **kwargs):
site = Site.objects.get(id=settings.SITE_ID)
if not kwargs["domain"]:
self.stdout.write(site.domain)
return
site.domain = kwargs["domain"]
site.name = "Kiwi TCMS"
site.save()
self.stdout.write("Domain updated successfully.")