import datetime

from django.conf import settings
from django.contrib.auth.models import (AbstractBaseUser, BaseUserManager,
                                        PermissionsMixin)
from django.db import models

# Create your models here.

class UserAccountManager(BaseUserManager):
    """ Manages the functions needed to create a user """
    def create_user(self, email, password=None, **extra_fields):
        """ Creates and saves a new user """
        if not email:
            raise ValueError("Users must have an email address")

        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)

        user.set_password(password)
        user.save(using=self._db)

        return user

    def create_superuser(self, email, password, **extra_fields):
        user = self.create_user(email, password, **extra_fields)

        user.is_superuser = True
        user.is_staff = True
        user.save(using=self._db)

        return user


class UserAccount(AbstractBaseUser, PermissionsMixin):
    """ General user account fields """

    username = models.CharField(max_length=2555, null=True, unique=True)
    email = models.EmailField(max_length=255, unique=True)
    first_name = models.CharField(max_length=255)
    last_name = models.CharField(max_length=255)
    avatar = models.ImageField(upload_to='avatar/', null=True, blank=True)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    get_newsletter = models.BooleanField(default=True, db_index=True)

    objects = UserAccountManager()

    USERNAME_FIELD = "email"
    REQUIRED_FIELDS = ["first_name", "last_name"]

    class Meta:
        ordering = ['-id']
        verbose_name = "User"
        verbose_name_plural = "Users"


    def get_short_name(self):
        return self.first_name

    def __str__(self):
        return self.email

    def get_or_create_userprofile(self):
        from profiles.models import UserProfile
        profile, _ = UserProfile.objects.get_or_create(user=self)
        return profile
    
    def get_avatar(self):
        profile = self.get_or_create_userprofile()
        return profile.avatar