diff --git a/aws_xray_sdk/ext/requests/patch.py b/aws_xray_sdk/ext/requests/patch.py index 66e7ef81..cff0c9ad 100644 --- a/aws_xray_sdk/ext/requests/patch.py +++ b/aws_xray_sdk/ext/requests/patch.py @@ -6,10 +6,9 @@ def patch(): - wrapt.wrap_function_wrapper( 'requests', - 'Session.request', + 'Session.send', _xray_traced_requests ) @@ -22,7 +21,7 @@ def patch(): def _xray_traced_requests(wrapped, instance, args, kwargs): - url = kwargs.get('url') or args[1] + url = args[0].url return xray_recorder.record_subsegment( wrapped, instance, args, kwargs, @@ -43,9 +42,10 @@ def _inject_header(wrapped, instance, args, kwargs): def requests_processor(wrapped, instance, args, kwargs, return_value, exception, subsegment, stack): - - method = kwargs.get('method') or args[0] - url = kwargs.get('url') or args[1] + + request = args[0] + method = request.method + url = request.url subsegment.put_http_meta(http.METHOD, method) subsegment.put_http_meta(http.URL, strip_url(url))