"""
This command create superuser object.
Run with>
python manage.py create_superuser users
"""

from django.core.management.base import AppCommand
from users.models import User


class Command(AppCommand):

    def handle(self, *args, **options):
        if User.objects.filter(email='admin@admin.com').count() < 1:
            user = User(email='admin@admin.com', name='Administrator', age=18,
                     is_staff=True, is_superuser=True, is_active=True)
            user.set_password('admin')
            user.save()
            print('User admin created')