automizor 0.4.5__tar.gz → 0.4.7__tar.gz

{automizor-0.4.5/automizor.egg-info → automizor-0.4.7}/PKG-INFO

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

automizor-0.4.7/automizor/__init__.py

@@ -0,0 +1 @@
+version = "0.4.7"

automizor-0.4.7/automizor/datastore/__init__.py

@@ -0,0 +1,93 @@
+from functools import lru_cache
+
+from ._container import DataStoreContainer
+
+
+@lru_cache
+def _get_datastore():
+    from ._datastore import DataStore
+
+    return DataStore()
+
+
+def get_store(name: str) -> DataStoreContainer:
+    """
+    Get a store container by name. The `DataStoreContainer` is a wrapper
+    around the data store that provides a get and set method to interact
+    with the data store.
+
+    Example usage:
+
+        .. code-block:: python
+
+            from automizor import datastore
+
+            # Get a data store countries
+            countries = datastore.get_store("countries")
+
+            # Initialize or update json store
+            countries.set([
+                {
+                    "name": "United States",
+                    "code": "US",
+                },
+                {
+                    "name": "Canada",
+                    "code": "CA",
+                },
+            ])
+
+            # Get values from json store
+            result = countries.get()
+
+            # Get a data store movies
+            movies = datastore.get_store("movies")
+
+            # Initialize or update kkv store
+            movies.set({
+                "US": {
+                    "action": {
+                        "Die Hard": 1988,
+                        "The Matrix": 1999
+                    }
+                }
+            })
+
+            # Get values from kkv store
+            result = movies.get("US")
+            result = movies.get("US", "action")
+
+            # Insert or update values
+            movies.set({
+                "US": {
+                    "action": {
+                        "Die Hard": 1988,
+                        "The Matrix": 1999,
+                        "John Wick": 2014
+                    },
+                    "comedy": {
+                        "The Hangover": 2009,
+                        "Superbad": 2007
+                    }
+                }
+            })
+
+            # Delete secondary key
+            movies.set({
+                "US": {
+                    "action": None
+                }
+            })
+
+            # Delete primary key
+            movies.set({
+                "US": None
+            })
+
+    """
+
+    datastore = _get_datastore()
+    return DataStoreContainer(
+        datastore=datastore,
+        name=name,
+    )

automizor-0.4.7/automizor/datastore/_container.py

@@ -0,0 +1,26 @@
+from dataclasses import dataclass
+
+from ._datastore import JSON, DataStore
+
+
+@dataclass
+class DataStoreContainer:
+    """
+    The `DataStoreContainer` is a wrapper around the data store that
+    provides a get and set method to interact with the data store.
+
+    Attributes:
+        datastore: The data store.
+        name: The name of the data store.
+    """
+
+    datastore: DataStore
+    name: str
+
+    def get(self, primary_key=None, secondary_key=None):
+        """Get values from the datastore."""
+        return self.datastore.get_values(self.name, primary_key, secondary_key)
+
+    def set(self, values: JSON):
+        """Set values in the datastore."""
+        self.datastore.set_values(self.name, values)

{automizor-0.4.5 → automizor-0.4.7}/automizor/datastore/_datastore.py

@@ -10,17 +10,18 @@ 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.
+    `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 get the type of data store, retrieve values, and set values in the store.
+    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.
+        session (requests.Session): The HTTP session used to make API requests.
     """
 
     def __init__(self):
@@ -28,19 +29,6 @@ class DataStore:
         self.session = requests.Session()
         self.session.headers.update(get_headers(self.token))
 
-    def type(self, name: str) -> str:
-        """
-        Gets the type of the specified data store.
-
-        Parameters:
-            name (str): The name of the data store.
-
-        Returns:
-            str: The type of the data store.
-        """
-
-        return self._get_type(name)
-
     def get_values(
         self,
         name: str,
@@ -52,8 +40,8 @@ class DataStore:
 
         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.
+            primary_key (str, optional): The primary key for the values.
+            secondary_key (str, optional): The secondary key for the values.
 
         Returns:
             JSON: The values from the data store.
@@ -72,25 +60,12 @@ class DataStore:
 
         return self._set_values(name, values)
 
-    def _get_type(self, name: str) -> str:
-        url = f"https://{self.url}/api/v1/workflow/datastore/{name}/"
-        try:
-            response = self.session.get(url, timeout=10)
-            response.raise_for_status()
-            return response.json().get("datastore_type")
-        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 _get_values(
         self,
         name: str,
         primary_key: str | None = None,
         secondary_key: str | None = None,
-    ) -> dict:
+    ) -> JSON:
         params = (
             {"primary_key": primary_key, "secondary_key": secondary_key}
             if primary_key or secondary_key

{automizor-0.4.5 → automizor-0.4.7/automizor.egg-info}/PKG-INFO

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

{automizor-0.4.5 → automizor-0.4.7}/automizor.egg-info/SOURCES.txt

@@ -11,6 +11,7 @@ automizor.egg-info/dependency_links.txt
 automizor.egg-info/requires.txt
 automizor.egg-info/top_level.txt
 automizor/datastore/__init__.py
+automizor/datastore/_container.py
 automizor/datastore/_datastore.py
 automizor/job/__init__.py
 automizor/job/_job.py

automizor-0.4.5/automizor/__init__.py

@@ -1 +0,0 @@
-version = "0.4.5"

automizor-0.4.5/automizor/datastore/__init__.py

@@ -1,115 +0,0 @@
-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.
-
-    The `DataStoreProxy` dynamically determines the type of data store being accessed
-    and adapts its behavior accordingly. For JSON data stores, it directly retrieves
-    values. For KKV stores, it provides a wrapper function that facilitates more complex
-    data retrieval operations based on primary and secondary keys.
-
-    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()
-        datastore_type = datastore.type(name)
-
-        def wrapper(primary_key=None, secondary_key=None):
-            return datastore.get_values(name, primary_key, secondary_key)
-
-        if datastore_type == "JSONDataStore":
-            return datastore.get_values(name)
-        return wrapper
-
-    def __setattr__(self, name, values: JSON):
-        datastore = _get_datastore()
-        datastore.set_values(name, values)
-
-
-sys.modules[__name__] = DataStoreProxy(__name__)