[networking] Add response extensions for impersonate info

test/test_networking.py

@@ -785,6 +785,25 @@ def test_supported_impersonate_targets(self, handler):
  assert res.status == 200
  assert std_headers['user-agent'].lower() not in res.read().decode().lower()
 
+ def test_response_extensions(self, handler):
+ with handler() as rh:
+ for target in rh.supported_targets:
+ request = Request(
+ f'http://127.0.0.1:{self.http_port}/gen_200', extensions={'impersonate': target})
+ res = validate_and_send(rh, request)
+ assert res.extensions['impersonate'] == rh._get_request_target(request)
+
+ def test_http_error_response_extensions(self, handler):
+ with handler() as rh:
+ for target in rh.supported_targets:
+ request = Request(
+ f'http://127.0.0.1:{self.http_port}/gen_404', extensions={'impersonate': target})
+ try:
+ validate_and_send(rh, request)
+ except HTTPError as e:
+ res = e.response
+ assert res.extensions['impersonate'] == rh._get_request_target(request)
+
 
 class TestRequestHandlerMisc:
  """Misc generic tests for request handlers, not related to request or validation testing"""

yt_dlp/networking/_curlcffi.py

@@ -132,6 +132,15 @@ def _check_extensions(self, extensions):
  extensions.pop('cookiejar', None)
  extensions.pop('timeout', None)
 
+ def send(self, request: Request) -> Response:
+ try:
+ response = super().send(request)
+ except HTTPError as e:
+ e.response.extensions['impersonate'] = self._get_request_target(request)
+ raise
+ response.extensions['impersonate'] = self._get_request_target(request)
+ return response
+
  def _send(self, request: Request):
  max_redirects_exceeded = False
  session: curl_cffi.requests.Session = self._get_instance(

yt_dlp/networking/common.py

@@ -517,6 +517,7 @@ def __init__(
  self.reason = reason or HTTPStatus(status).phrase
  except ValueError:
  self.reason = None
+ self.extensions = {}
 
  def readable(self):
  return self.fp.readable()