@@ -233,8 +233,8 @@ def _prepare_headers(headers: typing.Optional[Headers]) -> httpx.Headers:
233
233
return httpx .Headers (headers )
234
234
235
235
236
- def extract_parameters (args , kwargs ):
237
- if type (args [0 ]) == httpx .Request :
236
+ def _extract_parameters (args , kwargs ):
237
+ if isinstance (args [0 ], httpx .Request ) :
238
238
request : httpx .Request = args [0 ]
239
239
method = request .method .encode ()
240
240
url = request .url
@@ -253,10 +253,10 @@ def extract_parameters(args, kwargs):
253
253
return method , url , headers , stream , extensions
254
254
255
255
256
- def inject_propagation_headers (headers , args , kwargs ):
256
+ def _inject_propagation_headers (headers , args , kwargs ):
257
257
_headers = _prepare_headers (headers )
258
258
inject (_headers )
259
- if type (args [0 ]) == httpx .Request :
259
+ if isinstance (args [0 ], httpx .Request ) :
260
260
request : httpx .Request = args [0 ]
261
261
request .headers = _headers
262
262
else :
@@ -303,7 +303,7 @@ def handle_request(
303
303
if context .get_value ("suppress_instrumentation" ):
304
304
return self ._transport .handle_request (* args , ** kwargs )
305
305
306
- method , url , headers , stream , extensions = extract_parameters (
306
+ method , url , headers , stream , extensions = _extract_parameters (
307
307
args , kwargs
308
308
)
309
309
span_attributes = _prepare_attributes (method , url )
@@ -319,9 +319,9 @@ def handle_request(
319
319
if self ._request_hook is not None :
320
320
self ._request_hook (span , request_info )
321
321
322
- inject_propagation_headers (headers , args , kwargs )
322
+ _inject_propagation_headers (headers , args , kwargs )
323
323
response = self ._transport .handle_request (* args , ** kwargs )
324
- if type (response ) == httpx .Response :
324
+ if isinstance (response , httpx .Response ) :
325
325
response : httpx .Response = response
326
326
status_code = response .status_code
327
327
headers = response .headers
@@ -380,7 +380,7 @@ async def handle_async_request(
380
380
if context .get_value ("suppress_instrumentation" ):
381
381
return await self ._transport .handle_async_request (* args , ** kwargs )
382
382
383
- method , url , headers , stream , extensions = extract_parameters (
383
+ method , url , headers , stream , extensions = _extract_parameters (
384
384
args , kwargs
385
385
)
386
386
span_attributes = _prepare_attributes (method , url )
@@ -396,12 +396,12 @@ async def handle_async_request(
396
396
if self ._request_hook is not None :
397
397
await self ._request_hook (span , request_info )
398
398
399
- inject_propagation_headers (headers , args , kwargs )
399
+ _inject_propagation_headers (headers , args , kwargs )
400
400
401
401
response = await self ._transport .handle_async_request (
402
402
* args , ** kwargs
403
403
)
404
- if type (response ) == httpx .Response :
404
+ if isinstance (response , httpx .Response ) :
405
405
response : httpx .Response = response
406
406
status_code = response .status_code
407
407
headers = response .headers
0 commit comments