from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin, BaseUserManager
from django.db import models

from dockerhub.constants import TYPE_CHOICES, AccountTypes


class DockerUserManager(BaseUserManager):
    def create_user(self, username, email, password=None):
        user = self.model.objects.create(username=username, email=email)
        return user

    def create_active_user(self, username, email, password=None):
        user = self.model.objects.create(username=username, email=email, is_active=True)
        return user


class DockerUser(AbstractBaseUser, PermissionsMixin):
    username = models.CharField(max_length=30, unique=True)
    is_active = models.BooleanField(default=False)
    type = models.CharField(max_length=15, choices=TYPE_CHOICES, default=AccountTypes.USER)
    email = models.EmailField()

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = ['email', ]

    objects = DockerUserManager()