Add management command to change Base URL. Closes kiwitcms#971

tcms/management/management/commands/baseurl.py

@@ -0,0 +1,34 @@
+from django.core.management.base import BaseCommand
+from django.contrib.sites.models import Site
+
+
+class Command(BaseCommand):
+    help = "Sets the base URL of KiwiTCMS instance. " \
+           "If no arguments given returns current base URL."
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            'baseurl', nargs='?', default=None,
+            help='Base URL of KiwiTCMS instance',
+        )
+        parser.add_argument(
+            'basename', nargs='?', default='KiwiTCMS',
+            help='Optional site name. Defaults to "KiwiTCMS"',
+        )
+
+    def handle(self, *args, **kwargs):
+        if not kwargs['baseurl']:
+            try:
+                site = Site.objects.get(id=1)
+                self.stdout.write('URL: "%s", name: "%s"' % (site.domain, site.name))
+            except Site.DoesNotExist:
+                self.stdout.write('Base URL is not defined')
+            return
+        site, created = Site.objects.update_or_create(
+                            id=1,
+                            defaults={
+                                'domain': kwargs['baseurl'],
+                                'name': kwargs['basename'],
+                            }
+                        )
+        self.stdout.write('Base URL updated successfully.')