python(feat): updated export.protos to support parquet

Author: wei-qlu

python/lib/sift/exports/v1/exports_pb2.py

@@ -30,12 +30,14 @@ class _ExportOutputFormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrap
     EXPORT_OUTPUT_FORMAT_UNSPECIFIED: _ExportOutputFormat.ValueType  # 0
     EXPORT_OUTPUT_FORMAT_CSV: _ExportOutputFormat.ValueType  # 1
     EXPORT_OUTPUT_FORMAT_SUN: _ExportOutputFormat.ValueType  # 2
+    EXPORT_OUTPUT_FORMAT_PARQUET: _ExportOutputFormat.ValueType  # 3
 
 class ExportOutputFormat(_ExportOutputFormat, metaclass=_ExportOutputFormatEnumTypeWrapper): ...
 
 EXPORT_OUTPUT_FORMAT_UNSPECIFIED: ExportOutputFormat.ValueType  # 0
 EXPORT_OUTPUT_FORMAT_CSV: ExportOutputFormat.ValueType  # 1
 EXPORT_OUTPUT_FORMAT_SUN: ExportOutputFormat.ValueType  # 2
+EXPORT_OUTPUT_FORMAT_PARQUET: ExportOutputFormat.ValueType  # 3
 global___ExportOutputFormat = ExportOutputFormat
 
 @typing.final
@@ -225,19 +227,23 @@ class ExportDataResponse(google.protobuf.message.Message):
 
     PRESIGNED_URL_FIELD_NUMBER: builtins.int
     JOB_ID_FIELD_NUMBER: builtins.int
+    SERVICE_MESSAGE_FIELD_NUMBER: builtins.int
     presigned_url: builtins.str
     """Pre-signed URL of exported zip containing export files. This will be blank if
     the job is processed as a background job, in which case `job_id` will be non-empty.
     """
     job_id: builtins.str
     """This will be non-empty if the export is being processed asynchronously in a background job."""
+    service_message: builtins.str
+    """Optional message for the client to display in success or error notification (e.g. exclusions or 403 reason)."""
     def __init__(
         self,
         *,
         presigned_url: builtins.str = ...,
         job_id: builtins.str = ...,
+        service_message: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["job_id", b"job_id", "presigned_url", b"presigned_url"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["job_id", b"job_id", "presigned_url", b"presigned_url", "service_message", b"service_message"]) -> None: ...
 
 global___ExportDataResponse = ExportDataResponse
 
@@ -261,12 +267,16 @@ class GetDownloadUrlResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     PRESIGNED_URL_FIELD_NUMBER: builtins.int
+    SERVICE_MESSAGE_FIELD_NUMBER: builtins.int
     presigned_url: builtins.str
+    service_message: builtins.str
+    """Optional message for the client to display in success or error notification."""
     def __init__(
         self,
         *,
         presigned_url: builtins.str = ...,
+        service_message: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["presigned_url", b"presigned_url"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["presigned_url", b"presigned_url", "service_message", b"service_message"]) -> None: ...
 
 global___GetDownloadUrlResponse = GetDownloadUrlResponse

python/lib/sift/exports/v1/exports_pb2.pyi

@@ -38,6 +38,7 @@
 START = datetime(2025, 1, 1, tzinfo=timezone.utc)
 STOP = datetime(2025, 1, 2, tzinfo=timezone.utc)
 CSV = ExportOutputFormat.CSV
+PARQUET = ExportOutputFormat.PARQUET
 
 
 @pytest.fixture
@@ -240,6 +241,20 @@ def test_export_by_time_range(self, exports_api_sync, sift_client, nostromo_run)
             assert isinstance(job, Job)
 
 
+class TestParquetFormat:
+    @pytest.mark.asyncio
+    async def test_export_parquet_format(self, exports_api):
+        """Test that Parquet output format is accepted."""
+        await exports_api.export(
+            runs=["run-1"],
+            start_time=START,
+            stop_time=STOP,
+            output_format=PARQUET,
+        )
+        call_kwargs = exports_api._low_level_client.export_data.call_args.kwargs
+        assert call_kwargs["output_format"] == PARQUET
+
+
 class TestDictConversion:
     @pytest.mark.asyncio
     async def test_calculated_channel_dict_converted(self, exports_api):

python/lib/sift_client/_tests/resources/test_exports.py

@@ -71,7 +71,7 @@ async def export(
         You cannot provide both ``runs`` and ``assets`` at the same time.
 
         Args:
-            output_format: The file format for the export (CSV or Sun/WinPlot).
+            output_format: The file format for the export (CSV, Parquet, or Sun/WinPlot).
             runs: One or more Run objects or run IDs to export data from.
             assets: One or more Asset objects or asset IDs to export data from.
             start_time: Start of the time range to export. Required when using

python/lib/sift_client/resources/exports.py

@@ -593,7 +593,7 @@ class DataExportAPI:
         You cannot provide both ``runs`` and ``assets`` at the same time.
 
         Args:
-            output_format: The file format for the export (CSV or Sun/WinPlot).
+            output_format: The file format for the export (CSV, Parquet, or Sun/WinPlot).
             runs: One or more Run objects or run IDs to export data from.
             assets: One or more Asset objects or asset IDs to export data from.
             start_time: Start of the time range to export. Required when using

python/lib/sift_client/resources/sync_stubs/__init__.pyi

@@ -11,7 +11,9 @@ class ExportOutputFormat(Enum):
     Attributes:
         CSV: Comma-separated values format.
         SUN: Sun (WinPlot) format (not used in certain environments).
+        PARQUET: Apache Parquet columnar storage format.
     """
 
     CSV = ExportOutputFormatProto.EXPORT_OUTPUT_FORMAT_CSV
     SUN = ExportOutputFormatProto.EXPORT_OUTPUT_FORMAT_SUN
+    PARQUET = ExportOutputFormatProto.EXPORT_OUTPUT_FORMAT_PARQUET