rkutil/export_mysql_to_excel.py

import pymysql
import openpyxl
from openpyxl.styles import Font, PatternFill, Alignment, Border, Side

# ==================== 数据库配置 ====================
DB_CONFIG = {
    "host": "192.168.10.100",
    "port": 3308,
    "user": "root",
    "password": "Aa135790123",
    "database": "cqyt_new_prod",
    "charset": "utf8mb4",
}
OUTPUT_FILE = "export_medical_to_excel.xlsx"

# 需要导出的表名列表（来自 mysql表名.txt）
TARGET_TABLES = [
    "medical_announcement_info",
    "medical_banner",
    "medical_check_package",
    "medical_check_package_item",
    "medical_conclusion",
    "medical_defend_cancer_code",
    "medical_depart_hospital",
    "medical_diseases_dic",
    "medical_diseases_dic_rel",
    "medical_doctor_check_item",
    "medical_doctor_sign",
    "medical_emp_defend_cancer",
    "medical_emp_occup_import",
    "medical_emp_occupational",
    "medical_health_check_statistics",
    "medical_health_check_statistics_info",
    "medical_hospital",
    "medical_hospital_area",
    "medical_hospital_doctor_check",
    "medical_hospital_emp",
    "medical_hospital_except_count",
    "medical_hospital_except_date",
    "medical_hospital_item",
    "medical_hospital_item_rel",
    "medical_hospital_user",
    "medical_hospital_virtual",
    "medical_info",
    "medical_item",
    "medical_item_cost",
    "medical_item_mutex",
    "medical_item_package",
    "medical_item_package_model",
    "medical_item_package_model_append",
    "medical_item_package_model_append_sub",
    "medical_item_package_model_sub",
    "medical_item_package_sub",
    "medical_item_syn",
    "medical_limit_float",
    "medical_limit_model",
    "medical_model_float",
    "medical_order_info",
    "medical_order_queue",
    "medical_package_check",
    "medical_paper_question",
    "medical_paper_question_item",
    "medical_plan",
    "medical_plan_user",
    "medical_precheck_paper",
    "medical_precheck_paper_questions",
    "medical_record_model",
    "medical_record_model_base",
    "medical_record_model_config",
    "medical_serious_result_item",
    "medical_stat_year",
    "medical_uni_item_class",
    "medical_uni_item_class_office",
    "medical_uni_item_class_sicks",
    "medical_uni_item_hospital",
    "medical_uni_item_info",
    "medical_uni_item_risk_rel",
    "medical_uni_item_unusual_rel",
    "medical_user_defend_cancer_result",
    "medical_user_diseases",
    "medical_user_diseases_item",
    "medical_user_exception",
    "medical_user_exception_item",
    "medical_user_form",
    "medical_user_form_sub",
    "medical_user_ill",
    "medical_user_ill_sub",
    "medical_user_image",
    "medical_user_image_item",
    "medical_user_limit",
    "medical_user_paper",
    "medical_user_paper_answer",
    "medical_user_result",
    "medical_user_result_ext",
    "medical_user_result_item",
    "medical_user_result_item_tmp_unmatch",
    "medical_user_sign",
]
# ====================================================


def get_connection():
    """建立 MySQL 数据库连接"""
    return pymysql.connect(
        host=DB_CONFIG["host"],
        port=DB_CONFIG["port"],
        user=DB_CONFIG["user"],
        password=DB_CONFIG["password"],
        database=DB_CONFIG["database"],
        charset=DB_CONFIG["charset"],
    )


def get_table_comment(cursor, table_name):
    """获取表的中文注释"""
    cursor.execute(
        """
        SELECT TABLE_COMMENT
        FROM information_schema.TABLES
        WHERE TABLE_SCHEMA = %s AND TABLE_NAME = %s
        """,
        (DB_CONFIG["database"], table_name)
    )
    row = cursor.fetchone()
    return row[0] if row else ""


def get_columns_with_comments(cursor, table_name):
    """获取表的字段名和中文注释，按字段顺序排列"""
    cursor.execute(
        """
        SELECT COLUMN_NAME, COLUMN_COMMENT
        FROM information_schema.COLUMNS
        WHERE TABLE_SCHEMA = %s AND TABLE_NAME = %s
        ORDER BY ORDINAL_POSITION
        """,
        (DB_CONFIG["database"], table_name)
    )
    return cursor.fetchall()  # [(column_name, comment), ...]


def get_sample_rows(cursor, table_name, limit=3):
    """获取表的示例数据"""
    try:
        cursor.execute(f"SELECT * FROM `{table_name}` LIMIT %s", (limit,))
        return cursor.fetchall()
    except Exception:
        return []


