from django.db import models from django.conf import settings from common.models.baseModels import BaseModel from common.models.choiceModels import GenderChoices, CertificateCodeUseType from common.utils.codeManger import set_expire from common.utils.fileManager import DynamicUploadTo from django.contrib.postgres.fields import ArrayField from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin from common.models.choiceModels import NotificationType from django.utils import timezone from datetime import timedelta class UserManager(BaseUserManager): def create_user(self, email, password, **kwargs): user = self.model(email = email, **kwargs) if not password: user.set_unusable_password() else: user.set_password(password) user.save(using=self._db) return user def create_superuser(self, email=None, password=None, **extra_fields): superuser = self.create_user( email = email, password = password, ) superuser.is_staff = True superuser.is_superuser = True superuser.is_active = True superuser.save(using=self._db) return superuser class User(BaseModel, AbstractBaseUser, PermissionsMixin): email = models.EmailField(unique=True) realname = models.CharField(max_length=10, blank=True) # is_email_verified = models.BooleanField(default=False) is_plers_terms_of_service = models.BooleanField(default=False) is_terms_of_service_colio = models.BooleanField(default=False) is_consent_personal_info = models.BooleanField(default=False) is_consent_third_party_sharing = models.BooleanField(default=False) is_consent_marketing = models.BooleanField(default=False) phone = models.CharField(max_length=11, unique=True, blank=True) # is_phone_verified = models.BooleanField(default=False) nickname = models.CharField(max_length=20, unique=True, blank=True) gender = models.CharField(choices=GenderChoices.choices, max_length=1, blank=True) birth_date = models.CharField(max_length=10, blank=True) custom_url = models.CharField(max_length=20, default=None, null=True, blank=True, unique=True) is_custom_url = models.BooleanField(default=False) job_and_interests = ArrayField(models.CharField(max_length=20), default=list, blank=True) skills = ArrayField(models.CharField(max_length=20), default=list, blank=True) external_links = ArrayField(models.TextField(), default=list, blank=True) short_bio = models.CharField(max_length=100, blank=True) profile_image = models.ImageField(upload_to=DynamicUploadTo("user", "profile"), blank=True) banner_image = models.ImageField(upload_to=DynamicUploadTo("user", "banner"), blank=True) is_staff = models.BooleanField(default=False) is_active = models.BooleanField(default=True) objects = UserManager() USERNAME_FIELD = 'email' REQUIRED_FIELDS = [] def __str__(self): return self.nickname def delete(self, *args, **kwargs): if self.profile_image: self.profile_image.delete(save=False) if self.banner_image: self.banner_image.delete(save=False) super().delete(*args, **kwargs) class GithubToken(models.Model): user = models.OneToOneField(settings.AUTH_USER_MODEL, related_name='github_token', on_delete=models.CASCADE) access_token = models.CharField(max_length=255, blank=True, null=True) refresh_token = models.CharField(max_length=255, blank=True, null=True) scope = models.CharField(max_length=255, blank=True) token_type = models.CharField(max_length=32, default='bearer') expires_at = models.DateTimeField(blank=True, null=True) def is_expired(self) -> bool: return self.expires_at and timezone.now() >= self.expires_at