Python 数据库连接 关闭

Question

使用下面的代码可以打开连接，我该如何关闭？

import pyodbc
conn = pyodbc.connect('DRIVER=MySQL ODBC 5.1 driver;SERVER=localhost;DATABASE=spt;UID=who;PWD=testest') 

csr = conn.cursor()  
csr.close()
del csr

Answer 1

连接具有 PE​​P-249（Python 数据库 API 规范 v2.0）中指定的 close 方法：

import pyodbc
conn = pyodbc.connect('DRIVER=MySQL ODBC 5.1 driver;SERVER=localhost;DATABASE=spt;UID=who;PWD=testest') 

csr = conn.cursor()  
csr.close()
conn.close()     #<--- Close the connection

---

由于 pyodbc connection 和 cursor 都是上下文管理器，现在将其写成这样会更方便（也更可取）：

import pyodbc
conn = pyodbc.connect('DRIVER=MySQL ODBC 5.1 driver;SERVER=localhost;DATABASE=spt;UID=who;PWD=testest') 
with conn:
    crs = conn.cursor()
    do_stuff
    # conn.commit() will automatically be called when Python leaves the outer `with` statement
    # Neither crs.close() nor conn.close() will be called upon leaving the `with` statement!! 

请参阅https://github.com/mkleehammer/pyodbc/issues/43，了解为什么不调用 conn.close()。

请注意，与原始代码不同，这会导致调用conn.commit()。使用外部 with 语句来控制您希望何时调用 commit。

---

另请注意，无论您是否使用with 语句，根据the docs，

连接在被删除时会自动关闭（通常是在超出范围时），因此您通常不需要调用 [conn.close()]，但您可以根据需要显式关闭连接。

同样for cursors（我的重点）：

游标在被删除时会自动关闭（通常是在超出范围时），因此通常不需要调用 [csr.close()]。

Answer 2

您可以如下定义数据库类。此外，正如andrewf 建议的那样，使用上下文管理器进行游标访问。我将其定义为成员函数。 这样，它可以在应用程序代码的多个事务中保持连接打开，并节省到服务器的不必要的重新连接。

import pyodbc

class MS_DB():
    """ Collection of helper methods to query the MS SQL Server database.
    """

    def __init__(self, username, password, host, port=1433, initial_db='dev_db'):
        self.username = username
        self._password = password
        self.host = host
        self.port = str(port)
        self.db = initial_db
        conn_str = 'DRIVER=DRIVER=ODBC Driver 13 for SQL Server;SERVER='+ \
                    self.host + ';PORT='+ self.port +';DATABASE='+ \
                    self.db +';UID='+ self.username +';PWD='+ \ 
                    self._password +';'
        print('Connected to DB:', conn_str)
        self._connection = pyodbc.connect(conn_str)        
        pyodbc.pooling = False

    def __repr__(self):
        return f"MS-SQLServer('{self.username}', <password hidden>, '{self.host}', '{self.port}', '{self.db}')"

    def __str__(self):
        return f"MS-SQLServer Module for STP on {self.host}"

    def __del__(self):
        self._connection.close()
        print("Connection closed.")

    @contextmanager
    def cursor(self, commit: bool = False):
        """
        A context manager style of using a DB cursor for database operations. 
        This function should be used for any database queries or operations that 
        need to be done. 

        :param commit:
        A boolean value that says whether to commit any database changes to the database. Defaults to False.
        :type commit: bool
        """
        cursor = self._connection.cursor()
        try:
            yield cursor
        except pyodbc.DatabaseError as err:
            print("DatabaseError {} ".format(err))
            cursor.rollback()
            raise err
        else:
            if commit:
                cursor.commit()
        finally:
            cursor.close()

ms_db = MS_DB(username='my_user', password='my_secret', host='hostname')
with ms_db.cursor() as cursor:
        cursor.execute("SELECT @@version;")
        print(cur.fetchall())

Answer 3

您可以将整个连接包装在上下文管理器中，如下所示：

from contextlib import contextmanager
import pyodbc
import sys

@contextmanager
def open_db_connection(connection_string, commit=False):
    connection = pyodbc.connect(connection_string)
    cursor = connection.cursor()
    try:
        yield cursor
    except pyodbc.DatabaseError as err:
        error, = err.args
        sys.stderr.write(error.message)
        cursor.execute("ROLLBACK")
        raise err
    else:
        if commit:
            cursor.execute("COMMIT")
        else:
            cursor.execute("ROLLBACK")
    finally:
        connection.close()

然后在需要数据库连接的地方执行类似的操作：

with open_db_connection("...") as cursor:
    # Your code here

当您离开 with 块时，连接将关闭。如果发生异常或您没有使用with open_db_connection("...", commit=True) 打开块，这也会回滚事务。

Answer 4

根据 pyodbc 文档，与 SQL 服务器的连接 are not closed by default. 一些数据库驱动程序在调用 close() 时不会关闭连接以节省到服务器的往返行程。

要在调用 close() 时关闭连接，您应该将 pooling 设置为 False：

import pyodbc

pyodbc.pooling = False

Answer 5

您可以尝试关闭默认启用的池化。有关更多信息，请参阅this 讨论。

import pyodbc
pyodbc.pooling = False
conn = pyodbc.connect('DRIVER=MySQL ODBC 5.1 driver;SERVER=localhost;DATABASE=spt;UID=who;PWD=testest') 

csr = conn.cursor()  
csr.close()
del csr

Answer 6

处理连接的最常用方法，如果该语言没有像 .NET 中的 Using 这样的自关闭结构，那么您应该使用 try -> finally 来关闭对象。 pyodbc 可能确实有某种形式的自动关闭，但这是我做的代码以防万一：

conn = cursor = None
try:
    conn = pyodbc.connect('DRIVER=MySQL ODBC 5.1 driver;SERVER=localhost;DATABASE=spt;UID=who;PWD=testest') 

    cursor = conn.cursor()  

    # ... do stuff ...

finally:
    try: cursor.close()
    except: pass
    try: conn.close()
    except: pass