async-easy-model 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

async_easy_model/__init__.py

@@ -6,7 +6,7 @@ from typing import Optional, Any
 from .model import EasyModel, init_db, db_config
 from sqlmodel import Field, Relationship as SQLModelRelationship
 
-__version__ = "0.2.5"
+__version__ = "0.2.6"
 __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

@@ -1398,7 +1398,7 @@ async def init_db(migrate: bool = True, model_classes: List[Type[SQLModel]] = No
     if model_classes is None:
         model_classes = []
         # Get all model classes by inspecting the modules
-        for module_name, module in sys.modules.items():
+        for module_name, module in list(sys.modules.items()):
             if hasattr(module, "__dict__"):
                 for cls_name, cls in module.__dict__.items():
                     if isinstance(cls, type) and issubclass(cls, SQLModel) and cls != SQLModel and cls != EasyModel:

async_easy_model/visualization.py

@@ -268,6 +268,38 @@ class ModelVisualizer:
                 "is_required": False  # Changed: primary keys are not considered "required" for the diagram
             }
             
+            # For EasyModel tables, make sure created_at and updated_at are included
+            # These fields are automatically added by EasyModel but may not be in the model definitions
+            is_easy_model = False
+            # Check class hierarchy to determine if this is an EasyModel
+            for base in model_class.__mro__:
+                if base.__name__ == "EasyModel":
+                    is_easy_model = True
+                    break
+            
+            if is_easy_model:
+                # Add created_at timestamp field
+                fields["created_at"] = {
+                    "name": "created_at",
+                    "type": "datetime",
+                    "is_primary": False,
+                    "is_foreign_key": False,
+                    "foreign_key_reference": None,
+                    "is_virtual": False,
+                    "is_required": False  # Not required for user input as handled automatically
+                }
+                
+                # Add updated_at timestamp field
+                fields["updated_at"] = {
+                    "name": "updated_at",
+                    "type": "datetime",
+                    "is_primary": False,
+                    "is_foreign_key": False,
+                    "foreign_key_reference": None,
+                    "is_virtual": False,
+                    "is_required": False  # Not required for user input as handled automatically
+                }
+            
             # Get standard database fields
             if hasattr(model_class, "__annotations__"):
                 for field_name, field_type in model_class.__annotations__.items():
@@ -297,6 +329,9 @@ class ModelVisualizer:
                     # Check if it's a primary key
                     is_primary = field_name == "id"
                     
+                    # Check if it's an auto-generated timestamp field
+                    is_auto_timestamp = field_name in ["created_at", "updated_at"]
+                    
                     # Store field information
                     fields[field_name] = {
                         "name": field_name,
@@ -305,7 +340,7 @@ class ModelVisualizer:
                         "is_foreign_key": False,
                         "foreign_key_reference": None,
                         "is_virtual": False,
-                        "is_required": not is_optional and not is_primary  # Primary keys are not "required" for the diagram
+                        "is_required": not is_optional and not is_primary and not is_auto_timestamp  # Don't mark primary keys and auto timestamps as "required"
                     }
             
             # Get foreign key information and update fields
@@ -323,7 +358,7 @@ class ModelVisualizer:
                         "is_foreign_key": True,
                         "foreign_key_reference": fk_target,
                         "is_virtual": False,
-                        "is_required": field_name != "id"  # Only mark as required if not the id field
+                        "is_required": field_name != "id" and field_name not in ["created_at", "updated_at"]  # Don't mark auto timestamps as required
                     }
             
             # Get virtual relationship fields
@@ -374,7 +409,14 @@ class ModelVisualizer:
                 # Get fields for this model
                 fields = self._get_field_information(model_class)
                 
