使用“with 语句”关闭psyopcg2 中的连接的正确方法是什么？

Question

我想知道，使用 with 语句和 psyopcg2 关闭与 Postgres 数据库的连接的正确方法是什么。

import pandas as pd
import psycopg2
def create_df_from_postgres(params: dict,
                                   columns: str,
                                   tablename: str,
                                   ) -> pd.DataFrame:

    with psycopg2.connect(**params) as conn:
        data_sql = pd.read_sql_query(
          "SELECT " + columns + ", SUM(total)"
          " AS total FROM " + str(tablename),
          con=conn
          )
    # i need to close conection here:
        # conn.close()

    # or here:
    conn.close()
    return data_sql

这是处理连接的更好方法吗？

def get_ci_method_and_date(params: dict,
                           columns: str,
                           tablename: str,
                           ) -> pd.DataFrame:

    try:
        connection = psycopg2.connect(**params)
        data_sql = pd.read_sql_query('SELECT ' + columns +
                                     ' FROM ' + str(tablename),
                                     con=connection
                                     )
    finally:
        if(connection):
            connection.close()
    return data_sql

来自官方 psycopg 文档

警告与文件对象或其他资源不同，退出连接的 with 块不会关闭连接，而只会关闭与其关联的事务。如果你想确保连接在某个时间点后关闭，你仍然应该使用 try-catch 块：

conn = psycopg2.connect(DSN)
try:
    # connection usage
finally:
    conn.close()

Answer 1

关闭连接的正确方法：

\n

来自 psycopg 官方文档：

\n

\n

警告 与文件对象或其他资源不同，使用 block 退出连接\xe2\x80\x99 不会关闭连接，而只会关闭与\n关联的事务。如果您想确保连接在某个点之后关闭，您仍应使用 try-catch 块：

\n

\n

conn = psycopg2.connect(DSN)\ntry:\n    # connection usage\nfinally:\n    conn.close()\n

\n