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

async_easy_model/__init__.py

@@ -6,8 +6,8 @@ from typing import Optional, Any
 from .model import EasyModel, init_db, db_config
 from sqlmodel import Field, Relationship as SQLModelRelationship
 
-__version__ = "0.2.4"
-__all__ = ["EasyModel", "init_db", "db_config", "Field", "Relationship", "Relation", "enable_auto_relationships", "disable_auto_relationships", "process_auto_relationships", "MigrationManager", "check_and_migrate_models"]
+__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
 def Relationship(
@@ -43,4 +43,7 @@ from .relationships import Relation
 from .auto_relationships import enable_auto_relationships, disable_auto_relationships, process_auto_relationships
 
 # Add to __init__.py
-from .migrations import MigrationManager, check_and_migrate_models
+from .migrations import MigrationManager, check_and_migrate_models
+
+# Import the visualization helper
+from .visualization import ModelVisualizer

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

@@ -0,0 +1,669 @@
+"""
+Visualization utilities for EasyModel models.
+
+This module provides tools to visualize EasyModel database schemas,
+including tables, fields, relationships, and automatically generated virtual fields.
+"""
+
+import inspect
+import sys
+from typing import Dict, List, Optional, Type, Set, Any, Union
+from sqlmodel import SQLModel, Field
+from sqlalchemy import inspect as sa_inspect
+
+class ModelVisualizer:
+    """
+    Helper class for visualizing EasyModel database schemas.
+    
+    This class provides methods to generate visual representations of the database schema,
+    including tables, fields, relationships, and automatically generated virtual fields.
+    
+    Attributes:
+        model_registry (Dict[str, Type[SQLModel]]): Dictionary of registered models
+        title (str): Title for the Mermaid diagram
+    """
+    
+    def __init__(self, title: str = "EasyModel Table Schemas"):
+        """
+        Initialize the ModelVisualizer.
+        
+        Args:
+            title: Optional title for the Mermaid diagram
+        """
+        self.model_registry = {}
+        self.title = title
+        self._load_registered_models()
+    
+    def set_title(self, title: str) -> None:
+        """
+        Set or update the title for the Mermaid diagram.
+        
+        Args:
+            title: New title for the Mermaid diagram
+        """
+        self.title = title
+    
+    def _load_registered_models(self):
+        """
+        Load all registered EasyModel models.
+        This should be called after init_db has been executed.
+        """
+        # First try to get models from auto_relationships registry
+        try:
+            from .auto_relationships import _model_registry
+            if _model_registry:
+                self.model_registry = _model_registry.copy()
+                return
+        except (ImportError, AttributeError):
+            pass
+            
+        # Fallback: discover models from loaded modules
+        for module_name, module in sys.modules.items():
+            if hasattr(module, "__dict__"):
+                for cls_name, cls in module.__dict__.items():
+                    if (inspect.isclass(cls) and 
+                        issubclass(cls, SQLModel) and 
+                        cls != SQLModel and 
+                        hasattr(cls, "__tablename__")):
+                        self.model_registry[cls.__name__] = cls
+    
+    def _get_foreign_keys(self, model_class: Type[SQLModel]) -> Dict[str, str]:
+        """
+        Extract foreign key information from a model class.
+        
+        Args:
+            model_class: The SQLModel class to extract foreign keys from.
+            
+        Returns:
+            Dictionary mapping field names to their foreign key references.
+        """
+        foreign_keys = {}
+        
+        try:
+            # Check model annotations and __fields__ for foreign keys
+            if hasattr(model_class, "__annotations__"):
+                for field_name, field_type in model_class.__annotations__.items():
+                    if hasattr(model_class, field_name):
+                        field_value = getattr(model_class, field_name)
+                        if hasattr(field_value, "foreign_key"):
+                            if field_value.foreign_key:
+                                foreign_keys[field_name] = field_value.foreign_key
+            
+            # Check model attributes for Field instances with foreign_key
+            for attr_name in dir(model_class):
+                if attr_name.startswith("__") or attr_name in foreign_keys:
+                    continue
+                try:
+                    attr_value = getattr(model_class, attr_name)
+                    if hasattr(attr_value, "foreign_key"):
+                        if attr_value.foreign_key:
+                            foreign_keys[attr_name] = attr_value.foreign_key
+                except (AttributeError, TypeError):
+                    pass
+            
+            # Try to infer foreign keys from field names ending with _id
+            if hasattr(model_class, "__fields__"):
+                for field_name in model_class.__fields__:
+                    if field_name.endswith("_id") and field_name not in foreign_keys:
+                        related_name = field_name[:-3]  # Remove _id suffix
+                        # Check if there's a model with this name
+                        for model_name in self.model_registry:
+                            if model_name.lower() == related_name.lower():
+                                foreign_keys[field_name] = f"{model_name.lower()}.id"
+                                break
+        except Exception as e:
+            # Log but don't re-raise to ensure visualization continues
+            print(f"Warning: Error getting foreign keys for {model_class.__name__}: {str(e)}")
+        
+        return foreign_keys
+    
+    def _get_virtual_relationship_fields(self, model_class: Type[SQLModel]) -> Dict[str, Dict[str, Any]]:
+        """
+        Get virtual relationship fields that are automatically generated.
+        
+        Args:
+            model_class: The SQLModel class to extract virtual relationship fields from.
+            
+        Returns:
+            Dictionary of virtual relationship fields info.
+        """
+        virtual_fields = {}
+        table_name = getattr(model_class, "__tablename__", model_class.__name__.lower())
+        
+        try:
+            # Check for junction tables (tables with two foreign keys that form many-to-many)
+            foreign_keys = self._get_foreign_keys(model_class)
+            
+            # Track models that have relations via this model's foreign keys
+            related_models = set()
+            
+            # Determine if this is a junction table (has two foreign keys to different tables)
+            is_junction_table = False
+            if len(foreign_keys) >= 2:
+                referenced_tables = set()
+                for fk_target in foreign_keys.values():
+                    if "." in fk_target:
+                        referenced_tables.add(fk_target.split(".")[0])
+                
+                is_junction_table = len(referenced_tables) >= 2
+            
+            # For each foreign key, determine the virtual relationship field
+            for field_name, fk_target in foreign_keys.items():
+                if "." not in fk_target:
+                    continue
+                    
+                target_table = fk_target.split(".")[0]
+                
+                # Find the related model class
+                target_model = None
+                for name, cls in self.model_registry.items():
+                    cls_table_name = getattr(cls, "__tablename__", name.lower())
+                    if cls_table_name.lower() == target_table.lower():
+                        target_model = name
+                        related_models.add(name)
+                        break
+                
+                if not target_model:
+                    continue
+                
+                # The relationship field is typically named without the _id suffix
+                rel_name = field_name[:-3] if field_name.endswith("_id") else field_name
+                
+                # Skip if this looks like a duplicate relationship
+                if rel_name in virtual_fields:
+                    continue
+                
+                # Add the virtual relationship field (singular form - one instance)
+                virtual_fields[rel_name] = {
+                    "name": rel_name,
+                    "type": target_model,
+                    "is_list": False,  # One-to-one or many-to-one
+                    "related_model": target_model,
+                    "is_virtual": True,
+                    "is_required": False  # Virtual fields are usually optional
+                }
+            
+            # For junction tables, create the many-to-many virtual fields
+            if is_junction_table and len(related_models) >= 2:
+                # Don't add virtual fields to the junction table itself
+                return virtual_fields
+            
+            # For each model, check if it's referenced in a junction table to create many-to-many virtuals
+            for junction_name, junction_class in self.model_registry.items():
+                junction_table_name = getattr(junction_class, "__tablename__", junction_name.lower())
+                
+                # Skip if this is not a potential junction table
+                if junction_class == model_class:  # Skip self
+                    continue
+                
+                junction_fks = self._get_foreign_keys(junction_class)
+                if len(junction_fks) < 2:  # Not a potential junction
+                    continue
+                
+                # Check if this model is referenced by the potential junction table
+                this_model_referenced = False
+                other_referenced_models = set()
+                
+                for fk_target in junction_fks.values():
+                    if "." not in fk_target:
+                        continue
+                        
+                    target_table = fk_target.split(".")[0]
+                    
+                    if target_table == table_name:
+                        this_model_referenced = True
+                    else:
+                        # Find the model class for this target
+                        for other_name, other_cls in self.model_registry.items():
+                            other_table = getattr(other_cls, "__tablename__", other_name.lower())
+                            if other_table == target_table:
+                                other_referenced_models.add(other_name)
+                                break
+                
+                # If this is a junction table connecting this model to others
+                if this_model_referenced and other_referenced_models:
+                    for other_model in other_referenced_models:
+                        # Create a plural form of the model name for the relationship
+                        # This is a very simple pluralization, might need to improve
+                        other_model_lower = other_model.lower()
+                        plural_name = f"{other_model_lower}s"
+                        
+                        # Add the many-to-many virtual field (plural form)
+                        virtual_fields[plural_name] = {
+                            "name": plural_name,
+                            "type": f"List[{other_model}]",
+                            "is_list": True,  # Many-to-many
+                            "related_model": other_model,
+                            "is_virtual": True,
+                            "is_required": False
+                        }
+            
+        except Exception as e:
+            # Log but don't re-raise to ensure visualization continues
+            print(f"Warning: Error getting virtual fields for {model_class.__name__}: {str(e)}")
+        
+        return virtual_fields
+    
+    def _get_field_information(self, model_class: Type[SQLModel]) -> Dict[str, Dict[str, Any]]:
+        """
+        Extract field information from a model class.
+        
+        Args:
+            model_class: The SQLModel class to extract field information from.
+            
+        Returns:
+            Dictionary of field information with field names as keys.
+        """
+        fields = {}
+        
+        try:
+            # Make sure the 'id' field is always included
+            fields["id"] = {
+                "name": "id",
+                "type": "int",
+                "is_primary": True,
+                "is_foreign_key": False,
+                "foreign_key_reference": None,
+                "is_virtual": False,
+                "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():
+                    # Skip private fields
+                    if field_name.startswith("_"):
+                        continue
+                    
+                    # Get field type as string
+                    type_str = getattr(field_type, "__name__", str(field_type))
+                    
+                    # Check if field is optional
+                    is_optional = False
+                    if hasattr(field_type, "__origin__"):
+                        if field_type.__origin__ is Union:
+                            args = getattr(field_type, "__args__", [])
+                            if type(None) in args:
+                                is_optional = True
+                                # Simplify type for optional fields (remove Union and NoneType)
+                                other_types = [arg for arg in args if arg is not type(None)]
+                                if len(other_types) == 1:
+                                    # If there's only one other type (common case like Optional[str])
+                                    type_str = getattr(other_types[0], "__name__", str(other_types[0]))
+                                else:
+                                    # For more complex unions, just indicate it's a Union
+                                    type_str = "Union"
+                    
+                    # 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,
+                        "type": type_str,
+                        "is_primary": is_primary,
+                        "is_foreign_key": False,
+                        "foreign_key_reference": None,
+                        "is_virtual": False,
+                        "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
+            foreign_keys = self._get_foreign_keys(model_class)
+            for field_name, fk_target in foreign_keys.items():
+                if field_name in fields:
+                    fields[field_name]["is_foreign_key"] = True
+                    fields[field_name]["foreign_key_reference"] = fk_target
+                else:
+                    # Foreign key field wasn't in annotations, add it
+                    fields[field_name] = {
+                        "name": field_name,
+                        "type": "int",  # Most foreign keys are integers
+                        "is_primary": False,
+                        "is_foreign_key": True,
+                        "foreign_key_reference": fk_target,
+                        "is_virtual": False,
+                        "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
+            virtual_fields = self._get_virtual_relationship_fields(model_class)
+            for field_name, field_info in virtual_fields.items():
+                fields[field_name] = field_info
+        
+        except Exception as e:
+            # Log but don't re-raise to ensure visualization continue
+            print(f"Warning: Error processing fields for {model_class.__name__}: {str(e)}")
+        
+        return fields
+    
+    def _generate_mermaid_content(self) -> str:
+        """
+        Generate the raw Mermaid ER diagram content without markdown code fences.
+        This is used internally by both mermaid() and mermaid_link() methods.
+        
+        Returns:
+            String containing raw Mermaid ER diagram markup without markdown fences.
+        """
+        if not self.model_registry:
+            return f"---\ntitle: {self.title}\nconfig:\n    layout: elk\n---\nerDiagram\n    %% No models found. Run init_db first."
+        
+        # Start with the title section
+        lines = [
+            "---",
+            f"title: {self.title}",
+            "config:",
+            "    layout: elk",
+            "---",
+            "erDiagram"
+        ]
+        
+        # Keep track of rendered relationships to avoid duplicates
+        rendered_relationships = set()
+        processed_models = set()
+        
+        # Try to process all models, but continue even if some fail
+        for model_name, model_class in self.model_registry.items():
+            try:
+                table_name = getattr(model_class, "__tablename__", model_name.lower())
+                processed_models.add(model_name)
+                
+                # Add entity definition
+                lines.append(f"    {table_name} {{")
+                
+                # Get fields for this model
+                fields = self._get_field_information(model_class)
+                
+                # 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"]))
+                    
+                    # If it's a relationship, use the proper model name
+                    if field_info.get("is_virtual", False) and field_info.get("related_model"):
+                        related_model = field_info["related_model"]
+                        if field_info.get("is_list", False):
+                            # For list fields like 'tags', use the proper casing for model name
+                            field_type = f"{related_model}[]"
+                        else:
+                            # For single object reference, use the model name directly
+                            field_type = related_model
+                    
+                    # Format attributes with proper Mermaid syntax
+                    attrs_str = self._format_field_attributes(field_info)
+                    
+                    # 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("    }")
+            
+            except Exception as e:
+                lines.append(f"    %% Error defining {model_name}: {str(e)}")
+        
+        # Add relationships between models
+        for model_name, model_class in self.model_registry.items():
+            try:
+                if model_name not in processed_models:
+                    continue
+                    
+                table_name = getattr(model_class, "__tablename__", model_name.lower())
+                
+                # Get fields for this model
+                fields = self._get_field_information(model_class)
+                
+                # Add relationships based on foreign keys
+                for field_name, field_info in fields.items():
+                    if field_info.get("is_foreign_key", False) and field_info.get("foreign_key_reference"):
+                        # Parse the foreign key reference to get the target table
+                        fk_ref = field_info["foreign_key_reference"]
+                        target_table = fk_ref.split(".")[0] if "." in fk_ref else fk_ref
+                        
+                        # Create relationship ID to avoid duplicates
+                        rel_id = f"{table_name}_{target_table}_{field_name}"
+                        if rel_id in rendered_relationships:
+                            continue
+                        
+                        # Add the relationship
+                        lines.append(f"    {table_name} ||--o{{ {target_table} : \"{field_name}\"")
+                        rendered_relationships.add(rel_id)
+                
+                # Add many-to-many relationships
+                # Check if this model might be a junction table
+                if len(fields) >= 3:  # id + at least 2 foreign keys
+                    foreign_key_fields = [f for f in fields.values() if f.get("is_foreign_key", False)]
+                    
+                    if len(foreign_key_fields) >= 2:
+                        # This might be a junction table, try to render special M:N relationship
+                        for i, fk1 in enumerate(foreign_key_fields):
+                            for fk2 in foreign_key_fields[i+1:]:
+                                # Skip if either field doesn't have a foreign key reference
+                                if not fk1.get("foreign_key_reference") or not fk2.get("foreign_key_reference"):
+                                    continue
+                                    
+                                # Get the target tables
+                                target1 = fk1["foreign_key_reference"].split(".")[0]
+                                target2 = fk2["foreign_key_reference"].split(".")[0]
+                                
+                                # Skip self-references or duplicates
+                                if target1 == target2:
+                                    continue
+                                    
+                                # Create relationship IDs
+                                rel_id1 = f"{target1}_{target2}_m2m"
+                                rel_id2 = f"{target2}_{target1}_m2m"
+                                
+                                if rel_id1 in rendered_relationships or rel_id2 in rendered_relationships:
+                                    continue
+                                    
+                                # Add the many-to-many relationship directly between the end entities
+                                lines.append(f"    {target1} }}o--o{{ {target2} : \"many-to-many\"")
+                                rendered_relationships.add(rel_id1)
+            
+            except Exception as e:
+                lines.append(f"    %% Error processing relationships for {model_name}: {str(e)}")
+        
+        return "\n".join(lines)
+    
+    def _simplify_type_for_mermaid(self, type_str: str) -> str:
+        """
+        Simplify a Python type string for Mermaid ER diagram display.
+        
+        Args:
+            type_str: Python type string to simplify
+            
+        Returns:
+            A simplified type string suitable for Mermaid diagrams
+        """
+        # Remove common Python type prefixes
+        simplified = type_str.replace("typing.", "").replace("__main__.", "")
+        
+        # Handle common container types
+        if simplified.startswith("List["):
+            inner_type = simplified[5:-1]  # Extract the type inside List[]
+            # Preserve proper casing for model names
+            return f"{inner_type}[]"
+        
+        # Handle other complex types that might confuse Mermaid
+        if "[" in simplified or "Union" in simplified or "Optional" in simplified:
+            # For complex types, just return a more generic type name
+            if "str" in simplified:
+                return "string"
+            elif "int" in simplified or "float" in simplified:
+                return "number"
+            elif "bool" in simplified:
+                return "boolean"
+            elif "dict" in simplified or "Dict" in simplified:
+                return "object"
+            else:
+                return "any"
+        
+        # Simple type mapping to more Mermaid-friendly types
+        type_map = {
+            "str": "string",
+            "int": "number",
+            "float": "number",
+            "bool": "boolean",
+            "dict": "object",
+            "Dict": "object",
+            "datetime": "datetime",
+            "date": "date",
+            "time": "time",
+            "bytes": "binary",
+            "bytearray": "binary"
+        }
+        
+        return type_map.get(simplified, simplified)
+    
+    def _get_model_name_for_table(self, table_name: str) -> str:
+        """
+        Get the proper cased model name for a table name.
+        
+        Args:
+            table_name: The table name to find the model name for
+            
+        Returns:
+            The properly cased model name
+        """
+        for model_name, model_class in self.model_registry.items():
+            if getattr(model_class, "__tablename__", model_name.lower()) == table_name:
+                return model_name
+        return table_name  # Fallback to table name if no model found
+    
+    def _format_field_attributes(self, field_info: Dict[str, Any]) -> str:
+        """
+        Format field attributes according to Mermaid syntax.
+        Only the last attribute should have double quotes, and attributes are separated by spaces.
+        
+        Args:
+            field_info: Dictionary with field information
+            
+        Returns:
+            String with properly formatted attributes for the Mermaid diagram
+        """
+        attrs = []
+        
+        # Order is important: PK, FK, and then other attributes
+        if field_info.get("is_primary", False):
+            attrs.append("PK")
+        if field_info.get("is_foreign_key", False):
+            attrs.append("FK")
+        
+        # The comment attribute (should be last and in quotes)
+        comment = None
+        if field_info.get("is_virtual", False):
+            comment = "virtual"
+        elif field_info.get("is_required", False):
+            comment = "required"
+            
+        # Format the attribute string
+        if not attrs and not comment:
+            return ""
+        
+        result = ""
+        if attrs:
+            result = " " + " ".join(attrs)
+            
+        if comment:
+            result += f' "{comment}"'
+            
+        return result
+    
+    def mermaid(self) -> str:
+        """
+        Generate a Mermaid ER diagram for all registered models.
+        
+        Returns:
+            String containing Mermaid ER diagram markup in markdown format.
+        """
+        # Get the raw Mermaid content
+        content = self._generate_mermaid_content()
+        
+        # Wrap in markdown code fences
+        return f"```mermaid\n{content}\n```"
+    
+    def mermaid_link(self) -> str:
+        """
+        Generate a Mermaid Live Editor link for the ER diagram.
+        
+        Returns:
+            URL string that opens the diagram in Mermaid Live Editor.
+        """
+        # Code to handle mermaid.live base64 links
+        import base64, json, zlib
+        
+        def js_btoa(data):
+            return base64.b64encode(data)
+
+        def pako_deflate(data):
+            compress = zlib.compressobj(9, zlib.DEFLATED, 15, 8, zlib.Z_DEFAULT_STRATEGY)
+            compressed_data = compress.compress(data)
+            compressed_data += compress.flush()
+            return compressed_data
+
+        def gen_pako_link(graph_markdown: str):
+            jGraph = {"code": graph_markdown, "mermaid": {"theme": "default"}}
+            byte_str = json.dumps(jGraph).encode('utf-8')
+            deflated = pako_deflate(byte_str)
+            d_encode = js_btoa(deflated)
+            link = 'https://mermaid.live/edit#pako:' + d_encode.decode('ascii')
+            return link
+
+        # Get the raw Mermaid content directly (without markdown fences)
+        mermaid_content = self._generate_mermaid_content()
+        
+        # Generate the link
+        return gen_pako_link(mermaid_content)
+    
+    # Alias for backward compatibility
+    generate_mermaid_er_diagram = mermaid

{async_easy_model-0.2.4.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.4
+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
@@ -59,6 +60,7 @@ A simplified SQLModel-based ORM for async database operations in Python. async-e
 - 📝 Type hints for better IDE support
 - 🕒 Automatic `id`, `created_at` and `updated_at` fields provided by default
 - 🔄 Automatic schema migrations for evolving database models
+- 📊 Visualization of database schema using Mermaid ER diagrams
 
 ## Installation
 
@@ -250,6 +252,89 @@ deleted_count = await User.delete({"is_active": False})
 await Post.delete({"user": {"username": "john_doe"}, "is_published": False})
 ```
 
+## Database Schema Visualization
+
+The package includes a `ModelVisualizer` class that makes it easy to generate Entity-Relationship (ER) diagrams for your database models using Mermaid syntax.
+
+```python
+from async_easy_model import EasyModel, init_db, db_config, ModelVisualizer
+
+# Initialize your models and database
+await init_db()
+
+# Create a visualizer
+visualizer = ModelVisualizer()
+
+# Generate a Mermaid ER diagram
+er_diagram = visualizer.mermaid()
+print(er_diagram)
+
+# Generate a shareable link to view the diagram online
+er_link = visualizer.mermaid_link()
+print(er_link)
+
+# Customize the diagram title
+visualizer.set_title("My Project Database Schema")
+custom_diagram = visualizer.mermaid()
+```
+
+### Example Mermaid ER Diagram Output
+
+```mermaid
+---
+title: EasyModel Table Schemas
+config:
+    layout: elk
+---
+erDiagram
+    author {
+        number id PK
+        string name "required"
+        string email
+    }
+    book {
+        number id PK
+        string title "required"
+        number author_id FK
+        string isbn
+        number published_year
+        author author "virtual"
+        tag[] tags "virtual"
+    }
+    tag {
+        number id PK
+        string name "required"
+        book[] books "virtual"
+    }
+    book_tag {
+        number id PK
+        number book_id FK "required"
+        number tag_id FK "required"
+        book book "virtual"
+        tag tag "virtual"
+    }
+    review {
+        number id PK
+        number book_id FK "required"
+        number rating "required"
+        string comment
+        string reviewer_name "required"
+        book book "virtual"
+    }
+    book ||--o{ author : "author_id"
+    book_tag ||--o{ book : "book_id"
+    book_tag ||--o{ tag : "tag_id"
+    book }o--o{ tag : "many-to-many"
+    review ||--o{ book : "book_id"
+```
+
+The diagram automatically:
+- Shows all tables with their fields and data types
+- Identifies primary keys (PK) and foreign keys (FK)
+- Shows required fields and virtual relationships
+- Visualizes relationships between tables with proper cardinality
+- Properly handles many-to-many relationships
+
 ## Convenient Query Methods
 
 async-easy-model provides simplified methods for common query patterns:

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.4.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.4.dist-info/RECORD

@@ -1,10 +0,0 @@
-async_easy_model/__init__.py,sha256=qtmIxVNPohdcmh0KmQT1TArYRQRtA9aVj8RAa5qrApc,1642
-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-0.2.4.dist-info/LICENSE,sha256=uwDkl6oHbRltW7vYKNc4doJyhtwhyrSNFFlPpKATwLE,1072
-async_easy_model-0.2.4.dist-info/METADATA,sha256=kV2oSzYEJ2ac1ScXrWmTjr4EJtO8-e-LQawsBSykr5k,10720
-async_easy_model-0.2.4.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-async_easy_model-0.2.4.dist-info/top_level.txt,sha256=e5_47sGmJnyxz2msfwU6C316EqmrSd9RGIYwZyWx68E,17
-async_easy_model-0.2.4.dist-info/RECORD,,