Add missing data export methods

0xRy4n · Jul 31, 2023 · d3d3506 · d3d3506
1 parent aab7589
commit d3d3506
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/intercom_python_sdk/apis/data_export/api.py b/intercom_python_sdk/apis/data_export/api.py
@@ -73,3 +73,13 @@ def export(self, created_before: Union[int, datetime], created_after: Union[int,
     @get("/download/content/data/{job_identifier}")  # Leading slash as we aren't using the base URI
     def download(self, job_identifier: Path("job_identifier", str)):  # noqa # type: ignore
         """ Download a data export. """
+
+    @returns(dexport_schemas.DataExportJobSchema())  # type: ignore
+    @post("cancel/{job_identifier}")
+    def cancel(self, job_identifier: Path("job_identifier", str)):  # noqa # type: ignore
+        """ Cancel a data export. """
+
+    @returns(dexport_schemas.DataExportJobSchema())  # type: ignore
+    @post("content/data/{job_identifier}")
+    def get(self, job_identifier: Path("job_identifier", str)):  # noqa # type: ignore
+        """ Get a data export. """
diff --git a/intercom_python_sdk/apis/data_export/models.py b/intercom_python_sdk/apis/data_export/models.py
@@ -15,6 +15,7 @@
 import requests
 from typing import (
     TYPE_CHECKING,
+    Optional
 )
 
 # From Current Package
@@ -72,15 +73,15 @@ def download_url(self) -> str:
 
     def update(self, **kwargs) -> 'DataExportJob':
         """ Update this data export job to fetch it's current status and values. """
-        job = self.api_client.get_export_job(job_identifier=self.job_identifier)
-        self.__update_self(job)
+        job = self.api_client.get(job_identifier=self.job_identifier)
+        self.__update_self(job)  # type: ignore
 
         return self
 
     def cancel(self) -> 'DataExportJob':
         """ Cancel this data export job. """
-        job = self.api_client.cancel_export_job(job_identifier=self.job_identifier)
-        self.__update_self(job)
+        job = self.api_client.cancel(job_identifier=self.job_identifier)
+        self.__update_self(job)  # type: ignore
 
         return self