mirror of
https://github.com/Balshgit/different
synced 2025-09-12 01:20:43 +03:00
66 lines
2.6 KiB
Python
66 lines
2.6 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import AbstractUser
|
|
from .managers import CustomUserManager
|
|
|
|
|
|
class CustomUser(AbstractUser):
|
|
|
|
mobile = models.CharField(max_length=15, null=True, blank=True, unique=True,
|
|
help_text='Users mobile phone')
|
|
verification_code = models.CharField(max_length=10, unique=True, null=True, blank=True,
|
|
help_text='Verification code for bot account')
|
|
user_created = models.DateField(editable=False, auto_now_add=True, verbose_name='User created',
|
|
help_text='Date when user has been created')
|
|
activation_token = models.CharField(max_length=20, null=True, blank=True,
|
|
help_text='Activation token for any user')
|
|
email = models.EmailField(max_length=30, unique=True, blank=False, null=True, help_text='User email')
|
|
USERNAME_FIELD = 'username'
|
|
REQUIRED_FIELDS = ['email']
|
|
objects = CustomUserManager()
|
|
|
|
def __str__(self):
|
|
return self.username
|
|
|
|
|
|
# # ----- ToDO: Enable this to email verification --------
|
|
#
|
|
# from django.dispatch import receiver
|
|
# from django.conf import settings
|
|
# from .tasks import mail_send
|
|
# from server.settings.components.logging import main_logger
|
|
#
|
|
#
|
|
# def user_tokens() -> dict:
|
|
#
|
|
# tokens_dict = dict()
|
|
#
|
|
# def generate_token(token_length: int) -> str:
|
|
# from random import choice
|
|
# token_chars = 'QWERTYUIOPASDFGHJKLZXCVBNMqwertyuiopasdfghjklzxcvbnm1234567890'
|
|
# generated_token = ''
|
|
# for i in range(token_length):
|
|
# generated_token += choice(token_chars)
|
|
# return generated_token
|
|
#
|
|
# tokens_dict['activation_token'] = generate_token(token_length=20)
|
|
#
|
|
# return tokens_dict
|
|
#
|
|
#
|
|
# @receiver(models.signals.post_save, sender=CustomUser)
|
|
# def post_save_user_signal_handler(sender, instance, created, **kwargs):
|
|
#
|
|
# if created and instance.username != 'admin':
|
|
# instance.activation_token = user_tokens()['activation_token']
|
|
# instance.save()
|
|
# try:
|
|
# user = CustomUser.objects.get(username=instance.username)
|
|
# email = instance.email
|
|
# subject = 'Welcome to book bot administration'
|
|
# username = f'{instance.first_name} {instance.last_name}'
|
|
# text = f'https://{settings.DOMAIN_NAME}/accounts/complete_registration/{user.activation_token}'
|
|
#
|
|
# mail_send(to_email=email, subject=subject, username=username, text_content=text)
|
|
# except Exception as e:
|
|
# main_logger.error(f'Email not send to user {instance.username}. Reason: {e}')
|