automizor 0.4.7__py3-none-any.whl → 0.4.9__py3-none-any.whl

automizor/__init__.py

@@ -1 +1 @@
-version = "0.4.7"
+version = "0.4.9"

automizor/datastore/__init__.py

@@ -1,16 +1,15 @@
-from functools import lru_cache
-
 from ._container import DataStoreContainer
+from ._datastore import DataStore
 
 
-@lru_cache
-def _get_datastore():
-    from ._datastore import DataStore
-
-    return DataStore()
+def configure(api_token: str):
+    """
+    Configures the DataStore instance with the provided API token.
+    """
+    DataStore.configure(api_token)
 
 
-def get_store(name: str) -> DataStoreContainer:
+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
@@ -86,8 +85,14 @@ def get_store(name: str) -> DataStoreContainer:
 
     """
 
-    datastore = _get_datastore()
+    datastore = DataStore.get_instance()
     return DataStoreContainer(
         datastore=datastore,
         name=name,
     )
+
+
+__all__ = [
+    "configure",
+    "get_store",
+]

automizor/datastore/_container.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 
-from ._datastore import JSON, DataStore
+from automizor.utils import JSON
 
 
 @dataclass
@@ -14,6 +14,8 @@ class DataStoreContainer:
         name: The name of the data store.
     """
 
+    from ._datastore import DataStore
+
     datastore: DataStore
     name: str
 

automizor/datastore/_datastore.py

@@ -1,11 +1,7 @@
-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"]]
+from automizor.utils import JSON, get_api_config, get_headers
 
 
 class DataStore:
@@ -24,11 +20,23 @@ class DataStore:
         session (requests.Session): The HTTP session used to make API requests.
     """
 
-    def __init__(self):
-        self.url, self.token = get_api_config()
+    _instance = None
+
+    @classmethod
+    def configure(cls, api_token: str | None = None):
+        cls._instance = cls(api_token)
+
+    def __init__(self, api_token: str | None = None):
+        self.url, self.token = get_api_config(api_token)
         self.session = requests.Session()
         self.session.headers.update(get_headers(self.token))
 
+    @classmethod
+    def get_instance(cls):
+        if cls._instance is None:
+            cls.configure()
+        return cls._instance
+
     def get_values(
         self,
         name: str,
@@ -81,7 +89,7 @@ class DataStore:
                 exc.response, "Failed to get datastore values"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to get datastore values") from exc
+            raise AutomizorError(f"Failed to get datastore values: {exc}") from exc
 
     def _set_values(self, name: str, values: JSON) -> None:
         url = f"https://{self.url}/api/v1/workflow/datastore/{name}/values/"
@@ -93,4 +101,4 @@ class DataStore:
                 exc.response, "Failed to set datastore values"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to set datastore values") from exc
+            raise AutomizorError(f"Failed to set datastore values: {exc}") from exc

automizor/job/__init__.py

@@ -1,13 +1,13 @@
-from functools import lru_cache
+from automizor.utils import JSON
 
-from ._job import JSON
+from ._job import Job
 
 
-@lru_cache
-def _get_job():
-    from ._job import Job
-
-    return Job()
+def configure(api_token: str):
+    """
+    Configures the Job instance with the provided API token.
+    """
+    Job.configure(api_token)
 
 
 def get_context() -> dict:
@@ -22,7 +22,7 @@ def get_context() -> dict:
         AutomizorJobError: If retrieving the job context fails.
     """
 
-    job = _get_job()
+    job = Job.get_instance()
     return job.get_context()
 
 
@@ -38,11 +38,12 @@ def set_result(name: str, value: JSON):
     Note: Errors during file operations will raise unhandled exceptions.
     """
 
-    job = _get_job()
+    job = Job.get_instance()
     return job.set_result(name, value)
 
 
 __all__ = [
+    "configure",
     "get_context",
     "set_result",
 ]

automizor/job/_job.py

@@ -1,13 +1,10 @@
 import json
 import os
-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"]]
+from automizor.utils import JSON, get_api_config, get_headers
 
 
 class Job:
@@ -49,14 +46,26 @@ class Job:
         job.set_result("result_name", {"key": "value"})
     """
 
