diff options
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql/asyncpg.py')
| -rw-r--r-- | lib/sqlalchemy/dialects/postgresql/asyncpg.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/lib/sqlalchemy/dialects/postgresql/asyncpg.py b/lib/sqlalchemy/dialects/postgresql/asyncpg.py index 825558f26..dc3da224c 100644 --- a/lib/sqlalchemy/dialects/postgresql/asyncpg.py +++ b/lib/sqlalchemy/dialects/postgresql/asyncpg.py @@ -121,6 +121,7 @@ from ... import exc from ... import pool from ... import processors from ... import util +from ...engine import AdaptedConnection from ...sql import sqltypes from ...util.concurrency import asyncio from ...util.concurrency import await_fallback @@ -566,7 +567,7 @@ class AsyncAdapt_asyncpg_ss_cursor(AsyncAdapt_asyncpg_cursor): ) -class AsyncAdapt_asyncpg_connection: +class AsyncAdapt_asyncpg_connection(AdaptedConnection): __slots__ = ( "dbapi", "_connection", @@ -1045,5 +1046,8 @@ class PGDialect_asyncpg(PGDialect): return connect + def get_driver_connection(self, connection): + return connection._connection + dialect = PGDialect_asyncpg |