Flask-Login 获取登录属性异常:如何解决 \'str\' object has no attribute \'id\' 和 current_user.name 为 None 的问题?

flask-login 获取登录属性异常:如何解决 'str' object has no attribute 'id' 和 current_user.name 为 none 的问题?

flask-login获取登录属性异常

在使用flask-login管理登录时,遇到使用数据库模型属性作为标识符却报错attributeerror: 'str' object has no attribute 'id'以及current_user.name返回的属性为none的问题。

这可能是由于在初始化user类时未正确获取数据库模型属性。下面提供修改后的代码:

from flask_login import UserMixin
from sqlalchemy import Column, Integer, String
from database import SessionLocal, engine, Base

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

在修改后的代码中:

  • 创建了一个数据库会话,以访问数据库。
  • 使用查询获取用户id对应的用户。
  • 如果找到用户,则将其转换为user类并返回。

通过这些修改,可以正确获取数据库模型属性,从而解决报错和属性返回none的问题。

以上就是Flask-Login 获取登录属性异常:如何解决 'str' object has no attribute 'id' 和 current_user.name 为 None 的问题?的详细内容,更多请关注硕下网其它相关文章!