能否从Python上下文管理器返回两个变量？以MySQL场景为例

你的思路完全可行，上下文管理器的__enter__方法可以返回一个元组，再通过元组解构赋值在with语句中同时获取多个变量。不过你当前的使用语法有误，同时可以优化异常处理逻辑，以下是具体修正方案：

修正后的代码示例 #

上下文管理器类 #

class DatabaseConnect:
    def __init__(self, name):
        self.name = name
        self.cnx = None
        self.cursor = None

    def __enter__(self):
        # 假设 databaseConnect 是你实际创建MySQL连接的函数
        self.cnx = databaseConnect(self.name)
        self.cursor = self.cnx.cursor()
        # 返回包含连接和游标的元组
        return self.cnx, self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 确保游标和连接即使创建失败也不会报错
        if self.cursor:
            self.cursor.close()
        if self.cnx:
            # 根据是否发生异常决定提交或回滚
            if exc_type is None:
                self.cnx.commit()
            else:
                self.cnx.rollback()
            self.cnx.close()

正确使用方式 #

# 用元组解构获取两个变量
with DatabaseConnect("QueryableCases") as (cnx, cursor):
    # 执行数据库操作
    cursor.execute("SELECT * FROM some_table")
    results = cursor.fetchall()
    # ...其他业务逻辑

也可以用更简洁的写法（Python 3.1及以上版本支持）：

with DatabaseConnect("QueryableCases") as cnx, cursor:
    cursor.execute("INSERT INTO some_table VALUES (%s)", ("test_data",))

关键说明 #

• __enter__返回元组后，with语句的as部分可直接解构为多个变量，这是Python原生的元组解包特性。
• 优化了__exit__的异常处理：只有当代码块无异常时才提交事务，出现异常则回滚，避免错误数据被持久化。
• 增加了非空判断，防止连接或游标创建失败时调用close()方法抛出错误。

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