更新 User.py

master
wangsiyuan 2023-12-28 16:34:35 +08:00
parent eaea8e1f0f
commit ca8cbadccc
1 changed files with 6 additions and 5 deletions

View File

@ -1,14 +1,15 @@
import bcrypt import bcrypt
class User: class User:
def __init__(self, nickname, phone_number, password, identity, status): def __init__(self, name, number, password, status):
self.nickname = nickname # 用户昵称 self.name = name # 用户昵称
self.phone_number = phone_number # 手机号 self.number = number # 手机号
self.password = self.hash_password(password) # 哈希密码 self.password = self.hash_password(password) # 哈希密码
self.identity = identity # 身份(老师或学生)
self.status = status # 状态(是否可用) self.status = status # 状态(是否可用)
def hash_password(self, password): def hash_password(self, password):
return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8') return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
def __str__(self): def __str__(self):
return f"User({self.nickname}, {self.phone_number}, {'Teacher' if self.identity == 'teacher' else 'Student'}, {'Active' if self.status else 'Inactive'})" return f"User({self.name}, {self.number}, {'Active' if self.status else 'Inactive'})"