Use psycopg2 LoggingConnection with sqlalchemy

sa_log_psycopg2.py

"""Use psycopg2 LoggingConnection with SQLAlalchemy."""

import logging

import psycopg2.extras
import sqlalchemy as sa


logging.basicConfig(level=logging.DEBUG)

log = logging.getLogger()


class LoggingConnection(psycopg2.extras.LoggingConnection):

    def __init__(self, *args, **kwargs) -> None:
        super().__init__(*args, **kwargs)
        self.initialize(log)


engine = sa.create_engine('postgresql://postgres@/',
                          connect_args={'connection_factory': LoggingConnection})

with engine.connect() as conn:
    print(conn.scalar(sa.func.version()))