我可以将 pymysql.connect() 与 "with" 语句一起使用吗？

Question

下面是 pymysql 中的示例：

conn = pymysql.connect(...)
with conn.cursor() as cursor:
    cursor.execute(...)
    ...
conn.close()

我可以改用以下方法吗，或者这会留下挥之不去的联系吗？（执行成功）

import pymysql
with pymysql.connect(...) as cursor:
    cursor.execute('show tables')

(python 3、最新的pymysql)

Answer 1

这看起来不安全，如果您查看 here，__enter__ 和 __exit__ 函数是在 with 子句中调用的。对于 pymysql 连接，它们看起来像这样：

def __enter__(self):
    """Context manager that returns a Cursor"""
    return self.cursor()

def __exit__(self, exc, value, traceback):
    """On successful exit, commit. On exception, rollback"""
    if exc:
        self.rollback()
    else:
        self.commit()

所以看起来 exit 子句不会关闭连接，这意味着它会挥之不去。我不确定他们为什么这样做。不过，您可以制作自己的包装器来执行此操作。

您可以通过创建多个游标来回收连接(the source for cursors is here)游标方法如下所示：

def __enter__(self):
    return self

def __exit__(self, *exc_info):
    del exc_info
    self.close()

所以他们确实把自己关起来了。您可以创建单个连接并在 with 子句中将其与多个游标一起重用。

如果您想在 with 子句后面隐藏关闭连接的逻辑，例如一个上下文管理器，一个简单的方法是这样的：

from contextlib import contextmanager
import pymysql


@contextmanager
def get_connection(*args, **kwargs):
    connection = pymysql.connect(*args, **kwargs)
    try:
        yield connection
    finally:
        connection.close()

然后您可以像这样使用该上下文管理器：

with get_connection(...) as con:
    with con.cursor() as cursor:
        cursor.execute(...)

Answer 2

正如所指出的那样，游标会自行处理，但是几天前连接对上下文管理器的所有支持都被完全删除了，所以现在唯一的选择是写你的：

https://github.com/PyMySQL/PyMySQL/pull/763

https://github.com/PyMySQL/PyMySQL/issues/446

Answer 3

作为替代方案，因为我想支持连接的上下文管理器模式，所以我用猴子补丁实现了它。不是最好的方法，但它是某种东西。

import pymysql


MONKEYPATCH_PYMYSQL_CONNECTION = True


def monkeypatch_pymysql_connection():
    Connection = pymysql.connections.Connection

    def enter_patch(self):
        return self

    def exit_patch(self, exc, value, traceback):
        try:
            self.rollback()  # Implicit rollback when connection closed per PEP-249
        finally:
            self.close()

    Connection.__enter__ = enter_patch
    Connection.__exit__ = exit_patch


if MONKEYPATCH_PYMYSQL_CONNECTION:
    monkeypatch_pymysql_connection()
    MONKEYPATCH_PYMYSQL_CONNECTION = False  # Prevent patching more than once

这种方法适用于我的用例。我更愿意在 Connection class 中使用 __enter__ 和 __exit__ 方法。然而，这种方法在开发人员于 2018 年底解决该问题时被拒绝了。

我可以将 pymysql.connect() 与 "with" 语句一起使用吗？

Can I use pymysql.connect() with "with" statement?

python

with-statement

pymysql