altcodepro-polydb-python 2.5.4__py3-none-any.whl → 2.5.7__py3-none-any.whl

polydb/cloudDatabaseFactory.py

@@ -6,17 +6,17 @@ from typing import Dict, List, Optional
 
 from .base import SharedFilesAdapter
 
-from .adapters.AzureFileStorageAdapter import AzureFileStorageAdapter
-from .adapters.EFSAdapter import EFSAdapter
-from .adapters.PostgreSQLAdapter import PostgreSQLAdapter
-from .adapters.AzureBlobStorageAdapter import AzureBlobStorageAdapter
-from .adapters.BlockchainBlobAdapter import BlockchainBlobAdapter
-from .adapters.GCPStorageAdapter import GCPStorageAdapter
-from .adapters.S3CompatibleAdapter import S3CompatibleAdapter
-from .adapters.VercelBlobAdapter import VercelBlobAdapter
-
+# Deliberately NOT importing adapter classes at module level (they used to
+# be, redundantly -- every branch below already does its own lazy, local
+# `from .adapters.XAdapter import XAdapter` at the point of use, and
+# `from __future__ import annotations` above means the return-type hints
+# don't need a real import either). The top-level imports served no
+# purpose except forcing every cloud SDK to be installed just to
+# `import polydb` at all -- found by actually trying a minimal install.
 from .models import (
+    AWSSecretsManagerConfig,
     AzureFileConfig,
+    AzureKeyVaultConfig,
     AzureQueueConfig,
     AzureStorageConfig,
     AzureTableConfig,
@@ -29,22 +29,21 @@ from .models import (
     FirestoreConfig,
     GCPFileConfig,
     GCPPubSubConfig,
+    GCPSecretManagerConfig,
     GCPStorageConfig,
+    KafkaQueueConfig,
     MongoConfig,
     PartitionConfig,
     PostgreSQLConfig,
+    RabbitMQConfig,
     SQSAdapterConfig,
     StorageConfig,
+    VaultConfig,
     VercelKVConfig,
     VercelQueueConfig,
     VercelStorageConfig,
 )
 from .utils import setup_logger
-from .adapters.AzureQueueAdapter import AzureQueueAdapter
-from .adapters.GCPPubSubAdapter import GCPPubSubAdapter
-from .adapters.VercelQueueAdapter import VercelQueueAdapter
-from .adapters.BlockchainQueueAdapter import BlockchainQueueAdapter
-from .adapters.SQSAdapter import SQSAdapter
 
 # ============================================================
 # FACTORY
@@ -100,6 +99,11 @@ class CloudDatabaseFactory:
             return CloudProvider.GCP
         if os.getenv("VERCEL_ENV"):
             return CloudProvider.VERCEL
+        # README documents MONGODB_URI as a detection signal; the check was
+        # missing here entirely, so it silently fell through to Postgres
+        # instead (found by running the package's own test suite).
+        if os.getenv("MONGODB_URI"):
+            return CloudProvider.MONGODB
 
         return CloudProvider.POSTGRESQL
 
@@ -358,6 +362,8 @@ class CloudDatabaseFactory:
         | GCPPubSubAdapter
         | VercelQueueAdapter
         | BlockchainQueueAdapter
+        | KafkaQueueAdapter
+        | RabbitMQAdapter
     ):
         with self._lock:
             if name in self.instances:
@@ -368,6 +374,7 @@ class CloudDatabaseFactory:
                 cfg = StorageConfig(provider=self.provider, name=name)
 
             if cfg.provider == CloudProvider.AZURE:
+                from .adapters.AzureQueueAdapter import AzureQueueAdapter
 
                 connection_string = None
                 if isinstance(cfg, AzureQueueConfig):
@@ -391,6 +398,7 @@ class CloudDatabaseFactory:
                 )
 
             elif cfg.provider == CloudProvider.GCP:
+                from .adapters.GCPPubSubAdapter import GCPPubSubAdapter
 
                 topic = None
                 project_id = ""
@@ -404,6 +412,7 @@ class CloudDatabaseFactory:
                 )
 
             elif cfg.provider == CloudProvider.VERCEL:
+                from .adapters.VercelQueueAdapter import VercelQueueAdapter
 
                 url = ""
                 token = ""
@@ -413,6 +422,7 @@ class CloudDatabaseFactory:
                 instance = VercelQueueAdapter(url or "", token or "")
 
             elif cfg.provider == CloudProvider.BLOCKCHAIN:
+                from .adapters.BlockchainQueueAdapter import BlockchainQueueAdapter
 
                 rpc_url = ""
                 private_key = ""
@@ -430,12 +440,65 @@ class CloudDatabaseFactory:
                     contract_abi=contract_abi,
                 )
 
