altcodepro-polydb-python 2.1.0__tar.gz → 2.2.0__tar.gz

{altcodepro_polydb_python-2.1.0/src/altcodepro_polydb_python.egg-info → altcodepro_polydb_python-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.1.0
+Version: 2.2.0
 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
@@ -25,6 +25,8 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: psycopg2-binary>=2.9.11
 Requires-Dist: tenacity>=9.1.4
+Requires-Dist: redis>=6.4.0
+Requires-Dist: python-dotenv>=1.1.1
 Provides-Extra: aws
 Requires-Dist: boto3>=1.42.47; extra == "aws"
 Requires-Dist: botocore>=1.42.47; extra == "aws"
@@ -65,7 +67,6 @@ Requires-Dist: google-cloud-storage>=3.9.0; extra == "all"
 Requires-Dist: pymongo>=4.16.0; extra == "all"
 Requires-Dist: pika>=1.3.2; extra == "all"
 Requires-Dist: requests>=2.32.5; extra == "all"
-Requires-Dist: redis>=6.4.0; extra == "all"
 Provides-Extra: dev
 Requires-Dist: black>=26.1.0; extra == "dev"
 Requires-Dist: flake8>=7.3.0; extra == "dev"
@@ -78,7 +79,7 @@ Requires-Dist: pytest-mock>=3.15.1; extra == "test"
 Requires-Dist: moto>=5.1.21; extra == "test"
 Dynamic: license-file
 
-# PolyDB v3.0 - Enterprise Database Abstraction Layer
+# PolyDB v2.2.0 - Enterprise Database Abstraction Layer
 
 **Production-ready, cloud-independent database abstraction with full LINQ support, field-level audit, cache, and overflow storage**
 
@@ -243,8 +244,8 @@ users = db.read(User, {"role": "admin"})
 users = db.read(User, {"role": "admin"}, no_cache=True)
 
 # Manual invalidation
-from polydb.cache import CacheEngine
-cache = CacheEngine()
+from polydb.cache import RedisCacheEngine
+cache = RedisCacheEngine()
 cache.invalidate("User")
 cache.clear()
 ```

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/README.md

@@ -1,4 +1,4 @@
-# PolyDB v3.0 - Enterprise Database Abstraction Layer
+# PolyDB v2.2.0 - Enterprise Database Abstraction Layer
 
 **Production-ready, cloud-independent database abstraction with full LINQ support, field-level audit, cache, and overflow storage**
 
@@ -163,8 +163,8 @@ users = db.read(User, {"role": "admin"})
 users = db.read(User, {"role": "admin"}, no_cache=True)
 
 # Manual invalidation
-from polydb.cache import CacheEngine
-cache = CacheEngine()
+from polydb.cache import RedisCacheEngine
+cache = RedisCacheEngine()
 cache.invalidate("User")
 cache.clear()
 ```

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/example_usage.py

@@ -3,16 +3,14 @@
 Complete PolyDB usage example with all features
 """
 
-from polydb import (
-    DatabaseFactory,
-    polydb_model,
-    QueryBuilder,
-    Operator,
-    AuditContext,
-)
-
 
 # 1. Define models
+from polydb.audit.context import AuditContext
+from polydb.databaseFactory import DatabaseFactory
+from polydb.decorators import polydb_model
+from polydb.query import Operator, QueryBuilder
+
+
 @polydb_model
 class User:
     __polydb__ = {

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "altcodepro-polydb-python"
-version = "2.1.0"
+version = "2.2.0"
 description = "Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety"
-readme = {file = "README.md", content-type = "text/markdown"}
+readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.8"
 license-files = ["LICENSE"]
-authors = [{name = "AltCodePro"}]
+authors = [{ name = "AltCodePro" }]
 keywords = [
     "database",
     "cloud",
@@ -23,7 +23,7 @@ keywords = [
     "postgres",
     "mongodb",
     "dynamodb",
-    "s3"
+    "s3",
 ]
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -43,23 +43,22 @@ classifiers = [
 # Core dependencies - always installed
 dependencies = [
     "psycopg2-binary>=2.9.11",
-    "tenacity>=9.1.4"
+    "tenacity>=9.1.4",
+    "redis>=6.4.0",
+    "python-dotenv>=1.1.1",
 ]
 
 # Generic/Open-source stack (cheapest option)
 [project.optional-dependencies]
 
-aws = [
-    "boto3>=1.42.47",
-    "botocore>=1.42.47"
-]
+aws = ["boto3>=1.42.47", "botocore>=1.42.47"]
 
 azure = [
     "azure-core>=1.38.1",
     "azure-data-tables>=12.7.0",
     "azure-storage-blob>=12.28.0",
     "azure-storage-file-share>=12.24.0",
-    "azure-storage-queue>=12.15.0"
+    "azure-storage-queue>=12.15.0",
 ]
 
 gcp = [
@@ -68,26 +67,16 @@ gcp = [
     "google-cloud-core>=2.5.0",
     "google-cloud-firestore>=2.23.0",
     "google-cloud-pubsub>=2.35.0",
-    "google-cloud-storage>=3.9.0"
+    "google-cloud-storage>=3.9.0",
 ]
 
-mongodb = [
-    "pymongo>=4.16.0"
-]
+mongodb = ["pymongo>=4.16.0"]
 
-rabbitmq = [
-    "pika>=1.3.2"
-]
+rabbitmq = ["pika>=1.3.2"]
 
-vercel = [
-    "requests>=2.32.5"
-]
+vercel = ["requests>=2.32.5"]
 
-generic = [
-    "pymongo>=4.16.0",
-    "pika>=1.3.2",
-    "boto3>=1.42.47"
-]
+generic = ["pymongo>=4.16.0", "pika>=1.3.2", "boto3>=1.42.47"]
 
 all = [
     "boto3>=1.42.47",
@@ -103,17 +92,11 @@ all = [
     "pymongo>=4.16.0",
     "pika>=1.3.2",
     "requests>=2.32.5",
-    "redis>=6.4.0"
 ]
 
 
 # Development dependencies
-dev = [
-    "black>=26.1.0",
-    "flake8>=7.3.0",
-    "isort>=7.0.0",
-    "mypy>=1.19.1"
-]
+dev = ["black>=26.1.0", "flake8>=7.3.0", "isort>=7.0.0", "mypy>=1.19.1"]
 
 
 # Testing with all providers
@@ -121,7 +104,7 @@ test = [
     "pytest>=9.0.2",
     "pytest-cov>=7.0.0",
     "pytest-mock>=3.15.1",
-    "moto>=5.1.21"
+    "moto>=5.1.21",
 ]
 
 [project.urls]
@@ -131,7 +114,7 @@ Repository = "https://github.com/altcodepro/polydb-python"
 "Bug Tracker" = "https://github.com/altcodepro/polydb-python/issues"
 
 [tool.setuptools]
-package-dir = {"" = "src"}
+package-dir = { "" = "src" }
 
 [tool.setuptools.packages.find]
 where = ["src"]
@@ -167,4 +150,4 @@ addopts = [
     "--cov=polydb",
     "--cov-report=term-missing",
     "--cov-report=html",
-]
+]

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0/src/altcodepro_polydb_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.1.0
+Version: 2.2.0
 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
@@ -25,6 +25,8 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: psycopg2-binary>=2.9.11
 Requires-Dist: tenacity>=9.1.4
+Requires-Dist: redis>=6.4.0
+Requires-Dist: python-dotenv>=1.1.1
 Provides-Extra: aws
 Requires-Dist: boto3>=1.42.47; extra == "aws"
 Requires-Dist: botocore>=1.42.47; extra == "aws"
@@ -65,7 +67,6 @@ Requires-Dist: google-cloud-storage>=3.9.0; extra == "all"
 Requires-Dist: pymongo>=4.16.0; extra == "all"
 Requires-Dist: pika>=1.3.2; extra == "all"
 Requires-Dist: requests>=2.32.5; extra == "all"
-Requires-Dist: redis>=6.4.0; extra == "all"
 Provides-Extra: dev
 Requires-Dist: black>=26.1.0; extra == "dev"
 Requires-Dist: flake8>=7.3.0; extra == "dev"
@@ -78,7 +79,7 @@ Requires-Dist: pytest-mock>=3.15.1; extra == "test"
 Requires-Dist: moto>=5.1.21; extra == "test"
 Dynamic: license-file
 
-# PolyDB v3.0 - Enterprise Database Abstraction Layer
+# PolyDB v2.2.0 - Enterprise Database Abstraction Layer
 
 **Production-ready, cloud-independent database abstraction with full LINQ support, field-level audit, cache, and overflow storage**
 
@@ -243,8 +244,8 @@ users = db.read(User, {"role": "admin"})
 users = db.read(User, {"role": "admin"}, no_cache=True)
 
 # Manual invalidation
-from polydb.cache import CacheEngine
-cache = CacheEngine()
+from polydb.cache import RedisCacheEngine
+cache = RedisCacheEngine()
 cache.invalidate("User")
 cache.clear()
 ```

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/altcodepro_polydb_python.egg-info/SOURCES.txt

@@ -23,6 +23,7 @@ src/polydb/databaseFactory.py
 src/polydb/decorators.py
 src/polydb/errors.py
 src/polydb/factory.py
+src/polydb/json_safe.py
 src/polydb/models.py
 src/polydb/monitoring.py
 src/polydb/multitenancy.py

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/altcodepro_polydb_python.egg-info/requires.txt

@@ -1,5 +1,7 @@
 psycopg2-binary>=2.9.11
 tenacity>=9.1.4
+redis>=6.4.0
+python-dotenv>=1.1.1
 
 [all]
 boto3>=1.42.47
@@ -15,7 +17,6 @@ google-cloud-storage>=3.9.0
 pymongo>=4.16.0
 pika>=1.3.2
 requests>=2.32.5
-redis>=6.4.0
 
 [aws]
 boto3>=1.42.47

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/__init__.py

@@ -4,7 +4,7 @@ PolyDB - Enterprise Cloud-Independent Database Abstraction
 Full LINQ support, field-level audit, cache, soft delete, overflow storage
 """
 
-__version__ = "3.0.0"
+__version__ = "2.2.0"
 
 from .factory import CloudDatabaseFactory
 from .databaseFactory import DatabaseFactory
@@ -12,7 +12,7 @@ from .models import CloudProvider, PartitionConfig
 from .decorators import polydb_model
 from .query import QueryBuilder, Operator
 from .audit.context import AuditContext
-from .cache import CacheEngine
+from .cache import RedisCacheEngine as CacheEngine
 from .errors import (
     CloudDBError,
     DatabaseError,
@@ -31,34 +31,30 @@ from .errors import (
 
 __all__ = [
     # Factories
-    'CloudDatabaseFactory',
-    'DatabaseFactory',
-    
+    "CloudDatabaseFactory",
+    "DatabaseFactory",
     # Models & Config
-    'CloudProvider',
-    'PartitionConfig',
-    'polydb_model',
-    
+    "CloudProvider",
+    "PartitionConfig",
+    "polydb_model",
     # Query
-    'QueryBuilder',
-    'Operator',
-    
+    "QueryBuilder",
+    "Operator",
     # Audit & Cache
-    'AuditContext',
-    'CacheEngine',
-    
+    "AuditContext",
+    "CacheEngine",
     # Errors
-    'CloudDBError',
-    'DatabaseError',
-    'NoSQLError',
-    'StorageError',
-    'QueueError',
-    'ConnectionError',
-    'ValidationError',
-    'PolyDBError',
-    'ModelNotRegisteredError',
-    'InvalidModelMetadataError',
-    'UnsupportedStorageTypeError',
-    'AdapterConfigurationError',
-    'OperationNotSupportedError',
-]
+    "CloudDBError",
+    "DatabaseError",
+    "NoSQLError",
+    "StorageError",
+    "QueueError",
+    "ConnectionError",
+    "ValidationError",
+    "PolyDBError",
+    "ModelNotRegisteredError",
+    "InvalidModelMetadataError",
+    "UnsupportedStorageTypeError",
+    "AdapterConfigurationError",
+    "OperationNotSupportedError",
+]

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/AzureQueueAdapter.py

@@ -6,6 +6,8 @@ import os
 import threading
 from typing import Any, Dict, List
 
+from src.polydb.json_safe import json_safe
+
 class AzureQueueAdapter(QueueAdapter):
     """Azure Queue Storage with client reuse"""
 
@@ -36,7 +38,7 @@ class AzureQueueAdapter(QueueAdapter):
 
             if self._client:
                 queue_client = self._client.get_queue_client(queue_name)
-                response = queue_client.send_message(json.dumps(message))
+                response = queue_client.send_message(json.dumps(message,default=json_safe))
                 return response.id
             return ""
         except Exception as e:

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/AzureTableStorageAdapter.py

@@ -3,6 +3,7 @@ import os
 import threading
 from typing import Any, Dict, List, Optional
 from polydb.base.NoSQLKVAdapter import NoSQLKVAdapter
+from src.polydb.json_safe import json_safe
 from ..errors import NoSQLError, ConnectionError
 from ..retry import retry
 from ..types import JsonDict
@@ -56,7 +57,7 @@ class AzureTableStorageAdapter(NoSQLKVAdapter):
             data_copy['RowKey'] = rk
             
             # Check size
-            data_bytes = json.dumps(data_copy).encode()
+            data_bytes = json.dumps(data_copy,default=json_safe).encode()
             data_size = len(data_bytes)
             
             if data_size > self.AZURE_TABLE_MAX_SIZE:

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/DynamoDBAdapter.py

@@ -64,7 +64,7 @@ class DynamoDBAdapter(NoSQLKVAdapter):
             data_copy['SK'] = rk
             
             # Check size
-            data_bytes = json.dumps(data_copy).encode()
+            data_bytes = json.dumps(data_copy,default=json_safe).encode()
             data_size = len(data_bytes)
             
             if data_size > self.DYNAMODB_MAX_SIZE:

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/FirestoreAdapter.py

@@ -6,6 +6,7 @@ from google.cloud import firestore
 from google.cloud import storage
 from google.cloud.firestore import Client
 from polydb.base.NoSQLKVAdapter import NoSQLKVAdapter
+from src.polydb.json_safe import json_safe
 
 from ..errors import NoSQLError, ConnectionError
 from ..retry import retry
@@ -66,7 +67,7 @@ class FirestoreAdapter(NoSQLKVAdapter):
             data_copy['_rk'] = rk
             
             # Check size
-            data_bytes = json.dumps(data_copy).encode()
+            data_bytes = json.dumps(data_copy,default=json_safe).encode()
             data_size = len(data_bytes)
             
             if data_size > self.FIRESTORE_MAX_SIZE:

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/PostgreSQLAdapter.py

@@ -21,8 +21,10 @@ class PostgreSQLAdapter:
         self.logger = setup_logger(__name__)
         self.connection_string = os.getenv(
             "POSTGRES_CONNECTION_STRING",
-            os.getenv("POSTGRES_URL", "postgresql://user:password@localhost:5432/database"),
+            os.getenv("POSTGRES_URL", ""),
         )
+        if not self.connection_string:
+            raise ConnectionError("POSTGRES_CONNECTION_STRING or POSTGRES_URL must be set")
         self._pool = None
         self._lock = threading.Lock()
         self._initialize_pool()
@@ -42,24 +44,46 @@ class PostgreSQLAdapter:
         except Exception as e:
             raise ConnectionError(f"Failed to initialize PostgreSQL pool: {str(e)}")
 
-    def _get_connection(self):
+    def _get_connection(self) -> Any:
         if not self._pool:
             self._initialize_pool()
         return self._pool.getconn()  # type: ignore
 
-    def _return_connection(self, conn):
+    def _return_connection(self, conn: Any):
         if self._pool and conn:
             self._pool.putconn(conn)
 
+    def begin_transaction(self) -> Any:
+        """Begin a transaction and return the connection handle."""
+        conn = self._get_connection()
+        conn.autocommit = False  # Ensure transaction mode
+        return conn
+
+    def commit(self, tx: Any):
+        """Commit the transaction using the provided connection."""
+        if tx:
+            tx.commit()
+            self._return_connection(tx)
+
+    def rollback(self, tx: Any):
+        """Rollback the transaction using the provided connection."""
+        if tx:
+            tx.rollback()
+            self._return_connection(tx)
+
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
-    def insert(self, table: str, data: JsonDict) -> JsonDict:
+    def insert(self, table: str, data: JsonDict, tx: Optional[Any] = None) -> JsonDict:
         table = validate_table_name(table)
         for k in data.keys():
             validate_column_name(k)
 
-        conn = None
-        try:
+        conn = tx
+        own_conn = False
+        if not conn:
             conn = self._get_connection()
+            own_conn = True
+
+        try:
             cursor = conn.cursor()
 
             columns = ", ".join(data.keys())
@@ -71,15 +95,17 @@ class PostgreSQLAdapter:
             columns_list = [desc[0] for desc in cursor.description]
             result = dict(zip(columns_list, result_row))
 
-            conn.commit()
+            if own_conn:
+                conn.commit()
+
             cursor.close()
             return result
         except Exception as e:
-            if conn:
+            if own_conn:
                 conn.rollback()
             raise DatabaseError(f"Insert failed: {str(e)}")
         finally:
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
@@ -89,12 +115,16 @@ class PostgreSQLAdapter:
         query: Optional[Lookup] = None,
         limit: Optional[int] = None,
         offset: Optional[int] = None,
+        tx: Optional[Any] = None,
     ) -> List[JsonDict]:
         table = validate_table_name(table)
-        conn = None
+        conn = tx
+        own_conn = False
+        if not conn:
+            conn = self._get_connection()
+            own_conn = True
 
         try:
-            conn = self._get_connection()
             cursor = conn.cursor()
 
             sql = f"SELECT * FROM {table}"
@@ -131,15 +161,20 @@ class PostgreSQLAdapter:
         except Exception as e:
             raise DatabaseError(f"Select failed: {str(e)}")
         finally:
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
     def select_page(
-        self, table: str, query: Lookup, page_size: int, continuation_token: Optional[str] = None
+        self,
+        table: str,
+        query: Lookup,
+        page_size: int,
+        continuation_token: Optional[str] = None,
+        tx: Optional[Any] = None,
     ) -> Tuple[List[JsonDict], Optional[str]]:
         offset = int(continuation_token) if continuation_token else 0
-        results = self.select(table, query, limit=page_size + 1, offset=offset)
+        results = self.select(table, query, limit=page_size + 1, offset=offset, tx=tx)
 
         has_more = len(results) > page_size
         if has_more:
@@ -149,14 +184,24 @@ class PostgreSQLAdapter:
         return results, next_token
 
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
-    def update(self, table: str, entity_id: Union[Any, Lookup], data: JsonDict) -> JsonDict:
+    def update(
+        self,
+        table: str,
+        entity_id: Union[Any, Lookup],
+        data: JsonDict,
+        tx: Optional[Any] = None,
+    ) -> JsonDict:
         table = validate_table_name(table)
         for k in data.keys():
             validate_column_name(k)
 
-        conn = None
-        try:
+        conn = tx
+        own_conn = False
+        if not conn:
             conn = self._get_connection()
+            own_conn = True
+
+        try:
             cursor = conn.cursor()
 
             set_clause = ", ".join([f"{k} = %s" for k in data.keys()])
@@ -183,26 +228,32 @@ class PostgreSQLAdapter:
             columns = [desc[0] for desc in cursor.description]
             result = dict(zip(columns, result_row))
 
-            conn.commit()
+            if own_conn:
+                conn.commit()
+
             cursor.close()
             return result
         except Exception as e:
-            if conn:
+            if own_conn:
                 conn.rollback()
             raise DatabaseError(f"Update failed: {str(e)}")
         finally:
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
-    def upsert(self, table: str, data: JsonDict) -> JsonDict:
+    def upsert(self, table: str, data: JsonDict, tx: Optional[Any] = None) -> JsonDict:
         table = validate_table_name(table)
         for k in data.keys():
             validate_column_name(k)
 
-        conn = None
-        try:
+        conn = tx
+        own_conn = False
+        if not conn:
             conn = self._get_connection()
+            own_conn = True
+
+        try:
             cursor = conn.cursor()
 
             columns = ", ".join(data.keys())
@@ -226,24 +277,31 @@ class PostgreSQLAdapter:
             columns_list = [desc[0] for desc in cursor.description]
             result = dict(zip(columns_list, result_row))
 
-            conn.commit()
+            if own_conn:
+                conn.commit()
+
             cursor.close()
             return result
         except Exception as e:
-            if conn:
+            if own_conn:
                 conn.rollback()
             raise DatabaseError(f"Upsert failed: {str(e)}")
         finally:
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
-    def delete(self, table: str, entity_id: Union[Any, Lookup]) -> JsonDict:
+    def delete(
+        self, table: str, entity_id: Union[Any, Lookup], tx: Optional[Any] = None
+    ) -> JsonDict:
         table = validate_table_name(table)
-        conn = None
+        conn = tx
+        own_conn = False
+        if not conn:
+            conn = self._get_connection()
+            own_conn = True
 
         try:
-            conn = self._get_connection()
             cursor = conn.cursor()
 
             params = []
@@ -268,24 +326,31 @@ class PostgreSQLAdapter:
             columns = [desc[0] for desc in cursor.description]
             result = dict(zip(columns, result_row))
 
-            conn.commit()
+            if own_conn:
+                conn.commit()
+
             cursor.close()
             return result
         except Exception as e:
-            if conn:
+            if own_conn:
                 conn.rollback()
             raise DatabaseError(f"Delete failed: {str(e)}")
         finally:
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     @retry(max_attempts=3, delay=1.0, exceptions=(DatabaseError,))
-    def query_linq(self, table: str, builder: QueryBuilder) -> Union[List[JsonDict], int]:
+    def query_linq(
+        self, table: str, builder: QueryBuilder, tx: Optional[Any] = None
+    ) -> Union[List[JsonDict], int]:
         table = validate_table_name(table)
-        conn = None
+        conn = tx
+        own_conn = False
+        if not conn:
+            conn = self._get_connection()
+            own_conn = True
 
         try:
-            conn = self._get_connection()
             cursor = conn.cursor()
 
             if builder.count_only:
@@ -336,17 +401,24 @@ class PostgreSQLAdapter:
             cursor.execute(sql, params)
 
             if builder.count_only:
-                return cursor.fetchone()[0]
+                result = cursor.fetchone()[0]
+            else:
+                columns = [desc[0] for desc in cursor.description]
+                results = [dict(zip(columns, row)) for row in cursor.fetchall()]
+                result = results
 
-            columns = [desc[0] for desc in cursor.description]
-            results = [dict(zip(columns, row)) for row in cursor.fetchall()]
             cursor.close()
 
-            return results
+            if own_conn:
+                conn.commit()
+
+            return result
         except Exception as e:
+            if own_conn:
+                conn.rollback()
             raise DatabaseError(f"LINQ query failed: {str(e)}")
         finally:
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     def __del__(self):
@@ -361,14 +433,19 @@ class PostgreSQLAdapter:
         self,
         sql: str,
         params: Optional[List[Any]] = None,
+        tx: Optional[Any] = None,
         *,
         fetch: bool = False,
         fetch_one: bool = False,
     ) -> Union[None, JsonDict, List[JsonDict]]:
