# -*- coding: utf-8 -*- from django.db import models from django.conf import settings from django.core.validators import RegexValidator, MinLengthValidator from django.utils import timezone from django.utils.http import urlquote from django.utils.translation import ugettext_lazy as _ from django.core.mail import send_mail from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin from myauth import managers class DokUser(AbstractBaseUser, PermissionsMixin): """ (не)Самопальная модель пользователя - содрано у caktusgroup.com, также проблемы с добавлением юзера в админке решены при помощи stackoverflow, да хранит гугл имя его. """ username = models.CharField(_('first name'), max_length=30, blank=True, unique=True) email = models.EmailField(_('email address'), max_length=254, unique=True) first_name = models.CharField(_('first name'), max_length=30, blank=True) last_name = models.CharField(_('last name'), max_length=30, blank=True) is_staff = models.BooleanField(_('staff status'), default=False, help_text=_( 'Designates whether the user can log into this admin ' 'site.')) is_active = models.BooleanField(_('active'), default=True, help_text=_( 'Designates whether this user should be treated as ' 'active. Unselect this instead of deleting accounts.')) date_joined = models.DateTimeField(_('date joined'), default=timezone.now) profile = models.ForeignKey('customer.UserProfile', null=True, related_name='users') objects = managers.CustomUserManager() USERNAME_FIELD = 'username' REQUIRED_FIELDS = ['email'] class Meta: verbose_name = _('user') verbose_name_plural = _('users') def get_absolute_url(self): return f"/users/{urlquote(self.email)}/" def get_full_name(self): """ Returns the first_name plus the last_name, with a space in between. """ full_name = f'{self.first_name} {self.last_name}' return full_name.strip() def get_short_name(self): """ Returns the short name for the user. """ return self.first_name def email_user(self, subject, message, from_email=None): """ Sends an email to this User. """ send_mail(subject, message, from_email, [self.email]) class ConfirmEmail(models.Model): """ Подтверждение Email. """ user = models.OneToOneField(settings.AUTH_USER_MODEL, related_name='confirm_email', primary_key=True) is_confirmed = models.BooleanField(u'email подтвержден?', default=False) created_at = models.DateTimeField(u'Создан', auto_now_add=True) updated_at = models.DateTimeField(u'Изменен', auto_now=True) objects = managers.ConfirmEmailManager() class Meta: verbose_name = u'подтверждение email' verbose_name_plural = u'запросы подтверждения email' ordering = ['-created_at', ] def __str__(self): status = u'не подтвержден' if self.is_confirmed: status = u'подтвержден' return f'{self.user.email}, email {status}' class ResetKey(models.Model): """Ключ на восстановление пароля.""" user = models.OneToOneField( settings.AUTH_USER_MODEL, related_name='restore_key', primary_key=True ) key = models.CharField( u'Ключ доступа', max_length=40, db_index=True, validators=[ RegexValidator( regex='[0-9a-f]{40}', message='Введите значение длиной 40 символов, состоящее из цифр 0-9 и букв a-f.' ), MinLengthValidator(40), ] ) created_at = models.DateTimeField('создан', auto_now_add=True) updated_at = models.DateTimeField('изменен', auto_now=True) objects = managers.ResetKeyManager() class Meta: verbose_name = 'ключ восстановления пароля' verbose_name_plural = 'ключи восстановления паролей' ordering = ['-created_at', ] def __str__(self): return f'{self.user.email}, {self.key}'