Merge pull request #986 from yorickvP/yorickvp/mitm-stream

filter-pypi-responses: stream package requests that hit the proxy
nix-community · Jun 15, 2024 · 8235c8b · 8235c8b
2 parents 1f57898 + 318a87f
commit 8235c8b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkgs/fetchPipMetadata/filter-pypi-responses.py b/pkgs/fetchPipMetadata/filter-pypi-responses.py
@@ -85,6 +85,11 @@ def get_files_to_hide(pname, max_ts):
 """
 
 
+def responseheaders(flow: http.HTTPFlow) -> None:
+    if "/simple/" not in flow.request.url:
+        flow.response.stream = True
+
+
 def response(flow: http.HTTPFlow) -> None:
     if not "/simple/" in flow.request.url:
         return