PHP前端开发

使用 Flask-Login 登录时,如何正确获取数据库模型中用户的属性?

百变鹏仔 5个月前 (01-14) #Python
文章标签 如何正确

flask-login登录属性获取问题

在使用flask-login登录管理时,若使用数据库模型中的属性作为标识符,则应按照以下方式编写登录回调函数:

class user(usermixin, base):    # 定义数据库模型    __tablename__ = "users"    id = column(integer, primary_key=true, index=true)    name = column(string, unique=true, index=true)    password = column(string)    def __init__(self, user_id):        self.id = user_id.id        self.name = user_id.name    def get_id(self):        return str(self.id)@super_login.user_loaderdef load_user(user_id):    # 创建数据库会话    db = sessionlocal()    user = db.query(user).get(user_id)    if user:        return user(user)    return none

登录过程如下:

user_obj = model_login(name = username2)db_user = db.query(User).get(user_obj.id)user = User(db_user)login_user(user,remember = True)

这样,就可以通过user.id获取用户id,通过user.name获取用户名。

报错提示分析:

原本的代码中,传入user类的是数据库模型对象,而数据库模型对象没有id属性,因此提示attributeerror: 'str' object has no attribute 'id'错误。