Make sure we're importing redis the library (#2106)

...not the module, if there is one present.
getsentry · May 12, 2023 · ccdaed3 · ccdaed3
1 parent f805239
commit ccdaed3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sentry_sdk/integrations/redis.py b/sentry_sdk/integrations/redis.py
@@ -115,14 +115,14 @@ def __init__(self, max_data_size=_DEFAULT_MAX_DATA_SIZE):
     def setup_once():
         # type: () -> None
         try:
-            import redis
+            from redis import StrictRedis, client
         except ImportError:
             raise DidNotEnable("Redis client not installed")
 
-        patch_redis_client(redis.StrictRedis, is_cluster=False)
-        patch_redis_pipeline(redis.client.Pipeline, False, _get_redis_command_args)
+        patch_redis_client(StrictRedis, is_cluster=False)
+        patch_redis_pipeline(client.Pipeline, False, _get_redis_command_args)
         try:
-            strict_pipeline = redis.client.StrictPipeline  # type: ignore
+            strict_pipeline = client.StrictPipeline  # type: ignore
         except AttributeError:
             pass
         else: