使用 datamodel-code-generator 从 MySQL 生成 Python 模型

使用 datamodel-code-generator 从 MySQL 生成 Python 模型

简介

datamodel-code-generator 是一个强大的工具，可以从多种数据源（包括 MySQL）自动生成 Python 数据模型。本文将详细介绍如何使用它从 MySQL 数据库生成 Pydantic 模型。

安装必要组件

首先需要安装相关的 Python 包：

# 安装主要工具
pip install datamodel-code-generator

# 安装 MySQL 支持
pip install sqlalchemy
pip install pymysql

使用方法

1. 从数据库直接生成

最基本的用法是直接从数据库连接生成：

datamodel-codegen \
    --db-url="mysql+pymysql://username:password@localhost:3306/database_name" \
    --output models.py

2. 从 SQL 文件生成

如果您有数据库的 SQL 文件，也可以直接从 SQL 文件生成：

datamodel-codegen \
    --input your_schema.sql \
    --input-file-type sql \
    --output models.py

常用参数说明

基本参数

• --output: 输出文件路径
• --input: 输入文件（使用 SQL 文件时）
• --input-file-type: 输入文件类型
• --db-url: 数据库连接字符串

高级参数

• --target-python-version: 指定目标 Python 版本

  datamodel-codegen --db-url="..."