From 87c8da1a9e386565413cdf4af161291e4922f2bf Mon Sep 17 00:00:00 2001 From: Julien Palard Date: Thu, 20 Oct 2022 23:23:15 +0200 Subject: [PATCH] addist: Allow to be called with an exisitng site. --- ponyconf/management/commands/addsite.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ponyconf/management/commands/addsite.py b/ponyconf/management/commands/addsite.py index 17c0a24..070443c 100644 --- a/ponyconf/management/commands/addsite.py +++ b/ponyconf/management/commands/addsite.py @@ -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)))