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

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

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 2

Connections具有PEP-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:
    with conn.cursor() as crs:
        do_stuff
        # crs.close() will automatically be called when Python leaves the inner `with` statement
    # conn.commit() will automatically be called when Python leaves the outer `with` statement

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

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

连接会在删除时自动关闭（通常在超出范围时），因此您通常不需要调用[conn.close()]，但如果愿意，可以明确关闭连接。

同样for cursors（我的重点）：

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

Answer 3

您可以尝试关闭池，默认情况下启用池。有关详细信息，请参阅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 4

根据pyodbc文档，与SQL服务器are not closed by default.的连接有些数据库驱动程序在调用close（）时不会关闭连接，以便保存到服务器的往返。

要在致电 close（）时关闭连接，您应将池设置为False：

writing_type

Answer 5

您可以如下定义数据库类。另外，根据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 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

Python数据库连接关闭

6 个答案: