diff options
Diffstat (limited to 'lib/sqlalchemy/dialects/mysql/asyncmy.py')
| -rw-r--r-- | lib/sqlalchemy/dialects/mysql/asyncmy.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/lib/sqlalchemy/dialects/mysql/asyncmy.py b/lib/sqlalchemy/dialects/mysql/asyncmy.py index badf6e389..0fca338f5 100644 --- a/lib/sqlalchemy/dialects/mysql/asyncmy.py +++ b/lib/sqlalchemy/dialects/mysql/asyncmy.py @@ -31,6 +31,7 @@ This dialect should normally be used only with the from .pymysql import MySQLDialect_pymysql from ... import pool from ... import util +from ...engine import AdaptedConnection from ...util.concurrency import asynccontextmanager from ...util.concurrency import asyncio from ...util.concurrency import await_fallback @@ -171,7 +172,7 @@ class AsyncAdapt_asyncmy_ss_cursor(AsyncAdapt_asyncmy_cursor): return self.await_(self._cursor.fetchall()) -class AsyncAdapt_asyncmy_connection: +class AsyncAdapt_asyncmy_connection(AdaptedConnection): await_ = staticmethod(await_only) __slots__ = ("dbapi", "_connection", "_execute_mutex") @@ -321,5 +322,8 @@ class MySQLDialect_asyncmy(MySQLDialect_pymysql): return CLIENT.FOUND_ROWS + def get_driver_connection(self, connection): + return connection._connection + dialect = MySQLDialect_asyncmy |