diff options
Diffstat (limited to 'lib/sqlalchemy/dialects/mysql/asyncmy.py')
| -rw-r--r-- | lib/sqlalchemy/dialects/mysql/asyncmy.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/lib/sqlalchemy/dialects/mysql/asyncmy.py b/lib/sqlalchemy/dialects/mysql/asyncmy.py index f312cf79b..cde43398d 100644 --- a/lib/sqlalchemy/dialects/mysql/asyncmy.py +++ b/lib/sqlalchemy/dialects/mysql/asyncmy.py @@ -28,11 +28,10 @@ This dialect should normally be used only with the """ # noqa -import contextlib - from .pymysql import MySQLDialect_pymysql from ... import pool from ... import util +from ...util.concurrency import asynccontextmanager from ...util.concurrency import asyncio from ...util.concurrency import await_fallback from ...util.concurrency import await_only @@ -188,7 +187,7 @@ class AsyncAdapt_asyncmy_connection: self._execute_mutex = asyncio.Lock() self._ss_cursors = set() - @contextlib.asynccontextmanager + @asynccontextmanager async def _mutex_and_adapt_errors(self): async with self._execute_mutex: try: |