createsuperuser_if_none_exists.py
25 lines
| 860 B
| text/x-python
|
PythonLexer
r376 | import os | ||
r366 | from django.core.management.base import BaseCommand | ||
from django.contrib.auth import get_user_model | |||
class Command(BaseCommand): | |||
""" | |||
Create a superuser if none exist | |||
Example: | |||
manage.py createsuperuser_if_none_exists --user=admin123 --password=admin123 --email=admin123@igp.gob.pe | |||
""" | |||
def handle(self, *args, **options): | |||
User = get_user_model() | |||
r376 | username = os.environ.get('SIRM_USER', 'admin') | ||
password = os.environ.get('SIRM_PASSWORD', 'soporte') | |||
email = os.environ.get('SIRM_EMAIL', 'admin@igp.gob.pe') | |||
r366 | |||
if User.objects.exists(): | |||
self.stdout.write(f'Local user "{username}" currently exists') | |||
return | |||
User.objects.create_superuser(username=username, password=password, email=email) | |||
self.stdout.write(f'Local user "{username}" was created') |