Django项目知识点（二）

# 自定义用户模型
# 这里的user是app，User是模型 不是固定的
AUTH_USER_MODEL = 'user.User'

from  django.contrib.auth.models import AbstractUser,UserManager as _UserManager
# Create your models here.
class UserManager(_UserManager):
    '''
    自定义usermanage 修改时使用python manage createsupermanage命令时，我可以不提供email字段
    '''
    def create_superuser(self,username,password,email= None,**extra_fields):
        return  super().create_superuser(username=username,password=password,email=email,**extra_fields)

class User(AbstractUser):
    """
    add mobile, email_active fields to Django user model.
    """
    mobile = models.CharField('手机号', max_length=11, unique=True, help_text='手机号', error_messages={'unique': '此手机号码已注册'})
    email_active = models.BooleanField('邮箱状态', default=False)
    class Meta:
        db_table = 'tb_user'    # 指定数据库表名
        verbose_name = '用户'     # 在admin站点中显示名称
        verbose_name_plural = verbose_name  # 显示复数
    def __str__(self):
        return self.username
    # A list of the field names that will be prompted for
    # when create a user via createsuperuser management command.
    # 必需字段
    REQUIRED_FIELDS = ['mobile']
    # specify manager
    objects = UserManager()

from django_redis import get_redis_connection
class RegisterForm(forms.Form):
    """
    用户注册表单
    """
    username = forms.CharField(label='用户名', max_length=20, min_length=5, error_messages={
        'max_length': '用户名长度要小于20',
        'min_length': '用户名长度要大于5',
        'required': '用户名不能为空',
    })
    password = forms.CharField(label='密码', max_length=20, min_length=6, error_messages={
        'max_length': '密码长度要小于20',
        'min_length': '密码长度要大于5',
        'required': '密码不能为空',
    })
    password_repeat = forms.CharField(label='确认密码', max_length=20, min_length=6, error_messages={
        'max_length': '密码长度要小于20',
        'min_length': '密码长度要大于5',
        'required': '密码不能为空',
    })
    mobile = forms.CharField(label='手机号码', max_length=11, min_length=11, validators=[mobile_validator, ], error_messages={
        'max_length': '手机号码长度不正确',
        'min_length': '手机号码长度不正确',
        'required': '手机号码不能为空',
    })
    sms_code = forms.CharField(label='短信验证码', max_length=constants.SMS_CODE_LENGTH, min_length=constants.SMS_CODE_LENGTH, error_messages={
        'max_length': '短信验证码长度不正确',
        'min_length': '短信验证码长度不正确',
        'required': '短信验证码不能为空',
    })

    def clean_username(self):
        """
        校验用户名
        :return:
        """
        username = self.cleaned_data.get('username')

        if User.objects.filter(username=username).exists():
            raise forms.ValidationError('用户名已存在！')

        return username

    def clean_mobile(self):
        """
        校验手机号码
        :return:
        """
        mobile = self.cleaned_data.get('mobile')

        if User.objects.filter(mobile=mobile).exists():
            raise forms.ValidationError('手机号码已注册！')

        return mobile

    def clean(self):
        """
        联合校验
        :return:
        """
        clean_data = super().clean()
        # 校验密码是否一致
        password = clean_data.get('password')
        password_repeat = clean_data.get('password_repeat')

        if password != password_repeat:
            raise forms.ValidationError('两次密码不一致！')

        # 校验短信验证码
        sms_code = clean_data.get('sms_code')
        moblie = clean_data.get('mobile')

        redis_conn = get_redis_connection(alias='verify_code')
        real_code = redis_conn.get('sms_text_{}'.format(moblie))
        if (not real_code) or (real_code.decode('utf-8') != sms_code):
            raise forms.ValidationError('短信验证码错误!')