-        conn = None
+        conn = tx
+        own_conn = False
+        if not conn:
+            conn = self._get_connection()
+            own_conn = True
+
         cursor = None
         try:
-            conn = self._get_connection()
             cursor = conn.cursor()
 
             self.logger.debug("Executing raw SQL: %s", sql)
@@ -381,22 +458,25 @@ class PostgreSQLAdapter:
                 if row:
                     columns = [desc[0] for desc in cursor.description]
                     result = dict(zip(columns, row))
-                conn.commit()
+                if own_conn:
+                    conn.commit()
                 return result
 
             if fetch:
                 rows = cursor.fetchall()
                 columns = [desc[0] for desc in cursor.description]
                 results = [dict(zip(columns, r)) for r in rows]
-                conn.commit()
+                if own_conn:
+                    conn.commit()
                 return results
 
             # Non-fetch execution (DDL/DML)
-            conn.commit()
+            if own_conn:
+                conn.commit()
             return None
 
         except Exception as e:
-            if conn:
+            if own_conn:
                 try:
                     conn.rollback()
                 except Exception:
@@ -409,7 +489,7 @@ class PostgreSQLAdapter:
                     cursor.close()
                 except Exception:
                     pass
-            if conn:
+            if own_conn and conn:
                 self._return_connection(conn)
 
     @contextmanager

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/PubSubAdapter.py

@@ -7,6 +7,8 @@ import os
 import threading
 from typing import Any, Dict, List
 
+from src.polydb.json_safe import json_safe
+
 class PubSubAdapter(QueueAdapter):
     """GCP Pub/Sub with client reuse"""
 
@@ -42,7 +44,7 @@ class PubSubAdapter(QueueAdapter):
                 topic_path = self._publisher.topic_path(
                     self.project_id, queue_name or self.topic_name
                 )
-                data = json.dumps(message).encode("utf-8")
+                data = json.dumps(message,default=json_safe).encode("utf-8")
                 future = self._publisher.publish(topic_path, data)
                 return future.result()
             return ""

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/SQSAdapter.py

@@ -12,6 +12,8 @@ import os
 import threading
 from typing import Any, Dict, List
 
+from src.polydb.json_safe import json_safe
+
 
 class SQSAdapter(QueueAdapter):
     """AWS SQS with client reuse"""
@@ -45,7 +47,7 @@ class SQSAdapter(QueueAdapter):
                 self._initialize_client()
             if self._client:
                 response = self._client.send_message(
-                    QueueUrl=self.queue_url, MessageBody=json.dumps(message)
+                    QueueUrl=self.queue_url, MessageBody=json.dumps(message,default=json_safe)
                 )
                 return response["MessageId"]
             return ""

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/adapters/VercelKVAdapter.py

