async-easy-model 0.2.9__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.9"
+__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
@@ -107,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}
@@ -437,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)
@@ -769,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():
@@ -1383,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.9.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.9
+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.9.dist-info/RECORD

@@ -1,11 +0,0 @@
-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,,