async-easy-model 0.2.8__py3-none-any.whl → 0.2.9__py3-none-any.whl

async_easy_model/__init__.py

@@ -12,7 +12,7 @@ from typing import Optional, Any
 from .model import EasyModel, init_db, db_config
 from sqlmodel import Field, Relationship as SQLModelRelationship
 
-__version__ = "0.2.8"
+__version__ = "0.2.9"
 __all__ = ["EasyModel", "init_db", "db_config", "Field", "Relationship", "Relation", "enable_auto_relationships", "disable_auto_relationships", "process_auto_relationships", "MigrationManager", "check_and_migrate_models", "ModelVisualizer"]
 
 # Create a more user-friendly Relationship function

async_easy_model/model.py

@@ -73,15 +73,21 @@ class DatabaseConfig:
     def get_engine(self):
         """Get or create the SQLAlchemy engine."""
         if DatabaseConfig._engine is None:
-            kwargs = {}
+            # Apply connection pool configuration to all database types
+            # to prevent connection leaks and ensure proper resource management
+            kwargs = {
+                "pool_size": 10,        # Base number of connections in the pool
+                "max_overflow": 30,     # Additional connections allowed beyond pool_size
+                "pool_timeout": 30,     # Timeout in seconds for getting connection from pool
+                "pool_recycle": 1800,   # Recycle connections after 30 minutes
+                "pool_pre_ping": True,  # Verify connections before use
+            }
+            
+            # PostgreSQL-specific optimizations (if needed in the future)
             if self.db_type == "postgresql":
-                kwargs.update({
-                    "pool_size": 10,
-                    "max_overflow": 30,
-                    "pool_timeout": 30,
-                    "pool_recycle": 1800,
-                    "pool_pre_ping": True,
-                })
+                # PostgreSQL already has good defaults above
+                pass
+                
             DatabaseConfig._engine = create_async_engine(
                 self.get_connection_url(),
                 **kwargs
@@ -115,9 +121,24 @@ class EasyModel(SQLModel):
     @classmethod
     @contextlib.asynccontextmanager
     async def get_session(cls):
-        """Provide a transactional scope for database operations."""
-        async with db_config.get_session_maker()() as session:
+        """Provide a transactional scope for database operations.
+        
+        This method ensures proper session cleanup by:
+        - Explicitly rolling back transactions on exceptions
+        - Explicitly closing sessions in all cases
+        - Proper exception propagation
+        """
+        session = None
+        try:
+            session = db_config.get_session_maker()()
             yield session
+        except Exception:
+            if session:
+                await session.rollback()
+            raise
+        finally:
+            if session:
+                await session.close()
 
     @classmethod
     def _get_relationship_fields(cls) -> List[str]:

{async_easy_model-0.2.8.dist-info → async_easy_model-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-easy-model
-Version: 0.2.8
+Version: 0.2.9
 Summary: A simplified SQLModel-based ORM for async database operations
 Home-page: https://github.com/puntorigen/easy-model
 Author: Pablo Schaffner

async_easy_model-0.2.9.dist-info/RECORD

@@ -0,0 +1,11 @@
+async_easy_model/__init__.py,sha256=eGTq3OypupyVvLe2Zyn3ZFBeuU4v8EXW-XQ047vUMoM,1921
+async_easy_model/auto_relationships.py,sha256=V2LAzNi7y-keFk4C_m-byVRM-k_7nL5HEy9Ig3nEdq8,27756
+async_easy_model/migrations.py,sha256=rYDGCGlruSugAmPfdIF2-uhyG6UvC_2qbF3BXJ084qI,17803
+async_easy_model/model.py,sha256=pBWRAVhazZuI_rBOnILPttiCF_ZEomWBJGDmfdFp8Nk,64243
+async_easy_model/relationships.py,sha256=vR5BsJpGaDcecCcNlg9-ouZfxFXFQv5kOyiXhKp_T7A,3286
+async_easy_model/visualization.py,sha256=RVCdc8j3uUQe-zy3jXju_yhA13qJ8KWVbQ5fQyjyqkA,29973
+async_easy_model-0.2.9.dist-info/licenses/LICENSE,sha256=uwDkl6oHbRltW7vYKNc4doJyhtwhyrSNFFlPpKATwLE,1072
+async_easy_model-0.2.9.dist-info/METADATA,sha256=iVRM1qe6DfdFUSa5FYYES-oXgkKScm7m2BVClhlH-k8,12888
+async_easy_model-0.2.9.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+async_easy_model-0.2.9.dist-info/top_level.txt,sha256=e5_47sGmJnyxz2msfwU6C316EqmrSd9RGIYwZyWx68E,17
+async_easy_model-0.2.9.dist-info/RECORD,,

{async_easy_model-0.2.8.dist-info → async_easy_model-0.2.9.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.1)
+Generator: setuptools (77.0.3)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

async_easy_model-0.2.8.dist-info/RECORD

@@ -1,11 +0,0 @@
-async_easy_model/__init__.py,sha256=8wTdiTMrhwZjclRALXktyRIBJ4ONt-F0JG_CK60yz9A,1921
-async_easy_model/auto_relationships.py,sha256=V2LAzNi7y-keFk4C_m-byVRM-k_7nL5HEy9Ig3nEdq8,27756
-async_easy_model/migrations.py,sha256=rYDGCGlruSugAmPfdIF2-uhyG6UvC_2qbF3BXJ084qI,17803
-async_easy_model/model.py,sha256=f0eMcIaDOz9s01A4jpQ-T_VpCgFt67JF1puDYuhpdv4,63290
-async_easy_model/relationships.py,sha256=vR5BsJpGaDcecCcNlg9-ouZfxFXFQv5kOyiXhKp_T7A,3286
-async_easy_model/visualization.py,sha256=RVCdc8j3uUQe-zy3jXju_yhA13qJ8KWVbQ5fQyjyqkA,29973
-async_easy_model-0.2.8.dist-info/licenses/LICENSE,sha256=uwDkl6oHbRltW7vYKNc4doJyhtwhyrSNFFlPpKATwLE,1072
-async_easy_model-0.2.8.dist-info/METADATA,sha256=29ESU7RvrK94YP0v2iwy_v8f06nFLs96D3hPJJibO1U,12888
-async_easy_model-0.2.8.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-async_easy_model-0.2.8.dist-info/top_level.txt,sha256=e5_47sGmJnyxz2msfwU6C316EqmrSd9RGIYwZyWx68E,17
-async_easy_model-0.2.8.dist-info/RECORD,,