set on scope instead of span

Author: alexander-alderman-webb

sentry_sdk/integrations/django/__init__.py

@@ -479,38 +479,32 @@ def _after_get_response(request: "WSGIRequest") -> None:
         elif is_lazy:
             return
 
-        if (
-            user is None
-            or not is_authenticated(user)
-            or type(scope.streamed_span) is not StreamedSpan
-        ):
+        if user is None or not is_authenticated(user):
             return
 
-        segment_span = scope.streamed_span._segment
-
         user_id = None
         try:
             user_id = str(user.pk)
         except Exception:
             pass
         if user_id is not None:
-            segment_span.set_attribute(SPANDATA.USER_ID, user_id)
+            scope.set_attribute(SPANDATA.USER_ID, user_id)
 
         user_email = None
         try:
             user_email = user.email
         except Exception:
             pass
         if user_email is not None:
-            segment_span.set_attribute(SPANDATA.USER_EMAIL, user_email)
+            scope.set_attribute(SPANDATA.USER_EMAIL, user_email)
 
         username = None
         try:
             username = user.get_username()
         except Exception:
             pass
         if username is not None:
-            segment_span.set_attribute(SPANDATA.USER_NAME, username)
+            scope.set_attribute(SPANDATA.USER_NAME, username)
 
 
 def _patch_get_response() -> None:

tests/integrations/django/test_basic.py

@@ -532,7 +532,7 @@ def test_user_captured(
 
 @pytest.mark.forked
 @pytest_mark_django_db_decorator()
-def test_materialized_user_captured_on_segment_span(
+def test_materialized_user_captured(
     sentry_init,
     client,
     capture_events,