mysql 数据表文档生成

import mysql.connector

def generate_doc(db_config):
    conn = mysql.connector.connect(**db_config)
    cursor = conn.cursor(dictionary=True)

    cursor.execute("SHOW TABLES")
    tables = cursor.fetchall()

    doc = "# MySQL 数据库表文档\n\n"

    for table in tables:
        table_name = table['Tables_in_' + db_config['database']]
        doc += f"## {table_name}\n\n"
        cursor.execute(f"SHOW CREATE TABLE {table_name}")
        create_table_sql = cursor.fetchone()['Create Table']
        doc += f"{create_table_sql}\n\n"

        cursor.execute(f"DESC {table_name}")
        columns = cursor.fetchall()
        doc += "| 字段名 | 类型 | 描述 |\n| --- | --- | --- |\n"
        for column in columns:
            doc += f"| {column['Field']} | {column['Type']} | {column.get('Comment', '')} |\n"
        doc += "\n"

    cursor.close()
    conn.close()

    with open("mysql_doc.md", "w") as f:
        f.write(doc)

# 数据库配置
db_config = {
    'host': 'localhost',
    'user': 'root',
    'password': 'your_password',
    'database': 'your_database'
}

generate_doc(db_config)