avrokit 0.1.2__tar.gz → 0.1.3__tar.gz

{avrokit-0.1.2 → avrokit-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: avrokit
-Version: 0.1.2
+Version: 0.1.3
 Summary: Python utilities for working with Avro data files
 Project-URL: Homepage, https://github.com/brandtg/avrokit
 Project-URL: Repository, https://github.com/brandtg/avrokit

{avrokit-0.1.2 → avrokit-0.1.3}/avrokit/url/google.py

@@ -152,7 +152,16 @@ class GoogleCloudStorageURL(URL):
                     self._current_remote.upload_from_filename(self._current_local.name)
             finally:
                 # Cleanup resources
-                self._current_local.close()  # N.b. deletes on close
+                local = self._current_local
+                name = local.name
+                local.close()
+                # N.b. the temp file is created with delete=False so it survives an external
+                # close (e.g. DataFileWriter closing the stream before URL.close() uploads).
+                # Unlink it explicitly here to avoid leaking a file per open/close cycle.
+                try:
+                    os.unlink(name)
+                except FileNotFoundError:
+                    pass
                 self._current_client.close()
                 self._current_local = None
                 self._current_local_stream = None

{avrokit-0.1.2 → avrokit-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@
 
 [project]
 name = "avrokit"
-version = "0.1.2"
+version = "0.1.3"
 description = "Python utilities for working with Avro data files"
 authors = [{ name = "Greg Brandt", email = "brandt.greg@gmail.com" }]
 license = "Apache-2.0"

{avrokit-0.1.2 → avrokit-0.1.3}/tests/test_url_google.py

@@ -13,7 +13,7 @@ from filelock import FileLock
 
 from avrokit.io import avro_reader, avro_schema, avro_writer
 from avrokit.url.factory import parse_url
-from avrokit.url.google import google_cloud_storage_client
+from avrokit.url.google import GoogleCloudStorageURL, google_cloud_storage_client
 
 faker = Faker()
 
@@ -163,3 +163,50 @@ class TestGoogleCloudStorageURL:
         ]
         # No matches returns empty list
         assert parse_url(f"{base}/nonexistent_*.avro").expand() == []
+
+    def test_no_temp_file_leak(self, monkeypatch):
+        class FakeBlob:
+            def __init__(self):
+                self.content = b""
+
+            def download_to_file(self, f):
+                f.write(self.content)
+                f.flush()
+
+            def upload_from_filename(self, name):
+                with open(name, "rb") as f:
+                    self.content = f.read()
+
+        class FakeBucket:
+            def __init__(self):
+                self.blob_obj = FakeBlob()
+
+            def blob(self, path):
+                return self.blob_obj
+
+        class FakeClient:
+            def __init__(self):
+                self.bucket_obj = FakeBucket()
+
+            def bucket(self, name):
+                return self.bucket_obj
+
+            def close(self):
+                pass
+
+        fake_client = FakeClient()
+        monkeypatch.setattr("avrokit.url.google._create_client", lambda: fake_client)
+
+        tmpdir = tempfile.mkdtemp()
+        monkeypatch.setattr(tempfile, "tempdir", tmpdir)
+
+        url = GoogleCloudStorageURL("gs://test-bucket/test/leak.txt", mode="w")
+        with url as f:
+            f.write("hello")
+        assert fake_client.bucket_obj.blob_obj.content == b"hello"
+        assert os.listdir(tmpdir) == []
+
+        url = GoogleCloudStorageURL("gs://test-bucket/test/leak.txt", mode="r")
+        with url as f:
+            assert f.read() == "hello"
+        assert os.listdir(tmpdir) == []

{avrokit-0.1.2 → avrokit-0.1.3}/uv.lock

@@ -26,7 +26,7 @@ wheels = [
 
 [[package]]
 name = "avrokit"
-version = "0.1.0"
+version = "0.1.2"
 source = { editable = "." }
 dependencies = [
     { name = "avro" },