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

polydb/models.py

@@ -19,6 +19,9 @@ class CloudProvider(Enum):
     S3_COMPATIBLE = "s3_compatible"
     POSTGRESQL = "postgresql"
     BLOCKCHAIN = "blockchain"
+    VAULT = "vault"
+    KAFKA = "kafka"
+    RABBITMQ = "rabbitmq"
 
 
 @dataclass
@@ -86,6 +89,38 @@ class StorageConfig:
         self.name = name
 
 
+class AzureKeyVaultConfig(StorageConfig):
+    def __init__(self, name: str = "secrets", vault_url: Optional[str] = None):
+        super().__init__(CloudProvider.AZURE, name)
+        self.vault_url = vault_url
+
+
+class AWSSecretsManagerConfig(StorageConfig):
+    def __init__(self, name: str = "secrets", region: Optional[str] = None):
+        super().__init__(CloudProvider.AWS, name)
+        self.region = region
+
+
+class GCPSecretManagerConfig(StorageConfig):
+    def __init__(self, name: str = "secrets", project_id: Optional[str] = None):
+        super().__init__(CloudProvider.GCP, name)
+        self.project_id = project_id
+
+
+class VaultConfig(StorageConfig):
+    def __init__(
+        self,
+        name: str = "secrets",
+        url: Optional[str] = None,
+        token: Optional[str] = None,
+        mount_point: str = "secret",
+    ):
+        super().__init__(CloudProvider.VAULT, name)
+        self.url = url
+        self.token = token
+        self.mount_point = mount_point
+
+
 class AzureStorageConfig(StorageConfig):
     def __init__(
         self,
@@ -170,6 +205,48 @@ class VercelQueueConfig(StorageConfig):
         self.url = url
 
 
+class KafkaQueueConfig(StorageConfig):
+    def __init__(
+        self,
+        name: str = "kafka_queue",
+        bootstrap_servers: str = "",
+        group_id: str = "",
+        security_protocol: str = "",
+        sasl_mechanism: Optional[str] = None,
+        sasl_plain_username: Optional[str] = None,
+        sasl_plain_password: Optional[str] = None,
+        ssl_cafile: Optional[str] = None,
+    ):
+        super().__init__(CloudProvider.KAFKA, name)
+        self.bootstrap_servers = bootstrap_servers
+        self.group_id = group_id
+        self.security_protocol = security_protocol
+        self.sasl_mechanism = sasl_mechanism
+        self.sasl_plain_username = sasl_plain_username
+        self.sasl_plain_password = sasl_plain_password
+        self.ssl_cafile = ssl_cafile
+
+
+class RabbitMQConfig(StorageConfig):
+    def __init__(
+        self,
+        name: str = "rabbitmq",
+        url: str = "",
+        host: str = "",
+        port: int = 0,
+        username: str = "",
+        password: str = "",
+        virtual_host: str = "",
+    ):
+        super().__init__(CloudProvider.RABBITMQ, name)
+        self.url = url
+        self.host = host
+        self.port = port
+        self.username = username
+        self.password = password
+        self.virtual_host = virtual_host
+
+
 class BlockchainStorageConfig(StorageConfig):
     def __init__(self, name: str = "blockchain", ipfs_url: Optional[str] = None):
         super().__init__(CloudProvider.BLOCKCHAIN, name)

polydb/retry.py

@@ -25,6 +25,13 @@ _NON_RETRYABLE_MARKERS = (
     "ResourceNotFound",
     "InvalidArgument",
     "AuthenticationFailed",
+    # session_vars' allow-list rejection (PostgreSQLAdapter._apply_session_vars) --
+    # a caller passing a session variable name outside the fixed allow-list is a
+    # programming error, not a transient one; the identical resolved call would
+    # fail identically on every retry, so this only burns the full backoff window
+    # for no chance of a different outcome (same reasoning already applied to the
+    # markers above).
+    "not in the allow-list",
 )