-    def __init__(self):
+    _instance = None
+
+    @classmethod
+    def configure(cls, api_token: str | None = None):
+        cls._instance = cls(api_token)
+
+    def __init__(self, api_token: str | None = None):
         self._context_file = os.getenv("AUTOMIZOR_CONTEXT_FILE", None)
         self._job_id = os.getenv("AUTOMIZOR_JOB_ID", None)
 
-        self.url, self.token = get_api_config()
+        self.url, self.token = get_api_config(api_token)
         self.session = requests.Session()
         self.session.headers.update(get_headers(self.token))
 
+    @classmethod
+    def get_instance(cls):
+        if cls._instance is None:
+            cls.configure()
+        return cls._instance
+
     def get_context(self) -> dict:
         """
         Retrieves the context of the current job, which contains necessary information
@@ -128,4 +137,4 @@ class Job:
                 exc.response, "Failed to get job context"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to get job context") from exc
+            raise AutomizorError(f"Failed to get job context: {exc}") from exc

automizor/storage/__init__.py

@@ -1,17 +1,18 @@
 import json
 import mimetypes
-from functools import lru_cache
 from pathlib import Path
 from typing import List
 
-from ._storage import JSON
+from automizor.utils import JSON
 
+from ._storage import Storage
 
-@lru_cache
-def _get_storage():
-    from ._storage import Storage
 
-    return Storage()
+def configure(api_token: str):
+    """
+    Configures the Storage instance with the provided API token.
+    """
+    Storage.configure(api_token)
 
 
 def list_assets() -> List[str]:
@@ -22,7 +23,7 @@ def list_assets() -> List[str]:
         A list of all asset names.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     return storage.list_assets()
 
 
@@ -34,7 +35,7 @@ def delete_asset(name: str):
         name: The name identifier of the asset to delete.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     storage.delete_asset(name)
 
 
@@ -49,7 +50,7 @@ def get_bytes(name: str) -> bytes:
         The raw byte content of the asset.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     return storage.get_bytes(name)
 
 
@@ -65,7 +66,7 @@ def get_file(name: str, path: str) -> str:
         The path to the saved file, confirming the operation's success.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     return storage.get_file(name, path)
 
 
@@ -80,7 +81,7 @@ def get_json(name: str) -> JSON:
         The parsed JSON data, which can be a dict, list, or primitive data type.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     return storage.get_json(name)
 
 
@@ -95,7 +96,7 @@ def get_text(name: str) -> str:
         The content of the asset as a text string.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     return storage.get_text(name)
 
 
@@ -109,7 +110,7 @@ def set_bytes(name: str, data: bytes, content_type="application/octet-stream"):
         content_type: The MIME type of the asset.
     """
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     storage.set_bytes(name, data, content_type)
 
 
@@ -124,12 +125,12 @@ def set_file(name: str, path: str, content_type: str = None):
     """
 
     content = Path(path).read_bytes()
-    if content_type is None:
+    if not content_type:
         content_type, _ = mimetypes.guess_type(path)
         if content_type is None:
             content_type = "application/octet-stream"
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     storage.set_bytes(name, content, content_type)
 
 
@@ -146,7 +147,7 @@ def set_json(name: str, value: JSON, **kwargs):
     content = json.dumps(value, **kwargs).encode("utf-8")
     content_type = "application/json"
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     storage.set_bytes(name, content, content_type)
 
 
@@ -162,11 +163,12 @@ def set_text(name: str, text: str):
     content = text.encode("utf-8")
     content_type = "text/plain"
 
-    storage = _get_storage()
+    storage = Storage.get_instance()
     storage.set_bytes(name, content, content_type)
 
 
 __all__ = [
+    "configure",
     "list_assets",
     "delete_asset",
     "get_bytes",

automizor/storage/_storage.py

@@ -1,11 +1,9 @@
-from typing import Dict, List, Union
+from typing import List
 
 import requests
 
 from automizor.exceptions import AutomizorError, NotFound
-from automizor.utils import get_api_config, get_headers
-
-JSON = Union[str, int, float, bool, None, Dict[str, "JSON"], List["JSON"]]
+from automizor.utils import JSON, get_api_config, get_headers
 
 
 class Storage:
@@ -50,11 +48,23 @@ class Storage:
         text_data = storage.get_text("AssetName")
     """
 