@@ -3,6 +3,7 @@ import os
 import threading
 from typing import Any, Dict, List, Optional
 from polydb.base.NoSQLKVAdapter import NoSQLKVAdapter
+from src.polydb.json_safe import json_safe
 from ..errors import NoSQLError, StorageError
 from ..retry import retry
 from ..types import JsonDict
@@ -36,7 +37,7 @@ class VercelKVAdapter(NoSQLKVAdapter):
             data_copy['_rk'] = rk
             
             # Check size
-            data_bytes = json.dumps(data_copy).encode()
+            data_bytes = json.dumps(data_copy,default=json_safe).encode()
             data_size = len(data_bytes)
             
             if data_size > self.VERCEL_KV_MAX_SIZE:
@@ -69,7 +70,7 @@ class VercelKVAdapter(NoSQLKVAdapter):
                 response = requests.post(
                     f"{self.kv_url}/set/{key}",
                     headers={'Authorization': f'Bearer {self.kv_token}'},
-                    json={'value': json.dumps(reference_data)},
+                    json={'value': json.dumps(reference_data,default=json_safe)},
                     timeout=self.timeout
                 )
                 response.raise_for_status()
@@ -78,7 +79,7 @@ class VercelKVAdapter(NoSQLKVAdapter):
                 response = requests.post(
                     f"{self.kv_url}/set/{key}",
                     headers={'Authorization': f'Bearer {self.kv_token}'},
-                    json={'value': json.dumps(data_copy)},
+                    json={'value': json.dumps(data_copy,default=json_safe)},
                     timeout=self.timeout
                 )
                 response.raise_for_status()

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/audit/models.py

