automizor 0.4.4__py3-none-any.whl → 0.4.6__py3-none-any.whl

automizor/__init__.py

@@ -1 +1 @@
-version = "0.4.4"
+version = "0.4.6"

automizor/datastore/__init__.py

@@ -0,0 +1,107 @@
+import sys
+import types
+from functools import lru_cache
+
+from ._datastore import JSON
+
+
+@lru_cache
+def _get_datastore():
+    from ._datastore import DataStore
+
+    return DataStore()
+
+
+class DataStoreProxy(types.ModuleType):
+    """
+    `DataStoreProxy` acts as a dynamic interface for interacting with various types of
+    data stores within the `Automizor Platform`. It provides a convenient way to access
+    and manipulate data stored in JSON and Key-Key-Value (KKV) formats.
+
+    This class leverages the `Automizor DataStore` module to fetch and update data,
+    utilizing a caching mechanism to enhance performance. The primary interaction is
+    through attribute access and assignment, making it simple to work with different
+    data structures.
+
+    Example usage:
+
+        .. code-block:: python
+
+            from automizor import datastore
+
+            # Initialize or update json store
+            datastore.countries = {
+                "US": {
+                    "name": "United States",
+                    "capital": "Washington, D.C.",
+                    "population": 331449281,
+                    "area": 9833520
+                },
+                "CA": {
+                    "name": "Canada",
+                    "capital": "Ottawa",
+                    "population": 38005238,
+                    "area": 9984670
+                }
+            }
+
+            # Get values from json store
+            countries = datastore.countries()
+
+            # Initialize or update kkv store
+            datastore.movies = {
+                "US": {
+                    "action": {
+                        "Die Hard": 1988,
+                        "The Matrix": 1999
+                    }
+                }
+            }
+
+            # Get values from kkv store
+            movies = datastore.movies("US")
+            movies_action = datastore.movies("US", "action")
+
+            # Insert or update values
+            datastore.movies = {
+                "US": {
+                    "action": {
+                        "Die Hard": 1988,
+                        "The Matrix": 1999,
+                        "John Wick": 2014
+                    },
+                    "comedy": {
+                        "The Hangover": 2009,
+                        "Superbad": 2007
+                    }
+                }
+            }
+
+            # Delete secondary key
+            datastore.movies = {
+                "US": {
+                    "action": None
+                }
+            }
+
+            # Delete primary key
+            datastore.movies = {
+                "US": None
+            }
+
+    """
+
+    def __getattr__(self, name):
+        datastore = _get_datastore()
+
+        def wrapper(primary_key=None, secondary_key=None):
+            return datastore.get_values(name, primary_key, secondary_key)
+
+        return wrapper
+
+    def __setattr__(self, name, values: JSON):
+        datastore = _get_datastore()
+        datastore.set_values(name, values)
+
+
+sys.modules[__name__] = DataStoreProxy(__name__)

automizor/datastore/_datastore.py

@@ -0,0 +1,95 @@
+from typing import Dict, List, Union
+
+import requests
+
+from automizor.exceptions import AutomizorError
+from automizor.utils import get_api_config, get_headers
+
+JSON = Union[str, int, float, bool, None, Dict[str, "JSON"], List["JSON"]]
+
+
+class DataStore:
+    """
+    `DataStore` is a class designed to interface with the `Automizor Platform` to manage and
+    manipulate data stored in various formats. It supports operations to retrieve and update
+    data using a unified API.
+
+    The class initializes an HTTP session with the necessary headers for authentication, and
+    provides methods to retrieve values, and set values in the store.
+
+    Attributes:
+        url (str): The base URL for the API endpoint.
+        token (str): The authentication token for API access.
+        session (requests.Session): The HTTP session used for making API requests.
+    """
+
+    def __init__(self):
+        self.url, self.token = get_api_config()
+        self.session = requests.Session()
+        self.session.headers.update(get_headers(self.token))
+
+    def get_values(
+        self,
+        name: str,
+        primary_key: str | None = None,
+        secondary_key: str | None = None,
+    ) -> JSON:
+        """
+        Retrieves values from the specified data store.
+
+        Parameters:
+            name (str): The name of the data store.
+            primary_key (str, optional): The primary key for the values. Defaults to None.
+            secondary_key (str, optional): The secondary key for the values. Defaults to None.
+
+        Returns:
+            JSON: The values from the data store.
+        """
+
+        return self._get_values(name, primary_key, secondary_key)
+
+    def set_values(self, name: str, values: JSON) -> None:
+        """
+        Sets values in the specified data store.
+
+        Parameters:
+            name (str): The name of the data store.
+            values (JSON): The values to set in the data store.
+        """
+
+        return self._set_values(name, values)
+
+    def _get_values(
+        self,
+        name: str,
+        primary_key: str | None = None,
+        secondary_key: str | None = None,
+    ) -> dict:
+        params = (
+            {"primary_key": primary_key, "secondary_key": secondary_key}
+            if primary_key or secondary_key
+            else {}
+        )
+        url = f"https://{self.url}/api/v1/workflow/datastore/{name}/values/"
+        try:
+            response = self.session.get(url, timeout=10, params=params)
+            response.raise_for_status()
+            return response.json()
+        except requests.HTTPError as exc:
+            raise AutomizorError.from_response(
+                exc.response, "Failed to get datastore values"
+            ) from exc
+        except Exception as exc:
+            raise AutomizorError("Failed to get datastore values") from exc
+
+    def _set_values(self, name: str, values: JSON) -> None:
+        url = f"https://{self.url}/api/v1/workflow/datastore/{name}/values/"
+        try:
+            response = self.session.post(url, json=values, timeout=10)
+            response.raise_for_status()
+        except requests.HTTPError as exc:
+            raise AutomizorError.from_response(
+                exc.response, "Failed to set datastore values"
+            ) from exc
+        except Exception as exc:
+            raise AutomizorError("Failed to set datastore values") from exc

