Python SQLite3连续执行两次查询触发NoneType异常问题排查

根本原因 #

你的问题核心是Flask CLI命令未绑定应用上下文，导致连续调用get_db时，第二次调用无法获取到有效的应用上下文，最终get_db返回None，触发TypeError。

具体来说：

• 你的dummy_command是一个Flask CLI命令，但没有使用@with_appcontext装饰器。Flask CLI命令默认不会自动推送应用上下文，第一次调用query_db时可能通过内部隐式逻辑临时创建了上下文，但操作完成后上下文被立即销毁。
• 第二次调用query_db时，_app_ctx_stack.top已经是None，而你的get_db函数未处理这种情况，最终返回None，导致db[shard]抛出TypeError: 'NoneType' object has no attribute '__getitem__'。

另外，你的get_user_id函数存在逻辑缺陷：当前仅返回第二个分片的查询结果，完全忽略了第一个分片的结果，不符合分片查询的预期。

解决方案 #

1. 给CLI命令添加应用上下文装饰器

修改你的CLI命令，添加@with_appcontext装饰器，确保整个命令执行期间应用上下文保持活跃：

from flask.cli import with_appcontext

@click.command()
@with_appcontext  # 关键：绑定应用上下文
def dummy_command():
    dummy_db()

2. 增强get_db函数的健壮性

为了避免因上下文缺失导致的错误，可以在get_db中主动检查并推送应用上下文：

def get_db():
    """Opens a new database connection if there is none yet for the current application context. """
    top = _app_ctx_stack.top
    
    # 如果没有应用上下文，手动推送一个
    if top is None:
        ctx = app.app_context()
        ctx.push()
        top = _app_ctx_stack.top
    
    if not hasattr(top, 'sqlite_db'):
        top.sqlite_db = [
            sqlite3.connect(DATABASE_1, detect_types=sqlite3.PARSE_DECLTYPES),
            sqlite3.connect(DATABASE_2, detect_types=sqlite3.PARSE_DECLTYPES),
            sqlite3.connect(DATABASE_3, detect_types=sqlite3.PARSE_DECLTYPES)
        ]
        # 简化重复的row_factory设置
        for conn in top.sqlite_db:
            conn.row_factory = sqlite3.Row
    return top.sqlite_db

3. 修复get_user_id的查询逻辑

当前函数只返回第二个分片的结果，应该依次检查所有分片的查询结果：

def get_user_id(username):
    """Convenience method to look up the id for a username."""
    # 先查询分片1
    rv = query_db('select user_id from user where username = ?', 1, [username], one=True)
    if rv is not None:
        return rv[0]
    
    # 分片1无结果，查询分片2
    rv = query_db('select user_id from user where username = ?', 2, [username], one=True)
    if rv is not None:
        return rv[0]
    
    # 所有分片都无结果，返回None
    return None

验证 #

完成上述修改后，再次运行dummy_command，连续调用query_db时就能正常获取数据库连接列表，不会再出现TypeError，同时也能正确从所有分片中查询用户ID。

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