fix: Auto-enabling Redis and Pyramid integration

Author: untitaker

sentry_sdk/integrations/__init__.py

@@ -63,6 +63,8 @@ def iter_default_integrations(with_auto_enabling_integrations):
     "sentry_sdk.integrations.aiohttp.AioHttpIntegration",
     "sentry_sdk.integrations.tornado.TornadoIntegration",
     "sentry_sdk.integrations.sqlalchemy.SqlalchemyIntegration",
+    "sentry_sdk.integrations.redis.RedisIntegration",
+    "sentry_sdk.integrations.pyramid.PyramidIntegration",
 )
 
 

sentry_sdk/integrations/pyramid.py

@@ -4,17 +4,20 @@
 import sys
 import weakref
 
-from pyramid.httpexceptions import HTTPException
-from pyramid.request import Request
-
 from sentry_sdk.hub import Hub, _should_send_default_pii
 from sentry_sdk.utils import capture_internal_exceptions, event_from_exception
 from sentry_sdk._compat import reraise, iteritems
 
-from sentry_sdk.integrations import Integration
+from sentry_sdk.integrations import Integration, DidNotEnable
 from sentry_sdk.integrations._wsgi_common import RequestExtractor
 from sentry_sdk.integrations.wsgi import SentryWsgiMiddleware
 
+try:
+    from pyramid.httpexceptions import HTTPException
+    from pyramid.request import Request
+except ImportError:
+    raise DidNotEnable("Pyramid not installed")
+
 from sentry_sdk._types import MYPY
 
 if MYPY:
@@ -64,7 +67,6 @@ def __init__(self, transaction_style="route_name"):
     def setup_once():
         # type: () -> None
         from pyramid import router
-        from pyramid.request import Request
 
         old_call_view = router._call_view
 

sentry_sdk/integrations/redis.py

@@ -2,7 +2,7 @@
 
 from sentry_sdk import Hub
 from sentry_sdk.utils import capture_internal_exceptions
-from sentry_sdk.integrations import Integration
+from sentry_sdk.integrations import Integration, DidNotEnable
 
 from sentry_sdk._types import MYPY
 
@@ -16,7 +16,10 @@ class RedisIntegration(Integration):
     @staticmethod
     def setup_once():
         # type: () -> None
-        import redis
+        try:
+            import redis
+        except ImportError:
+            raise DidNotEnable("Redis client not installed")
 
         patch_redis_client(redis.StrictRedis)