summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/dialects/postgresql/asyncpg.py
diff options
context:
space:
mode:
authormike bayer <mike_mp@zzzcomputing.com>2021-09-18 14:00:16 +0000
committerGerrit Code Review <gerrit@ci3.zzzcomputing.com>2021-09-18 14:00:16 +0000
commit955e6bd558e15fa1b0cde9a944d6f53d202d91c2 (patch)
tree61a64b7361ab0890521771a5d185db787482eaaf /lib/sqlalchemy/dialects/postgresql/asyncpg.py
parentc50183274728544e40e7da4fd35cf240da5df656 (diff)
parent26140c08111da9833dd2eff0b5091494f253db46 (diff)
downloadsqlalchemy-955e6bd558e15fa1b0cde9a944d6f53d202d91c2.tar.gz
Merge "Surface driver connection object when using a proxied dialect"
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql/asyncpg.py')
-rw-r--r--lib/sqlalchemy/dialects/postgresql/asyncpg.py6
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