Flask使用pymysql时数据库连接报错后正确关闭的方案咨询

你的现有实现逻辑是正确的，但是可以通过上下文管理器进一步简化，消除冗余的错误标记位和手动关闭逻辑，简化后的实现如下：

@app.route('/get-data/', methods=['GET'])
def get_data():
    try:
        # 连接和游标均使用上下文管理器，自动处理资源回收
        with pymysql.connect(db_url, db_username, db_password, db_name) as conn:
            with conn.cursor() as cursor:
                cursor.execute('SELECT a_column FROM a_table WHERE a_condition = 0')
                results = cursor.fetchall()
    except Exception as ex:
        logging.error(f'{ex}')
        return Response('Database error', 500)
    return jsonify(results)

优化后的代码具备以下优势：

• 无需手动维护db_error标记，异常捕获后可以直接返回，不会出现连接泄漏
• 无需手动编写finally块关闭游标和连接，所有资源回收逻辑由上下文管理器自动处理
• 代码结构更清晰，数据库操作核心逻辑和异常处理逻辑完全分离

针对你提出的疑问，相关说明如下：

• with上下文管理器的资源释放逻辑确实依赖驱动的接口实现，核心判定标准是对应类是否实现了Python的上下文管理器协议，也就是__enter__和__exit__魔术方法。只要__exit__方法中封装了资源关闭逻辑，无论代码块内是否抛出异常，__exit__都会被自动触发，完成资源回收。
• 主流Python数据库驱动（包括你举例的pymysql、pyodbc，以及psycopg2、sqlite3等常用驱动）都符合Python DB API 2.0规范，均对连接、游标类实现了标准的上下文管理器支持，可以保证无论操作是否报错，都会自动关闭游标和连接，不会出现资源泄漏。
• 若使用相对小众的数据库驱动，可以直接查看驱动的源码实现，确认连接和游标类的__exit__方法是否包含close()调用，只要存在对应逻辑就可以放心使用上下文管理器。

补充说明：部分驱动的连接上下文管理器还会自动处理事务：如果with块内没有抛出异常，退出时会自动执行commit()，如果抛出异常则自动执行rollback()，这个特性可以进一步简化事务相关的代码逻辑。

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