Update export method to accept separate start/end datetimes #DATA-3443

Author: katiepeters

src/viam/app/data_client.py

@@ -340,7 +340,7 @@ async def get_latest_tabular_data(
         return response.time_captured.ToDatetime(), response.time_synced.ToDatetime(), struct_to_dict(response.payload)
 
     async def export_tabular_data(
-        self, part_id: str, resource_name: str, resource_subtype: str, method_name: str, interval: Optional[CaptureInterval] = None
+        self, part_id: str, resource_name: str, resource_subtype: str, method_name: str, start_time: Optional[datetime] = None, end_time: Optional[datetime] = None
     ) -> List[ExportTabularDataResponse]:
         """Obtain unified tabular data and metadata from the specified data source.
 
@@ -361,20 +361,20 @@ async def export_tabular_data(
             resource_name (str): The name of the requested resource that captured the data.
             resource_subtype (str): The subtype of the requested resource that captured the data.
             method_name (str): The data capture method name.
-            interval (CaptureInterval): Optional time interval to retrieve data for. Contains start and end fields as protobuf
-            Timestamps defining the time range to query.
+            start_time (datetime): Optional start time for requesting a specific range of data.
+            end_time (datetime): Optional end time for requesting a specific range of data.
 
         Returns:
             List[ExportTabularDataResponse]: The unified tabular data and metadata.
 
         For more information, see `Data Client API <https://docs.viam.com/appendix/apis/data-client/>`_.
         """
 
+        interval=CaptureInterval(start=datetime_to_timestamp(start_time), end=datetime_to_timestamp(end_time))
         request = ExportTabularDataRequest(
             part_id=part_id, resource_name=resource_name, resource_subtype=resource_subtype, method_name=method_name, interval=interval
         )
-        response: List[ExportTabularDataResponse] = await self._data_client.ExportTabularData(request, metadata=self._metadata)
-        return response
+        return await self._data_client.ExportTabularData(request, metadata=self._metadata)
 
     async def binary_data_by_filter(
         self,

tests/test_data_client.py

@@ -199,7 +199,7 @@ async def test_get_latest_tabular_data(self, service: MockData):
     async def test_export_tabular_data(self, service: MockData):
         async with ChannelFor([service]) as channel:
             client = DataClient(channel, DATA_SERVICE_METADATA)
-            tabular_data = await client.export_tabular_data(PART_ID, COMPONENT_NAME, COMPONENT_TYPE, METHOD, INTERVAL)
+            tabular_data = await client.export_tabular_data(PART_ID, COMPONENT_NAME, COMPONENT_TYPE, METHOD, START_DATETIME, END_DATETIME)
             assert tabular_data is not None
             for tabular_datum in tabular_data:
                 assert tabular_datum is not None