From: Federico Caselli <cfederico87@gmail.com>
Date: Thu, 15 Sep 2022 21:38:47 +0000 (+0200)
Subject: Make 652755e7571c8cc2ec4e8beab8ef6b1f180c496b compatible with py36
X-Git-Tag: rel_1_4_42~28
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3bd8a3625f17b124557b0d3346c51b39782a9fb6;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Make 652755e7571c8cc2ec4e8beab8ef6b1f180c496b compatible with py36

Change-Id: I8072146e9c4405a96b566392afa8e57661aa2a42
---

diff --git a/lib/sqlalchemy/ext/asyncio/engine.py b/lib/sqlalchemy/ext/asyncio/engine.py
index 5bfda15070..ba4dd39b6f 100644
--- a/lib/sqlalchemy/ext/asyncio/engine.py
+++ b/lib/sqlalchemy/ext/asyncio/engine.py
@@ -551,7 +551,7 @@ class AsyncConnection(ProxyComparable, StartableContext, AsyncConnectable):
         return self.start().__await__()
 
     async def __aexit__(self, type_, value, traceback):
-        task = asyncio.create_task(self.close())
+        task = asyncio.get_running_loop().create_task(self.close())
         await asyncio.shield(task)
 
 
@@ -607,7 +607,7 @@ class AsyncEngine(ProxyComparable, AsyncConnectable):
                 await self.transaction.__aexit__(type_, value, traceback)
                 await self.conn.close()
 
-            task = asyncio.create_task(go())
+            task = asyncio.get_running_loop().create_task(go())
             await asyncio.shield(task)
 
     def __init__(self, sync_engine):
diff --git a/lib/sqlalchemy/ext/asyncio/session.py b/lib/sqlalchemy/ext/asyncio/session.py
index 7346840579..61874378dd 100644
--- a/lib/sqlalchemy/ext/asyncio/session.py
+++ b/lib/sqlalchemy/ext/asyncio/session.py
@@ -627,7 +627,7 @@ class AsyncSession(ReversibleProxy):
         return self
 
     async def __aexit__(self, type_, value, traceback):
-        task = asyncio.create_task(self.close())
+        task = asyncio.get_running_loop().create_task(self.close())
         await asyncio.shield(task)
 
     def _maker_context_manager(self):
@@ -649,7 +649,7 @@ class _AsyncSessionContextManager:
             await self.trans.__aexit__(type_, value, traceback)
             await self.async_session.__aexit__(type_, value, traceback)
 
-        task = asyncio.create_task(go())
+        task = asyncio.get_running_loop().create_task(go())
         await asyncio.shield(task)