Merge pull request #26 from alma/fix/create-data-exports

Update DataExport creation params

Author: olance

alma/endpoints/exports.py

@@ -15,9 +15,14 @@ def create(
         payout_id: str = None,
         start: datetime = None,
         end: datetime = None,
+        include_child_accounts = False,
+        custom_fields = None,
     ):
         """ Create a new export"""
-        data = {"type": export_type.value}
+        data = {
+            "type": export_type.value,
+            "include_child_accounts": include_child_accounts,
+        }
         if payout_id:
             data["payout"] = payout_id
 
@@ -26,6 +31,15 @@ def create(
 
         if end:
             data["end"] = int(end.timestamp())
+        
+        if custom_fields:
+            if isinstance(custom_fields, str):
+                data["custom_fields"] = custom_fields
+            else:
+                try:
+                    data["custom_fields"] = ",".join(custom_fields)
+                except TypeError as e:
+                    raise TypeError(f"Expected comma-separated string or an iterable yielding strings for custom_fields, got {type(custom_fields)}") from e
 
         return self.request(self.EXPORTS_PATH).set_body(data).post().expectJson(Export)