import os

from django.core.management.base import BaseCommand
from django.core import management
from django.contrib.auth import get_user_model
from django.db import connection

SUPERUSER_USERNAME = 'admin'


class Command(BaseCommand):

    def handle(self, *args, **options):
        management.call_command('migrate', verbosity=1)

        if not get_user_model().objects.filter(username=SUPERUSER_USERNAME).exists():
            os.environ['DJANGO_SUPERUSER_PASSWORD'] = 'admin'
            management.call_command('createsuperuser', '--username', SUPERUSER_USERNAME, '--email', 'admin@admin.local', '--noinput', verbosity=0)