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)))