Merge pull request #31 from sypht-team/add_workflow_data_endpoints

Add workflow data endpoints

Author: wilmeryan

requirements.txt

@@ -1,2 +1,2 @@
 requests==2.26.0
-urllib3==1.26.0
+urllib3==1.26.5

setup.py

@@ -1,6 +1,6 @@
 from setuptools import find_packages, setup
 
-__version__ = "0.5.13"
+__version__ = "0.5.14"
 __pkg_name__ = "sypht"
 
 setup(
@@ -16,5 +16,5 @@
         "Programming Language :: Python :: 3.7",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
-    install_requires=["requests>=2.26.0", "urllib3>=1.26.0"],
+    install_requires=["requests>=2.26.0", "urllib3>=1.26.5"],
 )

sypht/client.py

@@ -114,7 +114,10 @@ def _authenticate_v1(self, endpoint, client_id, client_secret, audience):
     @staticmethod
     def _parse_response(response):
         if 200 <= response.status_code < 300:
-            return response.json()
+            try:
+                return response.json()
+            except json.decoder.JSONDecodeError:
+                return response.text
         else:
             raise Exception(
                 "Request failed with status code ({}): {}".format(
@@ -237,19 +240,26 @@ def get_validation_rules(self, company_id=None, rules_id=None, endpoint=None):
         company_id = company_id or self.company_id
         endpoint = urljoin(
             endpoint or self.base_endpoint,
-            f"workflows/rules/{company_id}/{rules_id}",
+            f"workflows/{company_id}/rules/{rules_id}",
         )
         headers = self._get_headers()
         headers["Accept"] = "application/json"
         headers["Content-Type"] = "application/json"
         return self._parse_response(self.requests.get(endpoint, headers=headers))
 
-    def set_validation_rules(self, validation_rules=None, schema=True, company_id=None, rules_id=None, endpoint=None):
+    def set_validation_rules(
+        self,
+        validation_rules=None,
+        schema=True,
+        company_id=None,
+        rules_id=None,
+        endpoint=None,
+    ):
         data = {"data": validation_rules, "schema": schema}
         company_id = company_id or self.company_id
         endpoint = urljoin(
             endpoint or self.base_endpoint,
-            f"workflows/rules/{company_id}/{rules_id}",
+            f"workflows/{company_id}/rules/{rules_id}",
         )
         headers = self._get_headers()
         headers["Accept"] = "application/json"
@@ -262,7 +272,43 @@ def delete_validation_rules(self, company_id=None, rules_id=None, endpoint=None)
         company_id = company_id or self.company_id
         endpoint = urljoin(
             endpoint or self.base_endpoint,
-            f"workflows/rules/{company_id}/{rules_id}",
+            f"workflows/{company_id}/rules/{rules_id}",
+        )
+        headers = self._get_headers()
+        headers["Accept"] = "application/json"
+        headers["Content-Type"] = "application/json"
+        return self._parse_response(self.requests.delete(endpoint, headers=headers))
+
+    def get_workflow_data(self, company_id=None, data_key=None, endpoint=None):
+        company_id = company_id or self.company_id
+        endpoint = urljoin(
+            endpoint or self.base_endpoint,
+            f"workflows/{company_id}/data/{data_key}",
+        )
+        headers = self._get_headers()
+        headers["Accept"] = "application/json"
+        headers["Content-Type"] = "application/json"
+        return self._parse_response(self.requests.get(endpoint, headers=headers))
+
+    def put_workflow_data(self, data, data_key, company_id=None, endpoint=None):
+        data = {"data": data}
+        company_id = company_id or self.company_id
+        endpoint = urljoin(
+            endpoint or self.base_endpoint,
+            f"workflows/{company_id}/data/{data_key}",
+        )
+        headers = self._get_headers()
+        headers["Accept"] = "application/json"
+        headers["Content-Type"] = "application/json"
+        return self._parse_response(
+            self.requests.put(endpoint, data=json.dumps(data), headers=headers)
+        )
+
+    def delete_workflow_data(self, rules_id=None, company_id=None, endpoint=None):
+        company_id = company_id or self.company_id
+        endpoint = urljoin(
+            endpoint or self.base_endpoint,
+            f"workflows/{company_id}/rules/{rules_id}",
         )
         headers = self._get_headers()
         headers["Accept"] = "application/json"