def make_header_style():
    """生成表头样式"""
    fill = PatternFill(fill_type="solid", fgColor="4472C4")
    font = Font(bold=True, color="FFFFFF", size=11)
    alignment = Alignment(horizontal="center", vertical="center", wrap_text=True)
    thin = Side(style="thin", color="FFFFFF")
    border = Border(left=thin, right=thin, top=thin, bottom=thin)
    return fill, font, alignment, border


def make_field_name_style():
    """字段名列样式（浅蓝底）"""
    fill = PatternFill(fill_type="solid", fgColor="DCE6F1")
    font = Font(bold=True, size=10)
    alignment = Alignment(horizontal="left", vertical="center")
    return fill, font, alignment


def make_table_title_style():
    """表名标题行样式（深绿底白字）"""
    fill = PatternFill(fill_type="solid", fgColor="375623")
    font = Font(bold=True, color="FFFFFF", size=12)
    alignment = Alignment(horizontal="left", vertical="center")
    return fill, font, alignment


def write_all_tables_to_sheet(ws, tables_data):
    """将所有表写入同一个 Sheet，每张表前有标题行"""
    header_fill, header_font, header_align, header_border = make_header_style()
    field_fill, field_font, field_align = make_field_name_style()
    title_fill, title_font, title_align = make_table_title_style()

    current_row = 1

    for table_name, table_comment, columns, sample_rows in tables_data:
        # ---------- 第1行：表名 + 表注释（合并A-E列） ----------
        title_text = f"{table_name}  {table_comment or ''}"
        ws.merge_cells(
            start_row=current_row, start_column=1,
            end_row=current_row, end_column=5
        )
        title_cell = ws.cell(row=current_row, column=1, value=title_text)
        title_cell.fill = title_fill
        title_cell.font = title_font
        title_cell.alignment = title_align
        ws.row_dimensions[current_row].height = 22
        current_row += 1

        # ---------- 第2行：列头 ----------
        headers = ["数据名", "中文注释", "示例数据1", "示例数据2", "示例数据3"]
        for col_idx, header in enumerate(headers, start=1):
            cell = ws.cell(row=current_row, column=col_idx, value=header)
            cell.fill = header_fill
            cell.font = header_font
            cell.alignment = header_align
            cell.border = header_border
        ws.row_dimensions[current_row].height = 18
        current_row += 1

        # ---------- 字段数据行 ----------
        for field_index, (col_name, comment) in enumerate(columns):
            cell_a = ws.cell(row=current_row, column=1, value=col_name)
            cell_a.fill = field_fill
            cell_a.font = field_font
            cell_a.alignment = field_align

            ws.cell(row=current_row, column=2, value=comment or "")

            for sample_col, sample_row in enumerate(sample_rows, start=3):
                val = sample_row[field_index] if field_index < len(sample_row) else ""
                ws.cell(row=current_row, column=sample_col,
                        value=str(val) if val is not None else "NULL")
            current_row += 1

        # 表与表之间空一行
        current_row += 1

    # ---------- 统一设置列宽 ----------
    ws.column_dimensions["A"].width = 28
    ws.column_dimensions["B"].width = 32
    ws.column_dimensions["C"].width = 30
    ws.column_dimensions["D"].width = 30
    ws.column_dimensions["E"].width = 30


def main():
    print(f"正在连接 MySQL {DB_CONFIG['host']}:{DB_CONFIG['port']} ...")
    conn = get_connection()
    cursor = conn.cursor()

    print(f"共 {len(TARGET_TABLES)} 张表待处理")

    tables_data = []
    for i, table_name in enumerate(TARGET_TABLES, start=1):
        print(f"[{i}/{len(TARGET_TABLES)}] 处理表: {table_name}")
        table_comment = get_table_comment(cursor, table_name)
        columns = get_columns_with_comments(cursor, table_name)
        if not columns:
            print(f"  ⚠ 字段为空或表不存在: {table_name}")
        else:
            print(f"  字段数: {len(columns)}, 表注释: {table_comment or '（无）'}")
        sample_rows = get_sample_rows(cursor, table_name)
        tables_data.append((table_name, table_comment, columns, sample_rows))

    cursor.close()
    conn.close()

    wb = openpyxl.Workbook()
    ws = wb.active
    ws.title = "表结构数据"

    write_all_tables_to_sheet(ws, tables_data)

    wb.save(OUTPUT_FILE)
    print(f"\n导出完成 -> {OUTPUT_FILE}")


if __name__ == "__main__":
    main()