@@ -7,6 +7,8 @@ import uuid
 import hashlib
 import json
 
+from src.polydb.json_safe import json_safe
+
 
 @dataclass
 class AuditRecord:
@@ -80,7 +82,7 @@ class AuditRecord:
         )
 
         record.hash = hashlib.sha256(
-            json.dumps(asdict(record), sort_keys=True).encode()
+            json.dumps(asdict(record), sort_keys=True,default=json_safe).encode()
         ).hexdigest()
 
         return record

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/base/NoSQLKVAdapter.py

@@ -6,6 +6,8 @@ import json
 import threading
 from typing import Any, Dict, List, Optional, Tuple, Union, TYPE_CHECKING
 
+from src.polydb.json_safe import json_safe
+
 from ..errors import NoSQLError, StorageError
 from ..retry import retry
 from ..query import QueryBuilder, Operator
@@ -36,7 +38,7 @@ class NoSQLKVAdapter:
             try:
                 pk = self.partition_config.partition_key_template.format(**data)
             except KeyError:
-                pk = f"default_{data.get(pk_field, hashlib.md5(json.dumps(data, sort_keys=True).encode()).hexdigest()[:8])}"
+                pk = f"default_{data.get(pk_field, hashlib.md5(json.dumps(data, sort_keys=True,default=json_safe).encode()).hexdigest()[:8])}"
         else:
             pk = str(data.get(pk_field, 'default'))
         
