altcodepro-polydb-python 2.3.13__tar.gz → 2.3.15__tar.gz

{altcodepro_polydb_python-2.3.13/src/altcodepro_polydb_python.egg-info → altcodepro_polydb_python-2.3.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.3.13
+Version: 2.3.15
 Summary: Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety
 Author: AltCodePro
 Project-URL: Homepage, https://github.com/altcodepro/polydb-python
@@ -38,6 +38,8 @@ Requires-Dist: web3>=7.16.0
 Requires-Dist: google-cloud-firestore>=2.27.0
 Requires-Dist: google-cloud-pubsub>=2.38.0
 Requires-Dist: pymongo>=4.17.0
+Requires-Dist: build>=1.5.0
+Requires-Dist: twine>=6.2.0
 Provides-Extra: aws
 Requires-Dist: boto3>=1.42.47; extra == "aws"
 Requires-Dist: botocore>=1.42.47; extra == "aws"

{altcodepro_polydb_python-2.3.13 → altcodepro_polydb_python-2.3.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "altcodepro-polydb-python"
-version = "2.3.13"
+version = "2.3.15"
 description = "Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
@@ -57,6 +57,8 @@ dependencies = [
     "google-cloud-firestore>=2.27.0",
     "google-cloud-pubsub>=2.38.0",
     "pymongo>=4.17.0",
+    "build>=1.5.0",
+    "twine>=6.2.0",
 ]
 
 # Generic/Open-source stack (cheapest option)

{altcodepro_polydb_python-2.3.13 → altcodepro_polydb_python-2.3.15/src/altcodepro_polydb_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.3.13
+Version: 2.3.15
 Summary: Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety
 Author: AltCodePro
 Project-URL: Homepage, https://github.com/altcodepro/polydb-python
@@ -38,6 +38,8 @@ Requires-Dist: web3>=7.16.0
 Requires-Dist: google-cloud-firestore>=2.27.0
 Requires-Dist: google-cloud-pubsub>=2.38.0
 Requires-Dist: pymongo>=4.17.0
+Requires-Dist: build>=1.5.0
+Requires-Dist: twine>=6.2.0
 Provides-Extra: aws
 Requires-Dist: boto3>=1.42.47; extra == "aws"
 Requires-Dist: botocore>=1.42.47; extra == "aws"

{altcodepro_polydb_python-2.3.13 → altcodepro_polydb_python-2.3.15}/src/altcodepro_polydb_python.egg-info/requires.txt

@@ -13,6 +13,8 @@ web3>=7.16.0
 google-cloud-firestore>=2.27.0
 google-cloud-pubsub>=2.38.0
 pymongo>=4.17.0
+build>=1.5.0
+twine>=6.2.0
 
 [all]
 boto3>=1.42.47

{altcodepro_polydb_python-2.3.13 → altcodepro_polydb_python-2.3.15}/src/polydb/PolyDB.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 from typing import Any, Dict, List, Optional, Tuple, Type, Union
 
 from .advanced_query import AdvancedQueryBuilder, QueryHelper
@@ -23,8 +24,6 @@ from .types import JsonDict, Lookup
 from .utils import setup_logger
 from .validation import ModelValidator, SchemaValidator
 
-ModelRef = Union[Type, str]
-
 
 class PolyDB:
     """
@@ -51,6 +50,7 @@ class PolyDB:
         storage_configs: Optional[List[StorageConfig]] = None,
         tenant_registry: Optional[TenantRegistry] = None,
         partition_config: Optional[PartitionConfig] = None,
+        redis_cache_url: Optional[str] = None,
         enable_retries: bool = True,
         enable_audit: bool = True,
         enable_audit_reads: bool = False,
@@ -59,6 +59,7 @@ class PolyDB:
         enable_monitoring: bool = False,
         enable_encryption: bool = False,
         enable_rls: bool = False,
+        soft_delete: bool = True,
     ) -> None:
         self.logger = setup_logger(self.__class__.__name__)
 
@@ -71,7 +72,6 @@ class PolyDB:
             provider=provider,
             cloud_factory=self.cloud,
             engines=engines,
-            tenant_registry=tenant_registry,
             enable_retries=enable_retries,
             enable_audit=enable_audit,
             enable_audit_reads=enable_audit_reads,
@@ -79,7 +79,10 @@ class PolyDB:
             use_redis_cache=use_redis_cache,
             enable_monitoring=enable_monitoring,
             enable_encryption=enable_encryption,
-            enable_rls=enable_rls,
+            soft_delete=soft_delete,
+            redis_cache_url=redis_cache_url
+            or os.getenv("REDIS_CACHE_URL")
+            or os.getenv("REDIS_URL"),
         )
 
         self.partition_config = partition_config
@@ -123,7 +126,7 @@ class PolyDB:
         return self.cloud.get_object_storage(name)
 
     def get_shared_files(self):
-        return self.cloud.get_shared_files()
+        return self.cloud.get_files()
 
     def get_queue(self):
         return self.cloud.get_queue()
@@ -137,7 +140,7 @@ class PolyDB:
 
     def create(
         self,
-        model: ModelRef,
+        model: str,
         data: JsonDict,
         *,
         engine_override: Optional[EngineOverride] = None,
@@ -146,7 +149,7 @@ class PolyDB:
 
     def read(
         self,
-        model: ModelRef,
+        model: str,
         query: Optional[Lookup] = None,
         *,
         limit: Optional[int] = None,
@@ -169,7 +172,7 @@ class PolyDB:
 
     def read_one(
         self,
-        model: ModelRef,
+        model: str,
         query: Lookup,
         *,
         no_cache: bool = False,
@@ -186,7 +189,7 @@ class PolyDB:
 
     def get(
         self,
-        model: ModelRef,
+        model: str,
         entity_id: Any,
         *,
         include_deleted: bool = False,
@@ -203,7 +206,7 @@ class PolyDB:
 
     def read_page(
         self,
-        model: ModelRef,
+        model: str,
         query: Lookup,
         *,
         page_size: int = 100,
@@ -222,7 +225,7 @@ class PolyDB:
 
     def update(
         self,
-        model: ModelRef,
+        model: str,
         entity_id: Any,
         data: JsonDict,
         *,
@@ -241,7 +244,7 @@ class PolyDB:
 
     def upsert(
         self,
-        model: ModelRef,
+        model: str,
         data: JsonDict,
         *,
         replace: bool = False,
@@ -256,7 +259,7 @@ class PolyDB:
 
     def delete(
         self,
-        model: ModelRef,
+        model: str,
         entity_id: Any,
         *,
         etag: Optional[str] = None,
@@ -278,12 +281,12 @@ class PolyDB:
     def query(self) -> QueryBuilder:
         return QueryBuilder()
 
-    def advanced_query(self) -> AdvancedQueryBuilder:
-        return AdvancedQueryBuilder()
+    def advanced_query(self, table: str) -> AdvancedQueryBuilder:
+        return AdvancedQueryBuilder(table)
 
     def query_linq(
         self,
-        model: ModelRef,
+        model: str,
         builder: QueryBuilder,
         *,
         engine_override: Optional[EngineOverride] = None,
@@ -426,7 +429,7 @@ class PolyDB:
         queue_name: str = "default",
         delay: Optional[int] = None,
     ) -> str:
-        queue = self.get_queue()
+        queue: Any = self.get_queue()
         if hasattr(queue, "publish"):
             return queue.publish(queue_name=queue_name, message=message, delay=delay)
         return self.send_queue(message, queue_name=queue_name)
@@ -438,7 +441,7 @@ class PolyDB:
         max_messages: int = 10,
         wait_seconds: int = 5,
     ) -> List[Dict[str, Any]]:
-        queue = self.get_queue()
+        queue: Any = self.get_queue()
         if hasattr(queue, "consume"):
             return queue.consume(
                 queue_name=queue_name,
@@ -492,7 +495,7 @@ class PolyDB:
 
     def set_cache(
         self,
-        model: ModelRef,
+        model: str,
         key: Any,
         value: Any,
         *,
@@ -504,7 +507,7 @@ class PolyDB:
 
     def get_cache(
         self,
-        model: ModelRef,
+        model: str,
         key: Any,
     ) -> Optional[Any]:
         if not self.cache:
@@ -513,7 +516,7 @@ class PolyDB:
 
     def invalidate_cache(
         self,
-        model: ModelRef,
+        model: str,
         key: Optional[Any] = None,
     ) -> None:
         if not self.cache:
@@ -530,7 +533,7 @@ class PolyDB:
 
     def warm_model_cache(
         self,
-        model: ModelRef,
+        model: str,
         queries: List[Any],
         *,
         ttl: int = 300,
@@ -541,7 +544,7 @@ class PolyDB:
 
     def warm_popular_queries(
         self,
-        model: ModelRef,
+        model: str,
         *,
         limit: int = 20,
         ttl: int = 300,
@@ -554,27 +557,27 @@ class PolyDB:
     # BATCH
     # ============================================================
 
-    def bulk_insert(self, model: ModelRef, records: List[JsonDict]) -> BatchResult:
+    def bulk_insert(self, model: str, records: List[JsonDict]) -> BatchResult:
         return self.batch.bulk_insert(model, records)
 
     def bulk_update(
         self,
-        model: ModelRef,
-        updates: List[Tuple[Any, JsonDict]],
+        model: str,
+        updates: List[Dict[str, Any]],  # {entity_id, data}
     ) -> BatchResult:
         return self.batch.bulk_update(model, updates)
 
-    def bulk_delete(self, model: ModelRef, entity_ids: List[Any]) -> BatchResult:
+    def bulk_delete(self, model: str, entity_ids: List[Any]) -> BatchResult:
         return self.batch.bulk_delete(model, entity_ids)
 
     # ============================================================
     # VALIDATION
     # ============================================================
 
-    def validate_model(self, model: ModelRef):
+    def validate_model(self, model: Type):
         return ModelValidator.validate_model(model)
 
-    def validate(self, model: ModelRef) -> None:
+    def validate(self, model: Type) -> None:
         ModelValidator.validate_and_raise(model)
 
     def validate_data(self, model: Any, data: JsonDict):
@@ -672,11 +675,7 @@ class PolyDB:
         self.rls.add_policy(model, name, policy_func, apply_to)
 
     def set_default_rls_filters(
-        self,
-        model: str,
-        *,
-        read_filters: Optional[Dict[str, Any]] = None,
-        write_filters: Optional[Dict[str, Any]] = None,
+        self, model: str, *, read_filters: Dict[str, Any], write_filters: Dict[str, Any]
     ) -> None:
         if not self.rls:
             raise RuntimeError("RLS is not enabled on this PolyDB instance.")
@@ -687,7 +686,8 @@ class PolyDB:
     # ============================================================
 
     def with_tenant(self, tenant_id: str) -> "PolyDB":
-        TenantContext.set_tenant(tenant_id, self.tenant_registry)
+        if self.tenant_registry:
+            TenantContext.set_tenant(tenant_id, self.tenant_registry)
         return self
 
     def get_tenant(self) -> Optional[TenantConfig]:

{altcodepro_polydb_python-2.3.13 → altcodepro_polydb_python-2.3.15}/src/polydb/adapters/PostgreSQLAdapter.py

@@ -32,25 +32,64 @@ class PostgreSQLAdapter:
         self._lock = threading.Lock()
         self._initialize_pool()
 
+    def _ping_connection(self, conn) -> bool:
+        """Test if connection is still alive"""
+        try:
+            with conn.cursor() as cur:
+                cur.execute("SELECT 1")
+            return True
+        except Exception:
+            return False
+
     def _initialize_pool(self):
         try:
             import psycopg2.pool
+            from urllib.parse import urlparse, parse_qs, urlencode, urlunparse
+
+            # Enhance connection string with better Azure settings
+            dsn = self.connection_string
+            if "postgresql://" in dsn:
+                parsed = urlparse(dsn)
+                query = parse_qs(parsed.query)
+                query.setdefault("connect_timeout", ["30"])
+                query.setdefault("keepalives", ["1"])
+                query.setdefault("keepalives_idle", ["30"])
+                query.setdefault("keepalives_interval", ["10"])
+                query.setdefault("keepalives_count", ["5"])
+
+                new_query = urlencode(query, doseq=True)
+                parsed = parsed._replace(query=new_query)
+                dsn = urlunparse(parsed)
 
             with self._lock:
                 if not self._pool:
                     self._pool = psycopg2.pool.ThreadedConnectionPool(
-                        minconn=int(os.getenv("POSTGRES_MIN_CONNECTIONS", "2")),
-                        maxconn=int(os.getenv("POSTGRES_MAX_CONNECTIONS", "20")),
-                        dsn=self.connection_string,
+                        minconn=int(os.getenv("POSTGRES_MIN_CONNECTIONS", "5")),
+                        maxconn=int(os.getenv("POSTGRES_MAX_CONNECTIONS", "30")),
+                        dsn=dsn,
                     )
-                    self.logger.info("PostgreSQL pool initialized")
+                    self.logger.info("PostgreSQL pool initialized with Azure optimizations")
         except Exception as e:
             raise ConnectionError(f"Failed to initialize PostgreSQL pool: {str(e)}")
 
     def _get_connection(self) -> Any:
         if not self._pool:
             self._initialize_pool()
-        return self._pool.getconn()  # type: ignore
+
+        try:
+            conn = self._pool.getconn()  # type: ignore
+
+            # Critical: Validate connection for Azure transient issues
+            if not self._ping_connection(conn):
+                self.logger.warning("Stale connection detected from pool, closing and retrying")
+                self._pool.putconn(conn, close=True)  # type: ignore
+                conn = self._pool.getconn()  # type: ignore # Get fresh connection
+
+            return conn
+
+        except Exception as e:
+            self.logger.error(f"Failed to acquire connection from pool: {e}")
+            raise ConnectionError(f"Could not obtain database connection: {e}") from e
 
     def _return_connection(self, conn: Any):
         if self._pool and conn:
@@ -59,6 +98,16 @@ class PostgreSQLAdapter:
     # ---------------------------------------------------------------------
     # TRANSACTIONS
     # ---------------------------------------------------------------------
+    def reset_pool(self):
+        """Reset the entire pool (call during startup or after major failures)"""
+        with self._lock:
+            if self._pool:
+                try:
+                    self._pool.closeall()
+                except:
+                    pass
+                self._pool = None
+            self._initialize_pool()
 
     def begin_transaction(self) -> Any:
         """Begin a transaction and return the connection handle."""

{altcodepro_polydb_python-2.3.13 → altcodepro_polydb_python-2.3.15}/src/polydb/base/ObjectStorageAdapter.py

@@ -18,6 +18,7 @@ class ObjectStorageAdapter(ABC):
         optimize: bool = True,
         media_type: Optional[str] = None,
         metadata: Dict[str, Any] | None = None,
+        container_name: Optional[str] = None,
     ) -> str:
         """Store object with optional optimization"""
         if optimize and media_type: