89 lines
3.8 KiB
Python
89 lines
3.8 KiB
Python
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(BaseModel):
|
|
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 |