@@ -46,16 +48,16 @@ class NoSQLKVAdapter:
             try:
                 rk = self.partition_config.row_key_template.format(**data)
             except KeyError:
-                rk = hashlib.md5(json.dumps(data, sort_keys=True).encode()).hexdigest()
+                rk = hashlib.md5(json.dumps(data, sort_keys=True,default=json_safe).encode()).hexdigest()
         else:
-            rk = data.get('id', hashlib.md5(json.dumps(data, sort_keys=True).encode()).hexdigest())
+            rk = data.get('id', hashlib.md5(json.dumps(data, sort_keys=True,default=json_safe).encode()).hexdigest())
         
         return str(pk), str(rk)
     
     @retry(max_attempts=3, delay=1.0, exceptions=(NoSQLError,))
     def _check_overflow(self, data: JsonDict) -> Tuple[JsonDict, Optional[str]]:
         """Check size and store in blob if needed"""
-        data_bytes = json.dumps(data).encode()
+        data_bytes = json.dumps(data,default=json_safe).encode()
         data_size = len(data_bytes)
         
         if data_size > self.max_size:
@@ -292,7 +294,7 @@ class NoSQLKVAdapter:
             seen = set()
             unique = []
             for r in results:
-                key = json.dumps(r, sort_keys=True)
+                key = json.dumps(r, sort_keys=True,default=json_safe)
                 if key not in seen:
                     seen.add(key)
                     unique.append(r)

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/cache.py

