Подключение к postgreSQL

← разместить еще код

Расшаренный код:

from urllib import parse
import psycopg2

class DBConnection:
    def __init__(self, DB_URL):
        self.url = DB_URL
        self.con = None

    def __enter__(self):
        url = parse.urlparse(self.url)
        self.con = psycopg2.connect(
            database=url.path[1:],
            user=url.username,
            password=url.password,
            host=url.hostname,
            port=url.port
        )
        return self.con.cursor()

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.con.commit()
        self.con.close()