condense span streaming checks

Author: alexander-alderman-webb

sentry_sdk/integrations/django/__init__.py

@@ -681,9 +681,7 @@ def connect(self: "BaseDatabaseWrapper") -> None:
         with capture_internal_exceptions():
             sentry_sdk.add_breadcrumb(message="connect", category="query")
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name="connect",
@@ -709,9 +707,7 @@ def _commit(self: "BaseDatabaseWrapper") -> None:
         if integration is None or not integration.db_transaction_spans:
             return real_commit(self)
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=SPANNAME.DB_COMMIT,
@@ -737,9 +733,7 @@ def _rollback(self: "BaseDatabaseWrapper") -> None:
         if integration is None or not integration.db_transaction_spans:
             return real_rollback(self)
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=SPANNAME.DB_ROLLBACK,

sentry_sdk/integrations/django/asgi.py

@@ -183,9 +183,7 @@ async def sentry_wrapped_callback(
         if not integration or not integration.middleware_spans:
             return await callback(request, *args, **kwargs)
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=request.resolver_match.view_name,

sentry_sdk/integrations/django/caching.py

@@ -62,9 +62,7 @@ def _instrument_call(
         op = OP.CACHE_PUT if is_set_operation else OP.CACHE_GET
         description = _get_span_description(method_name, args, kwargs)
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=description,

sentry_sdk/integrations/django/middleware.py

@@ -85,9 +85,7 @@ def _check_middleware_span(
         if function_basename:
             description = "{}.{}".format(description, function_basename)
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         middleware_span: "Union[Span, StreamedSpan]"
         if span_streaming:
             middleware_span = sentry_sdk.traces.start_span(

sentry_sdk/integrations/django/signals_handlers.py

@@ -65,9 +65,9 @@ def sentry_sync_receiver_wrapper(
             def wrapper(*args: "Any", **kwargs: "Any") -> "Any":
                 signal_name = _get_receiver_name(receiver)
 
-                client = sentry_sdk.get_client()
-                span_streaming = has_span_streaming_enabled(client.options)
-
+                span_streaming = has_span_streaming_enabled(
+                    sentry_sdk.get_client().options
+                )
                 if span_streaming:
                     with sentry_sdk.traces.start_span(
                         name=signal_name,

sentry_sdk/integrations/django/tasks.py

@@ -33,9 +33,7 @@ def _sentry_enqueue(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
 
         name = qualname_from_function(self.func) or "<unknown Django task>"
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=name,

sentry_sdk/integrations/django/templates.py

@@ -66,9 +66,7 @@ def patch_templates() -> None:
     @property  # type: ignore
     @ensure_integration_enabled(DjangoIntegration, real_rendered_content.fget)
     def rendered_content(self: "SimpleTemplateResponse") -> str:
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=_get_template_name_description(self.template_name),

sentry_sdk/integrations/django/views.py

@@ -31,9 +31,7 @@ def patch_views() -> None:
     old_render = SimpleTemplateResponse.render
 
     def sentry_patched_render(self: "SimpleTemplateResponse") -> "Any":
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name="serialize response",
@@ -100,9 +98,7 @@ def sentry_wrapped_callback(request: "Any", *args: "Any", **kwargs: "Any") -> "A
         if not integration or not integration.middleware_spans:
             return callback(request, *args, **kwargs)
 
-        client = sentry_sdk.get_client()
-        span_streaming = has_span_streaming_enabled(client.options)
-
+        span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
         if span_streaming:
             with sentry_sdk.traces.start_span(
                 name=request.resolver_match.view_name,