fix: close local files after read

Signed-off-by: Frost Ming <[email protected]>
frostming · Mar 27, 2024 · fc15716 · fc15716
1 parent 13609ac
commit fc15716
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/unearth/fetchers/sync.py b/src/unearth/fetchers/sync.py
@@ -26,6 +26,11 @@ def is_absolute_url(self) -> bool:
 httpx.URL.is_absolute_url = property(is_absolute_url)
 
 
+class FileByteStream(IteratorByteStream):
+    def close(self) -> None:
+        self._stream.close()  # type: ignore[attr-defined]
+
+
 class LocalFSTransport(httpx.BaseTransport):
     def handle_request(self, request: httpx.Request) -> httpx.Response:
         link = Link(str(request.url))
@@ -50,7 +55,7 @@ def handle_request(self, request: httpx.Request) -> httpx.Response:
             return httpx.Response(
                 status_code=200,
                 headers=headers,
-                stream=IteratorByteStream(path.open("rb")),
+                stream=FileByteStream(path.open("rb")),
             )