|
|
import os
|
|
|
from django.core.management.base import BaseCommand
|
|
|
from django.contrib.auth.models import User, Group
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
"""
|
|
|
Create a superuser and user if none exist
|
|
|
Example:
|
|
|
manage.py create_users
|
|
|
"""
|
|
|
|
|
|
help = "Create a superuser and user if none exist"
|
|
|
|
|
|
def handle(self, *args, **options):
|
|
|
|
|
|
users = {
|
|
|
'Superuser': {
|
|
|
'username': os.environ.get('SIRM_SUPER_USER', 'superuser'),
|
|
|
'password': os.environ.get('SIRM_SUPER_PASSWORD', 'SuperuseROJ'),
|
|
|
'email' : os.environ.get('SIRM_SUPER_EMAIL', 'superuser@igp.gob.pe')},
|
|
|
'Developer': {
|
|
|
'username': os.environ.get('SIRM_DEV_USER', 'developer'),
|
|
|
'password': os.environ.get('SIRM_DEV_PASSWORD', 'DevelopeROJ'),
|
|
|
'email' : os.environ.get('SIRM_DEV_EMAIL', 'developer@igp.gob.pe')},
|
|
|
'Operator': {
|
|
|
'username': os.environ.get('SIRM_USER', 'operator'),
|
|
|
'password': os.environ.get('SIRM_PASSWORD', 'OperatoROJ'),
|
|
|
'email' : os.environ.get('SIRM_EMAIL', 'operator@igp.gob.pe')}
|
|
|
}
|
|
|
|
|
|
for key, value in users.items():
|
|
|
if key == 'Superuser':
|
|
|
if User.objects.filter(is_superuser=True):
|
|
|
self.stdout.write(f'Local {key} currently exists')
|
|
|
else:
|
|
|
user, created = User.objects.get_or_create(username=value["username"], first_name=value["username"], email=value["email"], is_superuser = True, is_staff = True)
|
|
|
if created:
|
|
|
user.set_password(value["password"])
|
|
|
user.save()
|
|
|
self.stdout.write(f'Local {key} "{value["username"]}" was created')
|
|
|
else:
|
|
|
self.stdout.write(f'Unable to create this local superuser: "superuser already exists"')
|
|
|
else:
|
|
|
if User.objects.filter(groups__name=key):
|
|
|
if User.objects.filter(groups__name=key, is_superuser=True):
|
|
|
self.stdout.write(f"{key} group must not have a superuser, remove superusers and create a new user")
|
|
|
else:
|
|
|
self.stdout.write(f"Local {key} currently exists")
|
|
|
else:
|
|
|
user, created = User.objects.get_or_create(username=value["username"], first_name=value["username"], email=value["email"])
|
|
|
if created:
|
|
|
user.set_password(value["password"])
|
|
|
user.save()
|
|
|
self.stdout.write(f'Local {key} "{value["username"]}" was created')
|
|
|
|
|
|
group = Group.objects.get(name=key)
|
|
|
group.user_set.add(user)
|
|
|
self.stdout.write(f'Local {key} "{value["username"]}" was added to {key} group')
|
|
|
else:
|
|
|
self.stdout.write(f'Unable to create and join to {key} group this local user: "user already exists"')
|
