kodexa 7.4.412538129491__tar.gz → 7.4.412795054059__tar.gz

{kodexa-7.4.412538129491 → kodexa-7.4.412795054059}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kodexa
-Version: 7.4.412538129491
+Version: 7.4.412795054059
 Summary: Python SDK for the Kodexa Platform
 Author: Austin Redenbaugh
 Author-email: austin@kodexa.com

{kodexa-7.4.412538129491 → kodexa-7.4.412795054059}/kodexa/platform/client.py

@@ -4783,7 +4783,8 @@ class DocumentFamilyEndpoint(DocumentFamily, ClientEndpoint):
         self.client.put(url, body=document_status.model_dump(by_alias=True))
 
     def add_document(
-            self, document: Document, content_object: Optional[ContentObject] = None
+            self, document: Document, content_object: Optional[ContentObject] = None,
+            taxonomies: Optional[List[Taxonomy]] = None, data_store: Optional[Store] = None
     ):
         """
         Add a document to the document family.
@@ -4791,22 +4792,36 @@ class DocumentFamilyEndpoint(DocumentFamily, ClientEndpoint):
         Args:
             document (Document): The document to add.
             content_object (Optional[ContentObject]): The content object. Defaults to None.
+            taxonomies (Optional[List[Taxonomy]]): List of taxonomies to use. Defaults to None.
+            data_store (Optional[Store]): Data store to add document to. Defaults to None.
         """
         url = (
             f'/api/stores/{self.store_ref.replace(":", "/")}/families/{self.id}/objects'
         )
         if content_object is None:
             content_object = self.content_objects[-1]
+
+        params = {
+            "sourceContentObjectId": content_object.id,
+            "transitionType": "DERIVED",
+            "documentVersion": document.version
+        }
+
+        # If we have a store but no taxonomies or the other way around then we need to throw an error
+        if (data_store and not taxonomies) or (taxonomies and not data_store):
+            raise Exception("If you provide a data store you must also provide taxonomies and vice-versa")
+
+        if taxonomies:
+            params["taxonomyRefs"] = ",".join([taxonomy.ref for taxonomy in taxonomies])
+
+        if data_store:
+            params["dataStoreRef"] = data_store.ref
+
         self.client.post(
             url,
-            params={
-                "sourceContentObjectId": content_object.id,
-                "transitionType": "DERIVED",
-                "documentVersion": document.version,
-            },
+            params=params,
             files={"file": document.to_kddb()},
         )
-
     def export_as_zip(self) -> bytes:
         """
         Export the document family as bytes.
@@ -5093,6 +5108,7 @@ class DataStoreEndpoint(StoreEndpoint):
             path: Optional[str] = None,
             root_name: str = "",
             friendly_names=True,
+            project_id: Optional[str] = None,
     ) -> str:
         """Get the data objects export of the store
 
@@ -5102,6 +5118,7 @@ class DataStoreEndpoint(StoreEndpoint):
             path (Optional[str]): The path to the data object
             root_name (str): The root name of the data objects export
             friendly_names (bool): Whether to use friendly names. Defaults to True
+            project_id (Optional[str]): The project ID
 
         Returns:
             str: The data objects export of the store
@@ -5117,6 +5134,9 @@ class DataStoreEndpoint(StoreEndpoint):
 
         if path:
             params["path"] = path
+            
+        if project_id:
+            params["projectId"] = project_id
 
         if output_format == "csv" and not path:
             raise ValueError("CSV output requires a path")

{kodexa-7.4.412538129491 → kodexa-7.4.412795054059}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kodexa"
-version = "7.4.412538129491"
+version = "7.4.412795054059"
 description = "Python SDK for the Kodexa Platform"
 authors = ["Austin Redenbaugh <austin@kodexa.com>", "Philip Dodds <philip@kodexa.com>", "Romar Cablao <rcablao@kodexa.com>", "Amadea Paula Dodds <amadeapaula@kodexa.com>"]
 readme = "README.md"