mysql数据库文档生成

import mysql.connector
from mysql.connector import Error

def generate_db_doc(host, user, password, database):
    try:
        connection = mysql.connector.connect(host=host,
                                             user=user,
                                             password=password,
                                             database=database)
        if connection.is_connected():
            cursor = connection.cursor()
            cursor.execute("SHOW TABLES")
            tables = cursor.fetchall()
            doc = f"Database: {database}\n\n"
            for table in tables:
                table_name = table[0]
                doc += f"Table: {table_name}\n"
                cursor.execute(f"DESCRIBE {table_name}")
                columns = cursor.fetchall()
                for column in columns:
                    doc += f"  {column[0]} - {column[1]} ({column[2]})\n"
                doc += "\n"
            return doc
    except Error as e:
        print(f"Error: {e}")
    finally:
        if connection.is_connected():
            cursor.close()
            connection.close()

# 示例调用
doc = generate_db_doc('localhost', 'root', 'password', 'testdb')
print(doc)