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

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_loader
def 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'错误。

以上就是使用 Flask-Login 登录时，如何正确获取数据库模型中用户的属性？的详细内容，更多请关注硕下网其它相关文章！