From a7afa5001dffca5b8435f64eccff96524996d6a4 Mon Sep 17 00:00:00 2001 From: Interitio Date: Fri, 6 Jun 2025 22:25:05 +1000 Subject: [PATCH] fix(data) Fix for new pgsql lib version. --- src/data/connector.py | 4 ++-- src/data/cursor.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/data/connector.py b/src/data/connector.py index 7b25aed..6baf3b5 100644 --- a/src/data/connector.py +++ b/src/data/connector.py @@ -47,8 +47,8 @@ class Connector: return AsyncConnectionPool( self._conn_args, open=False, - min_size=4, - max_size=8, + min_size=1, + max_size=4, configure=self._setup_connection, kwargs=self._conn_kwargs ) diff --git a/src/data/cursor.py b/src/data/cursor.py index 5e01a8d..5d183e0 100644 --- a/src/data/cursor.py +++ b/src/data/cursor.py @@ -3,7 +3,7 @@ from typing import Optional from psycopg import AsyncCursor, sql from psycopg.abc import Query, Params -from psycopg._encodings import pgconn_encoding +from psycopg._encodings import conn_encoding logger = logging.getLogger(__name__) @@ -15,7 +15,7 @@ class AsyncLoggingCursor(AsyncCursor): elif isinstance(query, (sql.SQL, sql.Composed)): msg = query.as_string(self) elif isinstance(query, bytes): - msg = query.decode(pgconn_encoding(self._conn.pgconn), 'replace') + msg = query.decode(conn_encoding(self._conn.pgconn), 'replace') else: msg = repr(query) return msg