etlplus 0.10.1__tar.gz → 0.10.2__tar.gz

{etlplus-0.10.1/etlplus.egg-info → etlplus-0.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: etlplus
-Version: 0.10.1
+Version: 0.10.2
 Summary: A Swiss Army knife for simple ETL operations
 Home-page: https://github.com/Dagitali/ETLPlus
 Author: ETLPlus Team

{etlplus-0.10.1 → etlplus-0.10.2}/etlplus/enums.py

@@ -524,6 +524,7 @@ def coerce_http_method(
 
 def infer_file_format_and_compression(
     value: object,
+    filename: object | None = None,
 ) -> tuple[FileFormat | None, CompressionFormat | None]:
     """
     Infer data format and compression from a filename, extension, or MIME type.
@@ -532,6 +533,9 @@ def infer_file_format_and_compression(
     ----------
     value : object
         A filename, extension, MIME type, or existing enum member.
+    filename : object | None, optional
+        A filename to consult for extension-based inference (e.g. when
+        ``value`` is ``application/octet-stream``).
 
     Returns
     -------
@@ -555,7 +559,25 @@ def infer_file_format_and_compression(
     compression = CompressionFormat.try_coerce(mime)
     fmt = FileFormat.try_coerce(mime)
 
-    suffixes = PurePath(text).suffixes
+    is_mime = mime.startswith(
+        (
+            'application/',
+            'text/',
+            'audio/',
+            'image/',
+            'video/',
+            'multipart/',
+        ),
+    )
+    suffix_source: object | None = filename if filename is not None else text
+    if is_mime and filename is None:
+        suffix_source = None
+
+    suffixes = (
+        PurePath(str(suffix_source)).suffixes
+        if suffix_source is not None
+        else []
+    )
     if suffixes:
         normalized_suffixes = [suffix.casefold() for suffix in suffixes]
         compression = (

{etlplus-0.10.1 → etlplus-0.10.2/etlplus.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: etlplus
-Version: 0.10.1
+Version: 0.10.2
 Summary: A Swiss Army knife for simple ETL operations
 Home-page: https://github.com/Dagitali/ETLPlus
 Author: ETLPlus Team

{etlplus-0.10.1 → etlplus-0.10.2}/tests/unit/test_u_enums.py

@@ -119,25 +119,32 @@ class TestInferFileFormatAndCompression:
     """Unit test suite for :func:`infer_file_format_and_compression`."""
 
     @pytest.mark.parametrize(
-        'value,expected_format,expected_compression',
+        'value,filename,expected_format,expected_compression',
         [
-            ('data.csv.gz', FileFormat.CSV, CompressionFormat.GZ),
-            ('data.jsonl.gz', FileFormat.NDJSON, CompressionFormat.GZ),
-            ('data.zip', None, CompressionFormat.ZIP),
-            ('application/json; charset=utf-8', FileFormat.JSON, None),
-            ('application/gzip', None, CompressionFormat.GZ),
-            (FileFormat.GZ, None, CompressionFormat.GZ),
-            (CompressionFormat.ZIP, None, CompressionFormat.ZIP),
+            ('data.csv.gz', None, FileFormat.CSV, CompressionFormat.GZ),
+            ('data.jsonl.gz', None, FileFormat.NDJSON, CompressionFormat.GZ),
+            ('data.zip', None, None, CompressionFormat.ZIP),
+            ('application/json; charset=utf-8', None, FileFormat.JSON, None),
+            ('application/gzip', None, None, CompressionFormat.GZ),
+            (
+                'application/octet-stream',
+                'payload.csv.gz',
+                FileFormat.CSV,
+                CompressionFormat.GZ,
+            ),
+            (FileFormat.GZ, None, None, CompressionFormat.GZ),
+            (CompressionFormat.ZIP, None, None, CompressionFormat.ZIP),
         ],
     )
     def test_infers_format_and_compression(
         self,
         value: object,
+        filename: object | None,
         expected_format: FileFormat | None,
         expected_compression: CompressionFormat | None,
     ) -> None:
         """Test mixed inputs for format and compression inference."""
-        fmt, compression = infer_file_format_and_compression(value)
+        fmt, compression = infer_file_format_and_compression(value, filename)
         assert fmt is expected_format
         assert compression is expected_compression