-    def __init__(self):
-        self.url, self.token = get_api_config()
+    _instance = None
+
+    @classmethod
+    def configure(cls, api_token: str | None = None):
+        cls._instance = cls(api_token)
+
+    def __init__(self, api_token: str | None = None):
+        self.url, self.token = get_api_config(api_token)
         self.session = requests.Session()
         self.session.headers.update(get_headers(self.token))
 
+    @classmethod
+    def get_instance(cls):
+        if cls._instance is None:
+            cls.configure()
+        return cls._instance
+
     def list_assets(self) -> List[str]:
         """
         Retrieves a list of all asset names.
@@ -82,7 +92,7 @@ class Storage:
                 exc.response, "Failed to list assets"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to list assets") from exc
+            raise AutomizorError(f"Failed to list assets: {exc}") from exc
         return asset_names
 
     def delete_asset(self, name: str):
@@ -106,7 +116,7 @@ class Storage:
                 exc.response, "Failed to delete asset"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to delete asset") from exc
+            raise AutomizorError(f"Failed to delete asset: {exc}") from exc
 
     def get_bytes(self, name: str) -> bytes:
         """
@@ -228,7 +238,7 @@ class Storage:
                 exc.response, "Failed to create asset"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to create asset") from exc
+            raise AutomizorError(f"Failed to create asset: {exc}") from exc
 
     def _download_file(self, name: str, mode: str = "content"):
         url = self._get_asset_url(name)
@@ -250,7 +260,7 @@ class Storage:
                 exc.response, "Failed to download asset"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to download asset") from exc
+            raise AutomizorError(f"Failed to download asset: {exc}") from exc
 
     def _get_asset_url(self, name: str) -> str:
         url = f"https://{self.url}/api/v1/storage/asset/{name}/"
@@ -267,7 +277,7 @@ class Storage:
                 exc.response, "Failed to get asset URL"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to get asset URL") from exc
+            raise AutomizorError(f"Failed to get asset URL: {exc}") from exc
 
     def _update_asset(self, name: str, content: bytes, content_type: str):
         """
@@ -297,4 +307,4 @@ class Storage:
                 exc.response, "Failed to update asset"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to update asset") from exc
+            raise AutomizorError(f"Failed to update asset: {exc}") from exc

automizor/utils/__init__.py

@@ -1,26 +1,29 @@
 import os
 import platform
+from typing import Dict, List, Union
 
 from automizor import version
 from automizor.exceptions import AutomizorError
 
+JSON = Union[str, int, float, bool, None, Dict[str, "JSON"], List["JSON"]]
+
+
 OS_SYSTEM, OS_RELEASE, _ = platform.system_alias(
     platform.system(), platform.release(), platform.version()
 )
 
 
-def get_api_config() -> tuple[str, str]:
-    token_string = os.getenv("AUTOMIZOR_AGENT_TOKEN")
+def get_api_config(api_token: str | None = None) -> tuple[str, str]:
+    if api_token is None:
+        api_token = os.getenv("AUTOMIZOR_AGENT_TOKEN")
 
-    if not token_string:
-        raise AutomizorError("AUTOMIZOR_AGENT_TOKEN is not set.")
+        if not api_token:
+            raise AutomizorError("AUTOMIZOR_AGENT_TOKEN is not set.")
 
     try:
-        token, url = token_string.strip().split("@")
+        token, url = api_token.strip().split("@")
     except ValueError as exc:
-        raise AutomizorError(
-            "AUTOMIZOR_AGENT_TOKEN is not in the correct format."
-        ) from exc
+        raise AutomizorError("The API token is not in the correct format.") from exc
     return url, token
 
 

automizor/vault/__init__.py

@@ -1,14 +1,14 @@
-from functools import lru_cache
 from typing import Any, Dict
 
 from ._container import SecretContainer
+from ._vault import Vault
 
 
-@lru_cache
-def _get_vault():
-    from ._vault import Vault
-
-    return Vault()
+def configure(api_token: str):
+    """
+    Configures the Vault instance with the provided API token.
+    """
+    Vault.configure(api_token)
 
 
 def create_secret(
@@ -38,7 +38,7 @@ def create_secret(
         value=value,
     )
 
-    vault = _get_vault()
+    vault = Vault.get_instance()
     return vault.create_secret(secret)
 
 
@@ -56,7 +56,7 @@ def get_secret(name: str) -> SecretContainer:
         AutomizorVaultError: If retrieving the secret fails.
     """
 
