Flask Login搭配pymongo操作MongoDB无法访问已登录用户记录

触发TypeError: 'NoneType' object is not subscriptable报错的核心原因有两个：

1. load_user回调中直接使用字符串类型的user_id查询MongoDB的_id字段，但MongoDB默认自动生成的_id是ObjectId类型，类型不匹配导致find_one查询无结果返回None，后续初始化User类时对空值取字段直接触发报错。
2. User类未做空值兼容，也没有将用户文档字段映射为实例属性，无法直接通过current_user.xxx的形式读取数据库存储的用户字段。
   另外现有代码存在一处配置错误：login_manager.login_view = 'auth.login'指向的蓝图路由不存在，未登录访问受保护页面时会触发路由不存在报错，需要一并修正。

• 第一步：引入ObjectId类型，修正用户加载逻辑
  在pymongo导入区域新增ObjectId引入，修改load_user方法，将字符串格式的用户ID转为MongoDB支持的ObjectId类型再查询，同时增加空值判断：

  from bson.objectid import ObjectId # 新增导入，和pymongo导入代码放在一起
  
  # 修正login_view配置，指向实际存在的首页登录路由
  login_manager.login_view = 'homepage'
  
  @login_manager.user_loader
  def load_user(user_id):
      # 处理ID格式不合法、查询无结果的场景，直接返回None告知Flask-Login用户未登录
      try:
          user_json = mycol.find_one({'_id': ObjectId(user_id)})
      except:
          return None
      if not user_json:
          return None
      return User(user_json)
  

• 第二步：重写User类，自动映射用户文档字段
  修改User类初始化逻辑，自动将数据库返回的用户文档所有字段挂载为实例属性，后续新增用户字段不需要重复修改类代码，直接通过属性名即可访问：

  class User(UserMixin):
      def __init__(self, user_json):
          self.user_json = user_json
          # 遍历用户文档所有字段，映射为实例属性
          for key, value in user_json.items():
              # 特殊处理_id字段，映射为id属性适配UserMixin默认逻辑
              if key == '_id':
                  self.id = str(value)
              else:
                  setattr(self, key, value)
  
      def get_id(self):
          return self.id
  

• 第三步：验证字段访问效果
  修复完成后重启服务，即可在任意登录后可访问的路由、模板中直接读取用户任意字段，示例：

  @app.route("/profile")
  @login_required
  def profile():
      # 直接通过属性访问即可，不需要额外查询
      return f"登录成功，用户名：{current_user.username}，邮箱：{current_user.email}，用户ID：{current_user.id}"
  

• 该适配方案兼容后续用户表新增字段，比如新增头像、手机号、注册时间等字段后，不需要修改User类代码，直接通过current_user.对应字段名即可读取。
• 记住me功能开启后，cookie中存储的是字符串格式的用户ID，每次请求触发load_user时都会自动转成ObjectId查询数据库，不会出现类型不匹配的问题。

内容的提问来源于stack exchange，提问作者David