-                # Add fields
+                # Separate timestamp fields to place them at the bottom
+                timestamp_fields = {}
+                if "created_at" in fields:
+                    timestamp_fields["created_at"] = fields.pop("created_at")
+                if "updated_at" in fields:
+                    timestamp_fields["updated_at"] = fields.pop("updated_at")
+                
+                # Add regular fields first
                 for field_name, field_info in fields.items():
                     # Format type
                     field_type = self._simplify_type_for_mermaid(str(field_info["type"]))
@@ -395,6 +437,14 @@ class ModelVisualizer:
                     # Add field
                     lines.append(f"        {field_type} {field_name}{attrs_str}")
                 
+                # Add timestamp fields at the bottom
+                for field_name in ["created_at", "updated_at"]:
+                    if field_name in timestamp_fields:
+                        field_info = timestamp_fields[field_name]
+                        field_type = self._simplify_type_for_mermaid(str(field_info["type"]))
+                        attrs_str = self._format_field_attributes(field_info)
+                        lines.append(f"        {field_type} {field_name}{attrs_str}")
+                
                 # Close entity definition
                 lines.append("    }")
             
@@ -506,7 +556,7 @@ class ModelVisualizer:
             "bool": "boolean",
             "dict": "object",
             "Dict": "object",
-            "datetime": "date",
+            "datetime": "datetime",
             "date": "date",
             "time": "time",
             "bytes": "binary",

{async_easy_model-0.2.5.dist-info → async_easy_model-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: async-easy-model
-Version: 0.2.5
+Version: 0.2.6
 Summary: A simplified SQLModel-based ORM for async database operations
 Home-page: https://github.com/puntorigen/easy-model
 Author: Pablo Schaffner
@@ -32,6 +32,7 @@ Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: keywords
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary

async_easy_model-0.2.6.dist-info/RECORD

@@ -0,0 +1,11 @@
+async_easy_model/__init__.py,sha256=5OzRFnNcCv5QAkDM7Gkt9jW4K1BIxXoYy_KbCZXZWnw,1739
+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.6.dist-info/licenses/LICENSE,sha256=uwDkl6oHbRltW7vYKNc4doJyhtwhyrSNFFlPpKATwLE,1072
+async_easy_model-0.2.6.dist-info/METADATA,sha256=JRhh3mi73bMeLWAA83G3vkQl9O0avDkmasjxVVO8nFw,12888
+async_easy_model-0.2.6.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+async_easy_model-0.2.6.dist-info/top_level.txt,sha256=e5_47sGmJnyxz2msfwU6C316EqmrSd9RGIYwZyWx68E,17
+async_easy_model-0.2.6.dist-info/RECORD,,

{async_easy_model-0.2.5.dist-info → async_easy_model-0.2.6.dist-info}/WHEEL

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

async_easy_model-0.2.5.dist-info/RECORD

@@ -1,11 +0,0 @@
-async_easy_model/__init__.py,sha256=cpv45_jab9WbuemZehABoXL64VZ19R-L4-UOAbj8sls,1739
-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=Vq6NTUThuEKy_CVLAb6Dy5pcDjbBr49qgamTQj5GtZ0,63284
-async_easy_model/relationships.py,sha256=vR5BsJpGaDcecCcNlg9-ouZfxFXFQv5kOyiXhKp_T7A,3286
-async_easy_model/visualization.py,sha256=VdW89D103sF7nnB1BG3EnGcID1HF-1TVr_U0iaOuFjA,27355
-async_easy_model-0.2.5.dist-info/LICENSE,sha256=uwDkl6oHbRltW7vYKNc4doJyhtwhyrSNFFlPpKATwLE,1072
-async_easy_model-0.2.5.dist-info/METADATA,sha256=G6Ol1O_uihhCchSJnf0dCHEP0G3NEl4Nd_pBJwOoKiA,12866
-async_easy_model-0.2.5.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-async_easy_model-0.2.5.dist-info/top_level.txt,sha256=e5_47sGmJnyxz2msfwU6C316EqmrSd9RGIYwZyWx68E,17
-async_easy_model-0.2.5.dist-info/RECORD,,