{automizor-0.4.4.dist-info → automizor-0.4.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: automizor
-Version: 0.4.4
+Version: 0.4.6
 Summary: Python Automizor framework
 Home-page: https://github.com/automizor/automizor-python
 Author: Christian Fischer

{automizor-0.4.4.dist-info → automizor-0.4.6.dist-info}/RECORD

@@ -1,5 +1,7 @@
-automizor/__init__.py,sha256=Qk6wsgzl7cSMWWZ902odoQ-wBV5Fda2ShX82NN7uHzM,18
+automizor/__init__.py,sha256=DWNNbWBv-hyqVy76bKbnHWaDUbqL0BOtdkPSV_88dx0,18
 automizor/exceptions.py,sha256=P5imySIOtG3ZIk2kh41Yod4RnlgTj7Vf0P3M-RuxQJs,1382
+automizor/datastore/__init__.py,sha256=lOCAgf-976kgD1bo7rBi43Ahh9NK8TahOktTfHi0DBA,3016
+automizor/datastore/_datastore.py,sha256=f2SAmsn5HWMYN_WGar6q0PioIfEJQdWbGC-REKFI5Cw,3371
 automizor/job/__init__.py,sha256=DNRuT6cyPQBaPRG4vNalCmEvcJQl-73b5ZDFOfNOwIg,1019
 automizor/job/_job.py,sha256=_-ehqPwJdY89mWm3_VAuh7KRJdv-8iUu6iMAzwGLHM0,5235
 automizor/log/__init__.py,sha256=gEr2SuwN1FgX1NMnbphjg8_gfSic9L15H3WC868A-TQ,2104
@@ -10,8 +12,8 @@ automizor/utils/__init__.py,sha256=2trRoR5lljYKbYdxmioSlvzajjQM0Wnoq3bvF9lEZ6w,8
 automizor/vault/__init__.py,sha256=UjRiW3J0R9ABXc1gXIPyS3cqNCwMWxx0l-C0PsIg7R0,1699
 automizor/vault/_container.py,sha256=-2y7kASigoIVAebuQBk-0R_sI4gfmvjsMLuMg_tR1xA,1945
 automizor/vault/_vault.py,sha256=uRsjOjzsstZpYwJoHNg_cpv803Dzo4T2oF6hwiG3Eww,4688
-automizor-0.4.4.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-automizor-0.4.4.dist-info/METADATA,sha256=ov8bVxRLQVqu-Q5opOucEST3mEOJmQbpblVy1UF3fIA,668
-automizor-0.4.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-automizor-0.4.4.dist-info/top_level.txt,sha256=gScDy4I3tP6BMYAsTAlBXrxVh3E00zV0UioxwXJOI3Y,10
-automizor-0.4.4.dist-info/RECORD,,
+automizor-0.4.6.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+automizor-0.4.6.dist-info/METADATA,sha256=7njCCq1Bqgh7kwvoBAr-cdwr-vdNUxjPPS-DfvAWKMs,668
+automizor-0.4.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+automizor-0.4.6.dist-info/top_level.txt,sha256=gScDy4I3tP6BMYAsTAlBXrxVh3E00zV0UioxwXJOI3Y,10
+automizor-0.4.6.dist-info/RECORD,,