Use new asyncio support in redis package (Fixes #911)

miguelgrinberg · miguelgrinberg · commit 0e7691b77605 · 2022-04-23T16:28:28.000+01:00
diff --git a/src/socketio/asyncio_redis_manager.py b/src/socketio/asyncio_redis_manager.py
@@ -2,9 +2,12 @@
 import pickle
 
 try:
-    import aioredis
+    from redis import asyncio as aioredis
 except ImportError:
-    aioredis = None
+    try:
+        import aioredis
+    except ImportError:
+        aioredis = None
 
 from .asyncio_pubsub_manager import AsyncPubSubManager
 
@@ -39,8 +42,7 @@ def __init__(self, url='redis://localhost:6379/0', channel='socketio',
                  write_only=False, logger=None, redis_options=None):
         if aioredis is None:
             raise RuntimeError('Redis package is not installed '
-                               '(Run "pip install aioredis" in your '
-                               'virtualenv).')
+                               '(Run "pip install redis" in your virtualenv).')
         if not hasattr(aioredis.Redis, 'from_url'):
             raise RuntimeError('Version 2 of aioredis package is required.')
         self.redis_url = url
