44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: qabot
|
|||
|
|
@Author：虎
|
|||
|
|
@file： users.py
|
|||
|
|
@date：2023/9/4 10:09
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
import hashlib
|
|||
|
|
|
|||
|
|
from django.db import models
|
|||
|
|
|
|||
|
|
__all__ = ["User", "password_encrypt"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def password_encrypt(raw_password):
|
|||
|
|
"""
|
|||
|
|
密码 md5加密
|
|||
|
|
:param raw_password: 密码
|
|||
|
|
:return: 加密后密码
|
|||
|
|
"""
|
|||
|
|
md5 = hashlib.md5() # 2，实例化md5() 方法
|
|||
|
|
md5.update(raw_password.encode()) # 3，对字符串的字节类型加密
|
|||
|
|
result = md5.hexdigest() # 4，加密
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
class User(models.Model):
|
|||
|
|
email = models.EmailField(unique=True, verbose_name="邮箱")
|
|||
|
|
username = models.CharField(max_length=150, unique=True, verbose_name="用户名")
|
|||
|
|
password = models.CharField(max_length=150, verbose_name="密码")
|
|||
|
|
role = models.CharField(max_length=150, verbose_name="角色")
|
|||
|
|
is_active = models.BooleanField(default=True)
|
|||
|
|
|
|||
|
|
USERNAME_FIELD = 'username'
|
|||
|
|
REQUIRED_FIELDS = []
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "user"
|
|||
|
|
|
|||
|
|
def set_password(self, raw_password):
|
|||
|
|
self.password = password_encrypt(raw_password)
|
|||
|
|
self._password = raw_password
|