#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 从 mom系统数据库设计.xlsx 读取表与字段定义，更新 mom系统数据库设计-数据模型.md。 保留 md 的 1-3 节（概述、模块清单、公共字段）和 14 节及以后（枚举、ER、附录等）， 用 Excel 内容更新 2.2 表清单总览 和 4-13 节各模块表结构。 """ import re from pathlib import Path try: import openpyxl except ImportError: raise SystemExit("请安装 openpyxl: pip install openpyxl") # 路径 BASE = Path(__file__).resolve().parent XLSX_PATH = BASE / "mom系统数据库设计.xlsx" MD_PATH = BASE / "mom系统数据库设计-数据模型.md" # Excel 表清单中的「模块」-> 文档中的 (节号, 模块标题, 表前缀) # 节号与原文一致: 4基础数据 5生产 6质量 7设备 8仓库 9排班 10IOT 11工具 12系统 13打印 MODULE_DOC = { "主数据管理": (4, "基础数据模块", "md_*"), "生产管理": (5, "生产管理模块", "pro_*"), "质量管理": (6, "质量管理模块", "qc_*"), "设备管理": (7, "设备管理模块", "dv_*"), "仓储管理": (8, "仓库管理模块", "wm_*"), "排班管理": (9, "排班管理模块", "cal_*"), "IoT数据采集": (10, "IoT数据采集模块", "iot_*"), "工装夹具管理": (11, "工具管理模块", "tm_*"), "系统管理": (12, "系统管理模块", "sys_*"), "打印机配置": (13, "打印管理模块", "print_*"), } # Excel 工作表名 -> 表清单中的模块名 SHEET_TO_MODULE = { "MD主数据": "主数据管理", "PRO生产管理": "生产管理", "CAL排班管理": "排班管理", "QC质量管理": "质量管理", "WM仓储管理": "仓储管理", "DV设备管理": "设备管理", "TM工装夹具管理": "工装夹具管理", "MO模具管理": None, "SYS系统管理": "系统管理", "PRINT打印管理": "打印机配置", "IOT数采": "IoT数据采集", } def load_table_list(wb): """从「表清单」sheet 加载 (表名, 模块, 表描述, 备注)，并建 (模块, 表描述)->表名 映射""" ws = wb["表清单"] rows = list(ws.iter_rows(min_row=2, values_only=True)) table_list = [] desc_to_name = {} for row in rows: if not row or not row[1]: continue name = (row[1] or "").strip() module = (row[2] or "").strip() desc = (row[3] or "").strip() remark = (row[4] or "").strip() if len(row) > 4 else "" if not name: continue table_list.append((name, module, desc, remark)) if module and desc: key = (module, desc) if key not in desc_to_name: desc_to_name[key] = name return table_list, desc_to_name def parse_sheet_tables(ws, module_name, desc_to_name): """ 解析一个 sheet 中所有表块。 每个表块：一行 表名称 + 一行 中文名称 + 一行 列序号|列名|列中文名|列类型|列设置|备注 + 数据行。 返回 [(table_name, cn_name, remark, columns), ...] """ rows = list(ws.iter_rows(values_only=True)) result = [] i = 0 while i < len(rows): row = list(rows[i]) if rows[i] else [] # 找「表名称」在第二列 if len(row) >= 3 and row[1] == "表名称" and row[2]: excel_table_name = (row[2] or "").strip() cn_name = "" if i + 1 < len(rows): next_row = list(rows[i + 1]) if rows[i + 1] else [] if len(next_row) >= 3 and next_row[1] == "中文名称": cn_name = (next_row[2] or "").strip() # 表名解析：优先用 表清单 中 (模块, 中文名称) 对应的表名 if module_name and cn_name and (module_name, cn_name) in desc_to_name: table_name = desc_to_name[(module_name, cn_name)] else: table_name = excel_table_name if excel_table_name and len(excel_table_name) > 2 else (cn_name or excel_table_name) # 列头在 i+2 col_start = i + 2 if col_start >= len(rows): i += 1 continue header = list(rows[col_start]) if rows[col_start] else [] if len(header) < 5 or header[1] != "列序号": i += 1 continue columns = [] for j in range(col_start + 1, len(rows)): r = list(rows[j]) if rows[j] else [] if len(r) < 4: break col_no, col_name, col_cn, col_type, col_setting, col_remark = ( r[1], r[2] if len(r) > 2 else "", r[3] if len(r) > 3 else "", r[4] if len(r) > 4 else "", r[5] if len(r) > 5 else "", r[6] if len(r) > 6 else "" ) if col_name is None or not str(col_name).strip(): break col_name = str(col_name).strip() col_type = (col_type or "").strip() nullable = "N" if col_setting and "not null" in str(col_setting).lower() else "Y" desc = (col_cn or "").strip() if col_remark and str(col_remark).strip(): desc = desc + " " + str(col_remark).strip() columns.append({ "name": col_name.lower(), "type": col_type or "-", "nullable": nullable, "desc": desc or "-", }) result.append((table_name, cn_name, "", columns)) i = col_start + len(columns) + 1 continue i += 1 return result def collect_all_tables(wb, table_list, desc_to_name): """从各数据 sheet 收集表，按模块归类。返回 { 模块名: [(表名, 中文名, 备注, columns), ...] }""" by_module = {} for sheet_name in wb.sheetnames: if sheet_name == "表清单": continue module_name = SHEET_TO_MODULE.get(sheet_name) if not module_name or module_name not in MODULE_DOC: continue ws = wb[sheet_name] tables = parse_sheet_tables(ws, module_name, desc_to_name) for t in tables: by_module.setdefault(module_name, []).append(t) return by_module def table_name_to_lower(name): """将 Excel 表名转为小写（如 WM_WAREHOUSE -> wm_warehouse）""" if not name: return name return name.strip().lower() def gen_table_list_md(table_list): """生成 2.2 表清单总览 markdown""" module_order = list(MODULE_DOC.keys()) # 模块简写（用于表格第三列） module_short = { "主数据管理": "基础数据", "生产管理": "生产管理", "质量管理": "质量管理", "设备管理": "设备管理", "仓储管理": "仓库管理", "排班管理": "排班管理", "IoT数据采集": "IoT数据采集", "工装夹具管理": "工具管理", "系统管理": "系统管理", "打印机配置": "打印管理", } lines = [ "### 2.2 表清单总览", "", "| 序号 | 表名 | 中文名 | 模块 |", "|------|------|--------|------|", ] for idx, (name, module, desc, _) in enumerate(table_list, 1): short = module_short.get(module, module) cn_display = (desc or "").strip() or name # 表描述为空时用表名 lines.append(f"| {idx} | {table_name_to_lower(name)} | {cn_display} | {short} |") lines.append("") return "\n".join(lines) def gen_module_section(module_name, tables, section_num): """生成一个模块的 markdown（## 4. 基础数据模块 及下属 ### 4.x 表）""" if not tables: return "" title = MODULE_DOC[module_name][1] lines = [ f"## {section_num}. {title}", "", ] for idx, (table_name, cn_name, remark, columns) in enumerate(tables, 1): tname_lower = table_name_to_lower(table_name) lines.append(f"### {section_num}.{idx} {cn_name or table_name} ({tname_lower})") lines.append("") if remark: lines.append(f"> {remark}") lines.append("") lines.append("| 字段名 | 类型 | 允许空 | 说明 |") lines.append("|--------|------|--------|------|") for col in columns: lines.append(f"| {col['name']} | {col['type']} | {col['nullable']} | {col['desc']} |") lines.append("") lines.append("---") lines.append("") return "\n".join(lines) def main(): if not XLSX_PATH.exists(): raise SystemExit(f"Excel 文件不存在: {XLSX_PATH}") if not MD_PATH.exists(): raise SystemExit(f"Markdown 文件不存在: {MD_PATH}") wb = openpyxl.load_workbook(XLSX_PATH, read_only=True, data_only=True) table_list, desc_to_name = load_table_list(wb) by_module = collect_all_tables(wb, table_list, desc_to_name) wb.close() # 生成 2.2 表清单 table_list_md = gen_table_list_md(table_list) # 生成 4-12 节（按 MODULE_DOC 顺序） module_sections = [] for mod_name in MODULE_DOC: if mod_name not in by_module: continue num, _, _ = MODULE_DOC[mod_name] module_sections.append(gen_module_section(mod_name, by_module[mod_name], num)) new_content_4_13 = "\n".join(module_sections) # 读取现有 md md_text = MD_PATH.read_text(encoding="utf-8") # 替换 2.2 表清单总览：从 ### 2.2 表清单总览 到下一个 ## 之前 def replace_2_2(content): start = content.find("### 2.2 表清单总览") if start == -1: return content end = content.find("\n## ", start + 1) if end == -1: end = len(content) return content[:start] + table_list_md.rstrip() + "\n\n" + content[end:] md_text = replace_2_2(md_text) # 替换 ## 4. ... 到 ## 14. 之前 start_marker = "## 4. 基础数据模块" end_marker = "\n## 14. 枚举值定义" start = md_text.find(start_marker) end = md_text.find(end_marker) if start == -1: raise SystemExit("未找到「## 4. 基础数据模块」") if end == -1: end = len(md_text) md_text = md_text[:start] + new_content_4_13.rstrip() + "\n\n" + md_text[end:] MD_PATH.write_text(md_text, encoding="utf-8") print(f"已更新: {MD_PATH}") print(f" - 表清单: {len(table_list)} 张表") for mod_name, tbls in by_module.items(): print(f" - {mod_name}: {len(tbls)} 张表") if __name__ == "__main__": main()