addist: Allow to be called with an exisitng site.
This commit is contained in:
parent
cb93534a9e
commit
87c8da1a9e
|
@ -9,5 +9,8 @@ class Command(BaseCommand):
|
|||
parser.add_argument('domain', type=str)
|
||||
|
||||
def handle(self, *args, **options):
|
||||
site = Site.objects.create(domain=options['domain'], name=options['domain'])
|
||||
self.stdout.write(self.style.SUCCESS('Created site {}'.format(site)))
|
||||
site, created = Site.objects.get_or_create(domain=options['domain'], name=options['domain'])
|
||||
if created:
|
||||
self.stdout.write(self.style.SUCCESS('Created site {}'.format(site)))
|
||||
else:
|
||||
self.stdout.write(self.style.NOTICE('Site {} already exists'.format(site)))
|
||||
|
|
Loading…
Reference in New Issue