-    vault = _get_vault()
+    vault = Vault.get_instance()
     return vault.get_secret(name)
 
 
@@ -74,12 +74,13 @@ def set_secret(secret: SecretContainer) -> SecretContainer:
         AutomizorVaultError: If updating the secret fails.
     """
 
-    vault = _get_vault()
+    vault = Vault.get_instance()
     return vault.set_secret(secret)
 
 
 __all__ = [
     "SecretContainer",
+    "configure",
     "create_secret",
     "get_secret",
     "set_secret",

automizor/vault/_vault.py

@@ -42,11 +42,23 @@ class Vault:
         vault.set_secret(secret)
     """
 
-    def __init__(self):
-        self.url, self.token = get_api_config()
+    _instance = None
+
+    @classmethod
+    def configure(cls, api_token: str | None = None):
+        cls._instance = cls(api_token)
+
+    def __init__(self, api_token: str | None = None):
+        self.url, self.token = get_api_config(api_token)
         self.session = requests.Session()
         self.session.headers.update(get_headers(self.token))
 
+    @classmethod
+    def get_instance(cls):
+        if cls._instance is None:
+            cls.configure()
+        return cls._instance
+
     def create_secret(self, secret: SecretContainer) -> SecretContainer:
         """
         Creates a new secret. Stores the secret in the `Automizor API`.
@@ -110,7 +122,7 @@ class Vault:
                 exc.response, "Failed to create secret"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to create secret") from exc
+            raise AutomizorError(f"Failed to create secret: {exc}") from exc
 
     def _get_secret(self, name: str) -> SecretContainer:
         url = f"https://{self.url}/api/v1/vault/secret/{name}/"
@@ -123,7 +135,7 @@ class Vault:
                 exc.response, "Failed to get secret"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to get secret") from exc
+            raise AutomizorError(f"Failed to get secret: {exc}") from exc
 
     def _update_secret(self, secret: SecretContainer) -> SecretContainer:
         url = f"https://{self.url}/api/v1/vault/secret/{secret.name}/"
@@ -136,4 +148,4 @@ class Vault:
                 exc.response, "Failed to update secret"
             ) from exc
         except Exception as exc:
-            raise AutomizorError("Failed to update secret") from exc
+            raise AutomizorError(f"Failed to update secret: {exc}") from exc

{automizor-0.4.7.dist-info → automizor-0.4.9.dist-info}/METADATA

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

automizor-0.4.9.dist-info/RECORD

@@ -0,0 +1,20 @@
+automizor/__init__.py,sha256=ElLL68L1maenU2D5V6AoUj5l5Opu9EHBBw3V9qIFxV4,18
+automizor/exceptions.py,sha256=P5imySIOtG3ZIk2kh41Yod4RnlgTj7Vf0P3M-RuxQJs,1382
+automizor/datastore/__init__.py,sha256=_rAdRfKnTa4SuSLl9GkAYziBPPLsy6yGfS0HprgokOk,2418
+automizor/datastore/_container.py,sha256=rHRkSWaRgy2rk6Qy-7ADFKvuET_COZ2i9VD7gvRDnlM,745
+automizor/datastore/_datastore.py,sha256=EW8AnZPvr32Yz3bC6SxadRgihY7_G-XcwSfLV5CMzMg,3552
+automizor/job/__init__.py,sha256=pDqSfR4lAyJNCi-fpG6byUPidHz8sntBs0bB2lJFVAY,1118
+automizor/job/_job.py,sha256=oVv7utlpaIPVBSnBXcV9jf7B3UNsWCDLtPkw0m3IJio,5444
+automizor/log/__init__.py,sha256=gEr2SuwN1FgX1NMnbphjg8_gfSic9L15H3WC868A-TQ,2104
+automizor/log/_log.py,sha256=P9jAFXVANs5bAGH6-S0pzuSbRKEcX8VlQ_3uu690awE,2948
+automizor/storage/__init__.py,sha256=RF4ccSAd0UepvJx5NFiotKt9mVr5WiZCb_4qrJ5OKx0,4140
+automizor/storage/_storage.py,sha256=ah8sCpRQm0MOnZgOgiBQF1Oamq_mz1VWrV17z6XOuFw,11262
+automizor/utils/__init__.py,sha256=W0JNrn7GByjJXi60djS6vluo__Tj6AyJioqu9D41tto,952
+automizor/vault/__init__.py,sha256=21Ag2zQk7x27Bh0yt_kSO2MkgT3eafbdlKe0NuQEqr8,1796
+automizor/vault/_container.py,sha256=-2y7kASigoIVAebuQBk-0R_sI4gfmvjsMLuMg_tR1xA,1945
+automizor/vault/_vault.py,sha256=-Xi_xMQLRB9patFTiZ4Z817F9VEvhIvx-YOOH8z_BKc,5020
+automizor-0.4.9.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+automizor-0.4.9.dist-info/METADATA,sha256=uNVJ3mjxM7iZZRRzbzTCdCE6FlUYdkXiNqLCJZZSG_c,668
+automizor-0.4.9.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+automizor-0.4.9.dist-info/top_level.txt,sha256=gScDy4I3tP6BMYAsTAlBXrxVh3E00zV0UioxwXJOI3Y,10
+automizor-0.4.9.dist-info/RECORD,,

{automizor-0.4.7.dist-info → automizor-0.4.9.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

automizor-0.4.7.dist-info/RECORD

@@ -1,20 +0,0 @@
-automizor/__init__.py,sha256=m3H0nxC6dWftoBK74_V3kjhpRmKkLX6wTCtSlFvMbjI,18
-automizor/exceptions.py,sha256=P5imySIOtG3ZIk2kh41Yod4RnlgTj7Vf0P3M-RuxQJs,1382
-automizor/datastore/__init__.py,sha256=NA7gJ9qFoBQYNKWHzwPqzEMwo4u0ysXlVRczmzY27BI,2303
-automizor/datastore/_container.py,sha256=2WOthNSshwsRc_3vujcr8D1dK1sTfZvOBbLUTQEBYI0,713
-automizor/datastore/_datastore.py,sha256=4HkNsctxfat64d05EWqweqjZPtUMjvG92SbRyvXIYLc,3336
-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
-automizor/log/_log.py,sha256=P9jAFXVANs5bAGH6-S0pzuSbRKEcX8VlQ_3uu690awE,2948
-automizor/storage/__init__.py,sha256=bEY2R0Vk0cqscKYNnX-MM222XrxbLXOvOSBDgFmVPLo,3984
-automizor/storage/_storage.py,sha256=DXcARkFZ3edoDRrDiR02zkO7LmGkRtcKTHmgPsal0lc,10989
-automizor/utils/__init__.py,sha256=2trRoR5lljYKbYdxmioSlvzajjQM0Wnoq3bvF9lEZ6w,811
-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.7.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-automizor-0.4.7.dist-info/METADATA,sha256=htYUcD3zFZeSVoC-AsbOdQGJNv8warmJijBy2YxWMoU,668
-automizor-0.4.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-automizor-0.4.7.dist-info/top_level.txt,sha256=gScDy4I3tP6BMYAsTAlBXrxVh3E00zV0UioxwXJOI3Y,10
-automizor-0.4.7.dist-info/RECORD,,