@@ -8,6 +8,7 @@ import hashlib
 import threading
 from enum import Enum
 import redis
+from src.polydb.json_safe import json_safe
 
 class CacheStrategy(Enum):
     """Cache invalidation strategies"""
@@ -58,7 +59,7 @@ class RedisCacheEngine:
     
     def _make_key(self, model: str, query: Dict[str, Any]) -> str:
         """Generate cache key"""
-        query_str = json.dumps(query, sort_keys=True)
+        query_str = json.dumps(query, sort_keys=True,default=json_safe)
         query_hash = hashlib.md5(query_str.encode()).hexdigest()
         return f"{self.prefix}{model}:{query_hash}"
     
@@ -94,7 +95,7 @@ class RedisCacheEngine:
         ttl = ttl or self.default_ttl
         
         try:
-            data = json.dumps(value)
+            data = json.dumps(value,default=json_safe)
             self._client.setex(key, ttl, data)
             
             # Initialize access count

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/databaseFactory.py

@@ -17,6 +17,7 @@ from .types import JsonDict, Lookup, ModelMeta
 from .audit.manager import AuditManager
 from .audit.context import AuditContext
 from .query import Operator, QueryBuilder
+from dotenv import load_dotenv
 
 logger = logging.getLogger(__name__)
 
@@ -65,6 +66,11 @@ class DatabaseFactory:
         # Redis cache (only if explicitly enabled + URL present)
         self._cache: Optional[RedisCacheEngine] = None
         self.cache_warmer: Optional[CacheWarmer] = None
+        try:
+            load_dotenv()
+        except Exception:
+            pass
+
         if enable_cache and use_redis_cache:
             redis_url = os.getenv("REDIS_CACHE_URL")
             if redis_url:
