Set user properties on the scope using instead of

Author: ericapisani

sentry_sdk/integrations/quart.py

@@ -210,19 +210,22 @@ async def _request_websocket_started(app: "Quart", **kwargs: "Any") -> None:
                     request_websocket.query_string.decode("utf-8", errors="replace"),
                 )
 
-                current_user_id = _get_current_user_id_from_quart()
-                if current_user_id:
-                    sentry_sdk.get_current_scope().set_attribute(
-                        "user.id", current_user_id
-                    )
-
+                user_properties = {}
                 if len(request_websocket.access_route) >= 1:
                     segment.set_attribute(
                         "client.address", request_websocket.access_route[0]
                     )
                     segment.set_attribute(
                         "user.ip_address", request_websocket.access_route[0]
                     )
+                    user_properties["ip_address"] = request_websocket.access_route[0]
+
+                current_user_id = _get_current_user_id_from_quart()
+                if current_user_id:
+                    user_properties["id"] = current_user_id
+
+                if user_properties:
+                    sentry_sdk.get_current_scope().set_user(user_properties)
 
     evt_processor = _make_request_event_processor(app, request_websocket, integration)
     scope.add_event_processor(evt_processor)

tests/integrations/quart/test_quart.py

@@ -932,6 +932,7 @@ async def login():
     segment = spans[1]
     if send_default_pii and user_id is not None:
         assert segment["attributes"]["user.id"] == user_id
+        print("SEGMENT ATTRS", segment["attributes"])
     else:
         assert "user.id" not in segment.get("attributes", {})