276 lines
10 KiB
Python
276 lines
10 KiB
Python
#!/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()
|