From 3733d5dd1ac64dff102c34122c59f4843bed3e43 Mon Sep 17 00:00:00 2001 From: Antoine Nguyen Date: Thu, 10 Dec 2015 09:41:37 +0100 Subject: [PATCH] Management command for django 1.8 see #810 --- modoboa/core/management/commands/set_default_site.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/modoboa/core/management/commands/set_default_site.py b/modoboa/core/management/commands/set_default_site.py index 03f703267..78db35a1f 100644 --- a/modoboa/core/management/commands/set_default_site.py +++ b/modoboa/core/management/commands/set_default_site.py @@ -14,13 +14,17 @@ class Command(BaseCommand): """Management command to set the default site.""" - help = 'Set default site (see django.contrib.sites)' + help = "Set default site (see django.contrib.sites)" + + def add_arguments(self, parser): + """Define command arguments.""" + parser.add_argument("hostname", type=str) def handle(self, *args, **options): """Command entry point.""" - if not args: + if "hostname" not in options: raise CommandError("You must provide a hostname") site = Site.objects.get(pk=1) - site.domain = args[0] - site.name = args[0] + site.domain = options["hostname"] + site.name = options["hostname"] site.save()