### 使用Python将SQL建表语句转换为ER图的工具或库 在处理SQL建表语句并将其转换为ER图时，可以使用一些专门设计的工具和库来完成这一任务。以下是几个常用的工具或库，它们能够帮助实现从SQL到ER图的转换： #### 1. **SQLAlchemy + ERAlchemy** SQLAlchemy 是一个功能强大的ORM（对象关系映射）工具，而 ERAlchemy 是基于 SQLAlchemy 的 ER 图生成工具。通过解析 SQL 建表语句并将其导入到 SQLAlchemy 模型中，ERAlchemy 可以生成可视化的 ER 图。 - **步骤说明**： - 使用 SQLAlchemy 定义数据库模型[^2]。 - 利用 ERAlchemy 将 SQLAlchemy 模型转换为 ER 图。 ```python # 安装依赖 pip install sqlalchemy eralchemy # 示例代码：定义模型 from sqlalchemy import Column, Integer, String, ForeignKey from sqlalchemy.ext.declarative import declarative_base Base = declarative_base() class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String) class Address(Base): __tablename__ = 'addresses' id = Column(Integer, primary_key=True) user_id = Column(Integer, ForeignKey('users.id')) email = Column(String) # 生成 ER 图 !eralchemy -i models.py -o er_diagram.png ``` #### 2. **dbdiagram.io** dbdiagram.io 是一个在线工具，支持直接输入 SQL 建表语句并生成 ER 图。它还提供了一个 Python SDK，允许开发者通过脚本调用其 API。 - **优点**： - 支持多种数据库方言（如 MySQL、PostgreSQL 等）。 - 自动生成美观的 ER 图，并支持导出为 PNG 或 SVG 格式。 - **示例代码**： ```python import dbdiagram # 输入 SQL 建表语句 sql = """ CREATE TABLE users ( id INT PRIMARY KEY, name VARCHAR(50) ); CREATE TABLE addresses ( id INT PRIMARY KEY, user_id INT, email VARCHAR(50), FOREIGN KEY (user_id) REFERENCES users(id) ); """ # 调用 dbdiagram.io 的 API response = dbdiagram.generate(sql, output="er_diagram.png") ``` #### 3. **PyMySQL + Graphviz** 如果需要完全自定义解决方案，可以结合 PyMySQL 解析 SQL 建表语句，并使用 Graphviz 生成 ER 图。这种方法灵活性较高，但开发成本也较大。 - **流程描述**： - 使用 PyMySQL 连接到数据库并提取 DDL 语句[^1]。 - 解析 DDL 语句，提取表名、字段、主键、外键等信息。 - 使用 Graphviz 绘制节点和边，生成 ER 图。 - **示例代码**： ```python import pymysql from graphviz import Digraph # 连接数据库 def connect_db(host, user, password, database): conn = pymysql.connect( host=host, user=user, password=password, database=database ) return conn # 提取建表语句 conn = connect_db("localhost", "root", "password", "news_system") cursor = conn.cursor() cursor.execute("SHOW CREATE TABLE users;") create_table_sql = cursor.fetchone()[1] # 使用 Graphviz 绘制 ER 图 dot = Digraph(comment='ER Diagram') dot.node('users', 'users') dot.node('addresses', 'addresses') dot.edge('users', 'addresses', label='1:N') dot.render('er_diagram.gv', view=True) ``` #### 4. **SchemaCrawler** SchemaCrawler 是一个命令行工具，可以生成数据库的 ER 图。它支持多种输出格式，包括图形化 ER 图和文本报告。 - **安装与使用**： ```bash # 下载 SchemaCrawler wget https://www.schemacrawler.com/download.html # 生成 ER 图 java -jar schemacrawler.jar --server=mysql --user=root --password=password --info-level=standard --output-format=png --output-file=er_diagram.png ``` --- ###