@@ -108,17 +114,28 @@ class DatabaseFactory:
         return model.__name__ if isinstance(model, type) else str(model)
 
     def _current_tenant_id(self) -> Optional[str]:
-        tenant = TenantContext.get_tenant()
-        return tenant.tenant_id if tenant else None
+        # Prefer TenantContext if present
+        try:
+            tenant = TenantContext.get_tenant()
+            if tenant and tenant.tenant_id:
+                return tenant.tenant_id
+        except Exception:
+            pass
+
+        # Fallback to AuditContext
+        return AuditContext.tenant_id.get()
 
     def _current_actor_id(self) -> Optional[str]:
         return AuditContext.actor_id.get()
 
     def _inject_tenant(self, data: JsonDict) -> JsonDict:
         tenant_id = self._current_tenant_id()
-        if tenant_id and "tenant_id" not in data:
-            data = dict(data)
-            data["tenant_id"] = tenant_id
+
+        if not tenant_id:
+            raise ValueError("Tenant ID is required but not set in AuditContext or TenantContext")
+
+        data = dict(data)
+        data.setdefault("tenant_id", tenant_id)
         return data
 
     def _inject_audit_fields(self, data: JsonDict, is_create: bool = False) -> JsonDict:
@@ -293,15 +310,17 @@ class DatabaseFactory:
         meta = self._meta(model)
         tenant_id = self._current_tenant_id()
         actor_id = self._current_actor_id()
-
         query = self._apply_soft_delete_filter(query if not include_deleted else None)
-
         # Multi-tenancy & RLS filters
         if self.tenant_enforcer:
             query = self.tenant_enforcer.enforce_read(model_name, query or {})
         if self.rls:
             query = self.rls.enforce_read(model_name, query or {})
-
+        # Inject tenant filter (mandatory isolation)
+        tenant_id = self._current_tenant_id()
+        if tenant_id:
+            query = dict(query or {})
+            query.setdefault("tenant_id", tenant_id)
         use_external_cache = self._enable_cache and self._cache and getattr(meta, "cache", False)
         encrypted_fields = getattr(meta, "encrypted_fields", [])
 
@@ -408,7 +427,11 @@ class DatabaseFactory:
             query = self.tenant_enforcer.enforce_read(model_name, query or {})
         if self.rls:
             query = self.rls.enforce_read(model_name, query or {})
-
+        # Inject tenant filter (mandatory isolation)
+        tenant_id = self._current_tenant_id()
+        if tenant_id:
+            query = dict(query or {})
+            query.setdefault("tenant_id", tenant_id)
         encrypted_fields = getattr(meta, "encrypted_fields", [])
 
         def _op() -> Tuple[List[JsonDict], Optional[str]]:
@@ -497,6 +520,9 @@ class DatabaseFactory:
             nonlocal after_plain, success
 
             if meta.storage == "sql" and meta.table:
+                if tenant_id:
+                    if isinstance(entity_id, dict):
+                        entity_id.setdefault("tenant_id", tenant_id)
                 result = self._sql.update(meta.table, entity_id, data)
             else:
                 model_type = self._model_type(model)

altcodepro_polydb_python-2.2.0/src/polydb/json_safe.py

@@ -0,0 +1,8 @@
+import json
+from datetime import datetime
+
+def json_safe(obj):
+    if isinstance(obj, datetime):
+        return obj.isoformat()
+    return str(obj)
+

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/multitenancy.py

@@ -4,7 +4,7 @@ Multi-tenancy enforcement and isolation
 """
 from typing import Dict, Any, List, Optional, Callable
 from contextvars import ContextVar
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from enum import Enum
 
 
@@ -24,7 +24,7 @@ class TenantConfig:
     database_name: Optional[str] = None
     max_connections: int = 10
     storage_quota_gb: Optional[float] = None
-    features: List[str] = []
+    features: List[str] = field(default_factory=list)
 
 
 class TenantRegistry:

{altcodepro_polydb_python-2.1.0 → altcodepro_polydb_python-2.2.0}/src/polydb/security.py

@@ -11,6 +11,8 @@ import json
 from functools import wraps
 import logging
 
+from src.polydb.json_safe import json_safe
+
 logger = logging.getLogger(__name__)
 
 
@@ -48,7 +50,7 @@ class FieldEncryption:
         """Encrypt arbitrary value (serialize if non-str)"""
         if value is None:
             return ""
-        data = json.dumps(value) if not isinstance(value, str) else value
+        data = json.dumps(value,default=json_safe) if not isinstance(value, str) else value
         try:
             from cryptography.hazmat.primitives.ciphers.aead import AESGCM