pmdb_utils 0.3.2__tar.gz → 0.3.4__tar.gz

{pmdb_utils-0.3.2 → pmdb_utils-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pmdb_utils
-Version: 0.3.2
+Version: 0.3.4
 Summary: Add your description here
 Author-email: saebod <saebod@hotmail.com>
 Requires-Python: >=3.13

{pmdb_utils-0.3.2 → pmdb_utils-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pmdb_utils"
-version = "0.3.2"
+version = "0.3.4"
 description = "Add your description here"
 readme = "README.md"
 authors = [

pmdb_utils-0.3.4/src/pmdb_utils/dataacceslayer/storageoptions.py

@@ -0,0 +1,49 @@
+import os
+
+
+class StorageOptions:
+    """
+    Configuration class for AWS/MinIO storage credentials and endpoints.
+    Allows initialization from arguments or environment variables.
+    """
+
+    def __init__(
+        self,
+        access_key_id=None,
+        secret_access_key=None,
+        endpoint_url=None,
+        allow_http="true",
+    ):
+        """
+        Initialize DeltaStorageOptions.
+        Args:
+            aws_access_key_id (str, optional): AWS access key ID. Defaults to None.
+            aws_secret_access_key (str, optional): AWS secret access key. Defaults to None.
+            aws_endpoint_url (str, optional): AWS/MinIO endpoint URL. Defaults to None.
+            region (str, optional): AWS region. Defaults to None.
+            allow_http (str, optional): Allow HTTP connections. Defaults to "true".
+        """
+        self.access_key_id = access_key_id or os.getenv("MINIO_ROOT_USER")
+        self.secret_access_key = secret_access_key or os.getenv("MINIO_ROOT_PASSWORD")
+        self.endpoint_url = endpoint_url or os.getenv("MINIO_ENDPOINT_URL")
+        self.allow_http = allow_http
+
+    @property
+    def minio(self) -> dict:
+        """
+        Return a dictionary of credentials and endpoint for MinIO/AWS SDKs.
+        Returns:
+            dict: Dictionary with keys 'aws_access_key_id', 'aws_secret_access_key', 'endpoint_url', 'allow_http'.
+        """
+        if (
+            not self.access_key_id
+            or not self.secret_access_key
+            or not self.endpoint_url
+        ):
+            raise ValueError("AWS credentials and endpoint URL must be provided.")
+        return {
+            "aws_access_key_id": self.access_key_id,
+            "aws_secret_access_key": self.secret_access_key,
+            "endpoint_url": self.endpoint_url,
+            "allow_http": self.allow_http,
+        }