async-easy-model 0.2.8__py3-none-any.whl → 0.3.0__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.3.0"
 __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

@@ -14,6 +14,9 @@ import re
 
 T = TypeVar("T", bound="EasyModel")
 
+# Global database configuration instance (forward declaration)
+db_config = None
+
 class DatabaseConfig:
     _engine = None
     _session_maker = None
@@ -73,15 +76,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
@@ -101,13 +110,74 @@ class DatabaseConfig:
 # Global database configuration instance.
 db_config = DatabaseConfig()
 
+def _normalize_datetime_for_db(value: Any) -> Any:
+    """
+    Normalize datetime values for database compatibility.
+    
+    For PostgreSQL with TIMESTAMP WITHOUT TIME ZONE columns, converts timezone-aware
+    datetimes to timezone-naive UTC datetimes. For other databases, returns the value unchanged.
+    
+    Args:
+        value: The value to potentially normalize
+        
+    Returns:
+        The normalized value
+    """
+    global db_config
+    
+    # Only process datetime objects
+    if not isinstance(value, datetime):
+        return value
+    
+    # Only normalize for PostgreSQL
+    if db_config and db_config.db_type == "postgresql":
+        # If the datetime is timezone-aware, convert to UTC and make naive
+        if value.tzinfo is not None:
+            return value.astimezone(tz.utc).replace(tzinfo=None)
+    
+    return value
+
+def _normalize_data_for_db(data: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Normalize all datetime values in a data dictionary for database compatibility.
+    
+    Args:
+        data: Dictionary containing field values
+        
+    Returns:
+        Dictionary with normalized datetime values
+    """
+    normalized_data = {}
+    for key, value in data.items():
+        normalized_data[key] = _normalize_datetime_for_db(value)
+    return normalized_data
+
+def _get_normalized_datetime() -> datetime:
+    """
+    Get a datetime that's appropriate for the current database backend.
+    
+    Returns:
+        For PostgreSQL: timezone-naive UTC datetime
+        For SQLite: timezone-aware UTC datetime (for backward compatibility)
+    """
+    global db_config
+    
+    utc_now = datetime.now(tz.utc)
+    
+    # For PostgreSQL, return timezone-naive datetime
+    if db_config and db_config.db_type == "postgresql":
+        return utc_now.replace(tzinfo=None)
+    
+    # For SQLite and others, return timezone-aware datetime (backward compatibility)
+    return utc_now
+
 class EasyModel(SQLModel):
     """
     Base model class providing common async database operations.
     """
     id: Optional[int] = Field(default=None, primary_key=True)
-    created_at: Optional[datetime] = Field(default_factory=lambda: datetime.now(tz.utc))
-    updated_at: Optional[datetime] = Field(default_factory=lambda: datetime.now(tz.utc))
+    created_at: Optional[datetime] = Field(default_factory=_get_normalized_datetime)
+    updated_at: Optional[datetime] = Field(default_factory=_get_normalized_datetime)
 
     # Default table args with extend_existing=True to ensure all subclasses can redefine tables
     __table_args__ = {"extend_existing": True}
@@ -115,9 +185,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]:
@@ -416,6 +501,9 @@ class EasyModel(SQLModel):
             try:
                 processed_data = await cls._process_relationships_for_insert(session, data)
                 
+                # Normalize datetime values for database compatibility
+                processed_data = _normalize_data_for_db(processed_data)
+                
                 # Create the model instance
                 obj = cls(**processed_data)
                 session.add(obj)
@@ -748,9 +836,10 @@ class EasyModel(SQLModel):
                             if existing:
                                 raise ValueError(f"Cannot update {field_name} to '{new_value}': value already exists")
                 
-                # Apply the updates
+                # Apply the updates with datetime normalization
                 for key, value in data.items():
-                    setattr(record, key, value)
+                    normalized_value = _normalize_datetime_for_db(value)
+                    setattr(record, key, normalized_value)
                 
                 # Process many-to-many relationships if any
                 for rel_name, rel_data in many_to_many_data.items():
@@ -1362,7 +1451,7 @@ class EasyModel(SQLModel):
 def _update_updated_at(session, flush_context, instances):
     for instance in session.dirty:
         if isinstance(instance, EasyModel) and hasattr(instance, "updated_at"):
-            instance.updated_at = datetime.now(tz.utc)
+            instance.updated_at = _get_normalized_datetime()
 
 async def init_db(migrate: bool = True, model_classes: List[Type[SQLModel]] = None):
     """

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-easy-model
-Version: 0.2.8
+Version: 0.3.0
 Summary: A simplified SQLModel-based ORM for async database operations
 Home-page: https://github.com/puntorigen/easy-model
 Author: Pablo Schaffner
@@ -59,6 +59,7 @@ A simplified SQLModel-based ORM for async database operations in Python. async-e
 - 🛠️ Built on top of SQLModel and SQLAlchemy for robust performance
 - 📝 Type hints for better IDE support
 - 🕒 Automatic `id`, `created_at` and `updated_at` fields provided by default
+- ⏰ **PostgreSQL DateTime Compatibility**: Automatic timezone-aware to timezone-naive datetime conversion for PostgreSQL TIMESTAMP WITHOUT TIME ZONE columns
 - 🔄 Automatic schema migrations for evolving database models
 - 📊 Visualization of database schema using Mermaid ER diagrams
 

async_easy_model-0.3.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+async_easy_model/__init__.py,sha256=IXheLaQx-dpGPL6oBEaaIftZDihlFB7bwzEWfMlZ4Ts,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=XxlFOYiX8HT1B6PBB3RubuvB5j4aYKEyO-sPA202Y2U,66508
+async_easy_model/relationships.py,sha256=vR5BsJpGaDcecCcNlg9-ouZfxFXFQv5kOyiXhKp_T7A,3286
+async_easy_model/visualization.py,sha256=RVCdc8j3uUQe-zy3jXju_yhA13qJ8KWVbQ5fQyjyqkA,29973
+async_easy_model-0.3.0.dist-info/licenses/LICENSE,sha256=uwDkl6oHbRltW7vYKNc4doJyhtwhyrSNFFlPpKATwLE,1072
+async_easy_model-0.3.0.dist-info/METADATA,sha256=0PEppZL9WiSl1uKDqlC-f-l9rrrM-x5HVcRWBcRgsJU,13047
+async_easy_model-0.3.0.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+async_easy_model-0.3.0.dist-info/top_level.txt,sha256=e5_47sGmJnyxz2msfwU6C316EqmrSd9RGIYwZyWx68E,17
+async_easy_model-0.3.0.dist-info/RECORD,,

{async_easy_model-0.2.8.dist-info → async_easy_model-0.3.0.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,,