Надо ли закрывать соединение psycopg2 когда программа заканчивается кодом Process finished with exit code -1?

Question

[Bekkazy]

Я новичок в Python и тут в голове возник вопрос "А надо ли закрывать соединение, когда сам останавливаю программу в Pycharm?" (Программа работает в бесконечном цикле и останавливать приходится вручную). Потому что сейчас очень много тестирую, и не получится ли так, что будут висеть большое количество не закрытых соединений в БД?

Answer 1

[deepblack]

Используйте менеджер контекста:

from contextlib import closing
with closing(psycopg2.connect(...)) as conn:
    with conn.cursor() as cursor:
        cursor.execute('SELECT * FROM users LIMIT 5')
        for row in cursor:
            print(row)

UPD:

Могу порекомендовать ознакомиться с этим

Пример использования (выдернул из проекта)

from psycopg2.pool import ThreadedConnectionPool
from psycopg2.extras import RealDictCursor
from contextlib import contextmanager
from urllib.parse import urlparse

DATABASE_URL = 'postgresql://postgres:postgres@localhost/postgres'
url = urlparse(DATABASE_URL)

pool = ThreadedConnectionPool(1, 50,
                              database=url.path[1:],
                              user=url.username,
                              password=url.password,
                              host=url.hostname,
                              port=url.port)


@contextmanager
def get_db_connection():
    try:
        connection = pool.getconn()
        yield connection
    finally:
        pool.putconn(connection)


@contextmanager
def get_db_cursor(commit=False):
    with get_db_connection() as connection:
        cursor = connection.cursor(cursor_factory=RealDictCursor)
        try:
            yield cursor
            if commit:
                connection.commit()
        finally:
            cursor.close()


# Пример использования
with get_db_cursor() as cursor: 
    cursor.execute("SELECT city, temp_lo, temp_hi, date FROM weather;") 
    data = cursor.fetchone()
    
with get_db_cursor(commit=True) as cursor:
    cursor.callproc("SOME_PROC_NAME")

Comments

[Bekkazy]

Как можно интегрировать менеджер контекста в данный класс?
Ей надо передавать подключение?

psycopg2.connect(...)

import psycopg2
from psycopg2 import pool
import datetime


class PotgressConn:
    def __init__(self, config=None):

        self.conf = config

        pg_user = self.conf.get_config("Postgres.user")
        pg_pass = self.conf.get_config("Postgres.password")
        pg_host = self.conf.get_config("Postgres.host")
        pg_port = self.conf.get_config("Postgres.port")
        pg_db = self.conf.get_config("Postgres.database")

        self.__pool = psycopg2.pool.SimpleConnectionPool(1, 20, user=pg_user,
                                                         password=pg_pass,
                                                         host=pg_host,
                                                         port=pg_port,
                                                         database=pg_db)
        if self.__pool:
            print("PostgreSQL: Connection pool created successfully")

    def get_conn(self):
        if self.__pool:
            return self.__pool.getconn()

    def disconnect(self):
        if self.__pool:
            self.__pool.closeall
        print("PostgreSQL: connection pool is closed")

    def execute(self, query, transaction_type = 'select', fetch_type='all'):
        try:
            records = None
            ps_connection = self.__pool.getconn()
            if ps_connection:
                ps_cursor = ps_connection.cursor()
                ps_cursor.execute(query)

                if transaction_type in('insert', 'delete'):
                    ps_connection.commit()
                else:
                    if fetch_type == "all":
                        records = ps_cursor.fetchall()
                    elif fetch_type == "many":
                        records = ps_cursor.fetchmany()
                    else:
                        print(ps_cursor.statusmessage)
                        records = ps_cursor.fetchone()

                ps_cursor.close()
                return records
        except Exception as err:
            raise err
        finally:
            self.disconnect()

    def insert_one(self, table_name, fields: list, values: list):
        if self.__pool:
            colums = ""
            for i in fields:
                colums = colums + i + ", "
            colums = colums[:-2]

            val = ""
            for j in values:
                val = val + "'" + str(j) + "', "
            val = val[:-2]
            query = "INSERT INTO {} ({}) VALUES({});".format(table_name, colums, val)
            res = self.execute(query, 'insert', 'one')
        return res

[deepblack]

Bekkazy, дополнил ответ, думаю допилите под свои нужды сами.
Там вполне наглядный пример

[Bekkazy]

Андрей, Спасибо