diff --git a/models/User.py b/models/User.py index 0888d7e..30344df 100644 --- a/models/User.py +++ b/models/User.py @@ -1,14 +1,15 @@ import bcrypt + class User: - def __init__(self, nickname, phone_number, password, identity, status): - self.nickname = nickname # 用户昵称 - self.phone_number = phone_number # 手机号 + def __init__(self, name, number, password, status): + self.name = name # 用户昵称 + self.number = number # 手机号 self.password = self.hash_password(password) # 哈希密码 - self.identity = identity # 身份(老师或学生) self.status = status # 状态(是否可用) def hash_password(self, password): return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8') + def __str__(self): - return f"User({self.nickname}, {self.phone_number}, {'Teacher' if self.identity == 'teacher' else 'Student'}, {'Active' if self.status else 'Inactive'})" \ No newline at end of file + return f"User({self.name}, {self.number}, {'Active' if self.status else 'Inactive'})"