mysql生成doc

import mysql.connector
from mysql.connector import Error

def generate_doc(host, database, user, password):
    try:
        connection = mysql.connector.connect(host=host,
                                             database=database,
                                             user=user,
                                             password=password)

        if connection.is_connected():
            cursor = connection.cursor()
            cursor.execute("SHOW TABLES")
            tables = cursor.fetchall()

            doc = "# MySQL Database Documentation\n\n"

            for table in tables:
                table_name = table[0]
                doc += f"## Table: {table_name}\n\n"
                cursor.execute(f"DESCRIBE {table_name}")
                columns = cursor.fetchall()
                doc += "| Field | Type | Null | Key | Default | Extra |\n"
                doc += "| --- | --- | --- | --- | --- | --- |\n"
                for column in columns:
                    doc += f"| {column[0]} | {column[1]} | {column[2]} | {column[3]} | {column[4]} | {column[5]} |\n"
                doc += "\n"

            with open("database_doc.md", "w") as file:
                file.write(doc)

            print("Documentation generated successfully.")

    except Error as e:
        print(f"Error: {e}")

    finally:
        if connection.is_connected():
            cursor.close()
            connection.close()

# Example usage
generate_doc("localhost", "mydatabase", "root", "password")