Don't open connnection with loop for python 3.10

Originally posted in NoneGG/aredis#208
alisaifee · Jan 5, 2022 · d1de656 · d1de656
1 parent 0d9b4da
commit d1de656
Showing 1 changed file with 26 additions and 7 deletions.
diff --git a/coredis/connection.py b/coredis/connection.py
@@ -593,11 +593,21 @@ def __init__(self, host='127.0.0.1', port=6379, password=None,
         self.socket_keepalive_options = socket_keepalive_options or {}
 
     async def _connect(self):
+        if LOOP_DEPRECATED:
+            connection = asyncio.open_connection(
+                host=self.host,
+                port=self.port,
+                ssl=self.ssl_context
+            )
+        else:
+            connection = asyncio.open_connection(
+                host=self.host,
+                port=self.port,
+                ssl=self.ssl_context,
+                loop=self.loop
+            )
         reader, writer = await exec_with_timeout(
-            asyncio.open_connection(host=self.host,
-                                    port=self.port,
-                                    ssl=self.ssl_context,
-                                    loop=self.loop),
+            connection,
             self._connect_timeout,
             loop=self.loop
         )
@@ -642,10 +652,19 @@ def __init__(self, path='', password=None,
         }
 
     async def _connect(self):
+        if LOOP_DEPRECATED:
+            connection = asyncio.open_unix_connection(
+                path=self.path,
+                ssl=self.ssl_context
+            )
+        else:
+            connection = asyncio.open_unix_connection(
+                path=self.path,
+                ssl=self.ssl_context,
+                loop=self.loop
+            )
         reader, writer = await exec_with_timeout(
-            asyncio.open_unix_connection(path=self.path,
-                                         ssl=self.ssl_context,
-                                         loop=self.loop),
+            connection,
             self._connect_timeout,
             loop=self.loop
         )