+            elif cfg.provider == CloudProvider.KAFKA:
+                from .adapters.KafkaQueueAdapter import KafkaQueueAdapter
+
+                bootstrap_servers = ""
+                group_id = ""
+                security_protocol = ""
+                sasl_mechanism = None
+                sasl_plain_username = None
+                sasl_plain_password = None
+                ssl_cafile = None
+                if isinstance(cfg, KafkaQueueConfig):
+                    bootstrap_servers = cfg.bootstrap_servers
+                    group_id = cfg.group_id
+                    security_protocol = cfg.security_protocol
+                    sasl_mechanism = cfg.sasl_mechanism
+                    sasl_plain_username = cfg.sasl_plain_username
+                    sasl_plain_password = cfg.sasl_plain_password
+                    ssl_cafile = cfg.ssl_cafile
+                instance = KafkaQueueAdapter(
+                    bootstrap_servers=bootstrap_servers or "",
+                    group_id=group_id or "",
+                    security_protocol=security_protocol or "",
+                    sasl_mechanism=sasl_mechanism or "",
+                    sasl_plain_username=sasl_plain_username or "",
+                    sasl_plain_password=sasl_plain_password or "",
+                    ssl_cafile=ssl_cafile or "",
+                )
+
+            elif cfg.provider == CloudProvider.RABBITMQ:
+                from .adapters.RabbitMQAdapter import RabbitMQAdapter
+
+                url = ""
+                host = ""
+                port = 0
+                username = ""
+                password = ""
+                virtual_host = ""
+                if isinstance(cfg, RabbitMQConfig):
+                    url = cfg.url
+                    host = cfg.host
+                    port = cfg.port
+                    username = cfg.username
+                    password = cfg.password
+                    virtual_host = cfg.virtual_host
+                instance = RabbitMQAdapter(
+                    url=url or "",
+                    host=host or "",
+                    port=port or 0,
+                    username=username or "",
+                    password=password or "",
+                    virtual_host=virtual_host or "",
+                )
+
             else:
                 raise NotImplementedError(
                     f"Queue adapter is not supported for {self.provider.value}"
                 )
 
-            self.instances["queue"] = instance
+            self.instances[name] = instance
             return instance
 
     def get_files(self, name: str = "files"):
@@ -490,3 +553,54 @@ class CloudDatabaseFactory:
 
             self.instances[name] = instance
             return instance
+
+    # --------------------------------------------------------
+    # SECRETS
+    # --------------------------------------------------------
+    def get_secrets(self, name: str = "secrets"):
+        """Cloud-agnostic secrets: Azure Key Vault / AWS Secrets Manager /
+        GCP Secret Manager / HashiCorp Vault (self-hosted), dispatched the
+        same way as every other adapter type. Unlike the data adapters,
+        this deliberately does NOT fall back to self.provider by default --
+        a factory configured for provider=postgresql or provider=mongodb
+        has no secrets manager of its own, so an unconfigured/unrecognized
+        provider here falls back to Vault (self-hosted, cloud-agnostic)
+        rather than raising."""
+        with self._lock:
+            cache_key = f"secrets::{name}"
+            if cache_key in self.instances:
+                return self.instances[cache_key]
+
+            cfg = self.configs.get(name)
+            provider = cfg.provider if cfg else self.provider
+
+            if provider == CloudProvider.AZURE:
+                from .adapters.AzureKeyVaultAdapter import AzureKeyVaultAdapter
+
+                vault_url = cfg.vault_url if isinstance(cfg, AzureKeyVaultConfig) else None
+                instance = AzureKeyVaultAdapter(vault_url=vault_url or "")
+
+            elif provider == CloudProvider.AWS:
+                from .adapters.AWSSecretsManagerAdapter import AWSSecretsManagerAdapter
+
+                region = cfg.region if isinstance(cfg, AWSSecretsManagerConfig) else None
+                instance = AWSSecretsManagerAdapter(region=region or "")
+
+            elif provider == CloudProvider.GCP:
+                from .adapters.GCPSecretManagerAdapter import GCPSecretManagerAdapter
+
+                project_id = cfg.project_id if isinstance(cfg, GCPSecretManagerConfig) else None
+                instance = GCPSecretManagerAdapter(project_id=project_id or "")
+
+            else:
+                from .adapters.VaultAdapter import VaultAdapter
+
+                url, token, mount_point = None, None, "secret"
+                if isinstance(cfg, VaultConfig):
+                    url, token, mount_point = cfg.url, cfg.token, cfg.mount_point
+                instance = VaultAdapter(
+                    url=url or "", token=token or "", mount_point=mount_point
+                )
+
+            self.instances[cache_key] = instance
+            return instance