sqlalchemy-schemadisplay

import os from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.orm import declarative_base from sqlalchemy_schemadisplay import create_schema_graph # 1. Define your SQLAlchemy models Base = declarative_base() class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String(50), nullable=False) email = Column(String(100), unique=True) class Product(Base): __tablename__ = 'products' id = Column(Integer, primary_key=True) name = Column(String(50), nullable=False) price = Column(Integer) # 2. Create the graph and save it to a file try: # Using Base.metadata directly graph = create_schema_graph( metadata=Base.metadata, show_datatypes=True, show_labels=True, rankdir='LR', # Left-to-right layout orientation='portrait' ) output_file = 'schema.png' graph.write_png(output_file) print(f"Schema diagram saved to {output_file}") except Exception as e: print(f"An error occurred: {e}") print("Make sure Graphviz is installed on your system PATH and 'graphviz' Python package is installed.") # Optional: Clean up (if you created a temporary database for reflection) # os.remove('test.db') if os.path.exists('test.db') else None