tiny refactor

Authored by: bashonly
This commit is contained in:
bashonly 2024-05-04 11:31:24 -05:00
parent 0b95a4f83f
commit ff84e9ec24
No known key found for this signature in database
GPG Key ID: 783F096F253D15B0
1 changed files with 3 additions and 2 deletions

View File

@ -133,12 +133,13 @@ class CurlCFFIRH(ImpersonateRequestHandler, InstanceStoreMixin):
extensions.pop('timeout', None)
def send(self, request: Request) -> Response:
target = self._get_request_target(request)
try:
response = super().send(request)
except HTTPError as e:
e.response.extensions['impersonate'] = self._get_request_target(request)
e.response.extensions['impersonate'] = target
raise
response.extensions['impersonate'] = self._get_request_target(request)
response.extensions['impersonate'] = target
return response
def _send(self, request: Request):