raw_data_check.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. r"""输入数据放置体检 + 增量放置指导 (2026-09-17 用户令 3)。
  4. 用户要的是两件事:
  5. ① **检查**放进去的输入数据(特别是**增量**放的)是否符合规则;
  6. ② 给出**放置指导**(该放哪、命名怎么写、放完跑什么)。
  7. ## 规则 (每条都能指到具体文件, 依据 = 数据目录约定 + 各摄入脚本的真实读法)
  8. R1 顶层只许 场站目录 / 共享资料(西门子4.0技术资料) / 说明文件(*.txt|*.md);
  9. 常见错误: 把 10 个 csv 直接解到 data/raw 根下, 或把整包 zip 丢在根上。
  10. R2 场站目录下只许约定子目录 (src/windscada/config.py::STATION_SUBDIRS);
  11. 常见错误: 解压时多套一层包名目录 (如 `10分钟数据/10分钟数据/WTG01.csv`)。
  12. R3 各源类的结构与命名:
  13. scada_10min/ scada_1min/ 平铺 `<机组>.csv`, 机组集合要与场配置一致
  14. 故障报警/ .xls/.xlsx/.xml, 不许多套层
  15. 风机故障记录/ 年目录 + .xls/.xlsx/.rar/.jpg
  16. 油样报告/ 两级 `台号/部件/*.pdf` (维护页按此分目录)
  17. windcms/ 含 CMS 原始导出: `*_decode.json` 或 `<包名>/measurement/**`
  18. m5_cms_tcm/ 含 handoff 正本 (handoff_vibration_v2.json / component_history.json) + 厂家报告/
  19. R4 可读性抽样: CSV 能按 utf-8/gbk 解出非空表头; TCM `*_decode.json` 能解析出场站/机组字段。
  20. (`--deep` 才做, 默认只查结构与命名, 免得在 GB 级目录上耗时)
  21. R5 **增量语义** (给了 --src 时): 逐件对比"源包会落到哪"与"现在有什么", 出三清单:
  22. 新增 / 相同(将跳过) / **冲突**(同名不同大小 ⇒ 会覆盖已摄入的数据, 必须人来定夺)。
  23. R6 时间空洞: scada_* 的文件名里若带年月, 检查是否缺月/重月 (数据层页面的时间轴会缺一段)。
  24. R7 体量提示: 单文件 > 2 GB、目录 > 60 GB、或混进 zip/rar (应先解压再放)。
  25. R8 放置指导: 按"新增了哪类源"给出该跑的重算命令 (台账/SCADA/振动/全量)。
  26. ## 退出码
  27. 0 合规(可能带提示) · 5 结构或命名违例 · 6 增量冲突(同名不同大小) · 7 缺源类/时间空洞 · 8 仅有提示
  28. ## 用法
  29. python scripts/raw_data_check.py # 体检现有 data/raw/<场站>
  30. python scripts/raw_data_check.py --deep # 加可读性抽样
  31. python scripts/raw_data_check.py --src <现场包目录> # 增量体检: 出新增/相同/冲突三清单
  32. python scripts/raw_data_check.py --src <现场包> --json
  33. python scripts/raw_data_check.py --write-doc # 把规则表与结论写进放置指导文档
  34. """
  35. from __future__ import annotations
  36. import argparse
  37. import json
  38. import os
  39. import pathlib
  40. import re
  41. import sys
  42. import zipfile
  43. ROOT = pathlib.Path(__file__).resolve().parents[1]
  44. sys.path.insert(0, str(ROOT))
  45. from src import paths as P # noqa: E402
  46. RC_OK, RC_STRUCT, RC_CLASH, RC_GAP, RC_NOTE = 0, 5, 6, 7, 8
  47. TOP_OK_DIRS = ('西门子4.0技术资料',) # data/raw 下与"场站目录"并列的共享资料
  48. TOP_OK_FILES = ('.txt', '.md', '.pdf') # 说明类文件
  49. ZIP_MOD = {'.zip', '.rar', '.7z', '.tar', '.gz', '.zst'}
  50. # Access 读写 .mdb 时留下的锁/临时文件 (不是源数据; 2026-09-19 实逮: MDB 重建期间 scada_mdb 下出现 .laccdb)
  51. ACCESS_TMP_SUFFIX = {".laccdb", ".ldb", ".tmp"}
  52. BIG_FILE = 2 * 1024 ** 3
  53. BIG_DIR = 60 * 1024 ** 3
  54. # 各类源的期望 (后缀集合, 说明) —— 与摄入脚本的真实读法一致 (见 place_raw_data.py 的映射表注释)
  55. EXPECT = {
  56. 'scada_10min': ({'.csv'}, '平铺 <机组>.csv (每台一个文件, 不许再套层)'),
  57. 'scada_1min': ({'.csv'}, '平铺 <机组>.csv'),
  58. '故障报警': ({'.xls', '.xlsx', '.xml'}, '年度/季度报警导出 (.xls) + XML 报警导出'),
  59. '风机故障记录': ({'.xls', '.xlsx', '.rar', '.jpg', '.png'}, '年目录 + 月度汇总表/现场照片'),
  60. '油样报告': ({'.pdf'}, '两级: <台号>/<部件>/*.pdf'),
  61. 'windcms': ({'.json', '.pdf', '.docx'}, 'CMS 原始导出 *_decode.json(可套 <包名>/measurement/ 层) + 厂家报告/'),
  62. 'm5_cms_tcm': ({'.json', '.docx', '.pdf', '.md'}, 'handoff 正本 + 厂家报告/'),
  63. # 2026-09-17 用户令: 现场交来的年度 MDB 归档(25年.zip / 26年.zip)就地放在这里 —— 它就是
  64. # 原始收资(Access 按月分类),取代此前的"已导出 CSV"两个 zip(10分钟数据.zip / 1分钟数据.zip)。
  65. # 允许 .zip(原样归档)+ .mdb(已解出时);解出的目录层次 `<年>/<月>/` 由 R2 的子目录规则放行。
  66. 'scada_mdb': ({'.zip', '.mdb'}, '年度归档 <年>年.zip(内层 <年>/<月>/<年-月-类>.zip → .mdb)'),
  67. }
  68. DATE_RE = re.compile(r'(20\d{2})[-_年]?(0[1-9]|1[0-2])')
  69. def station_dir(farm: str | None = None) -> pathlib.Path:
  70. """当前场的原始件目录: data/raw/<场站目录名>。"""
  71. from src.windscada import config as C
  72. try:
  73. d = C.raw_station_dir(farm)
  74. except BaseException:
  75. d = P.RAW_ROOT / (farm or P.farm())
  76. return pathlib.Path(d)
  77. def sizes_map(base: pathlib.Path) -> dict:
  78. """{相对路径: 字节数} —— 只算文件。"""
  79. out = {}
  80. if base.is_dir():
  81. for f in base.rglob('*'):
  82. if f.is_file():
  83. out[f.relative_to(base).as_posix()] = f.stat().st_size
  84. return out
  85. def check_tree(st: pathlib.Path, deep=False) -> tuple[list, list, dict]:
  86. """规则体检 → (问题列表[(level, 项, 说明, rc)], 提示列表, 统计)。"""
  87. res, notes = [], []
  88. stat = dict(exists=st.is_dir(), files=0, bytes=0, subdirs={}, units=[])
  89. # R1 data/raw 顶层
  90. if P.RAW_ROOT.is_dir():
  91. for p in sorted(P.RAW_ROOT.iterdir()):
  92. if p.is_dir():
  93. if p.name != st.name and p.name not in TOP_OK_DIRS:
  94. res.append(('!', f'data/raw/{p.name}', f'顶层目录未在约定内 (场站目录应为 {st.name}/; '
  95. f'共享资料放 {TOP_OK_DIRS})', RC_STRUCT))
  96. elif p.suffix.lower() not in TOP_OK_FILES:
  97. res.append(('!', f'data/raw/{p.name}', 'data/raw 顶层只放场站目录与说明文件; '
  98. '源件要放进 <场站>/<源类>/', RC_STRUCT))
  99. if not st.is_dir():
  100. notes.append(('i', f'{P.rel(st)}', '尚未放置输入数据 —— 页面会显示"无产物/无数据" (属预期); '
  101. '放置见 docs/输入数据放置指导_v0.1.md', RC_OK))
  102. return res, notes, stat
  103. # R2 场站内子目录 + R3 各源类
  104. from src.windscada import config as C
  105. allowed = set(C.STATION_SUBDIRS)
  106. for d in sorted(st.iterdir()):
  107. if d.is_file():
  108. if d.suffix.lower() in ZIP_MOD:
  109. res.append(('!', P.rel(d), '压缩包直接放在场站目录下 —— 先解压再按源类落位', RC_NOTE))
  110. else:
  111. notes.append(('i', P.rel(d), '场站目录下的散装文件 (多半是现场随手拷的; 建议归入对应源类子目录)', RC_OK))
  112. continue
  113. if d.name not in allowed:
  114. res.append(('!', f'{P.rel(d)}', f'未约定的子目录 (约定: {sorted(allowed)})'
  115. ' —— 常见成因: 解压时多套了一层包名', RC_STRUCT))
  116. continue
  117. files = [f for f in d.rglob('*') if f.is_file()]
  118. nbytes = sum(f.stat().st_size for f in files)
  119. stat['subdirs'][d.name] = dict(files=len(files), bytes=nbytes)
  120. stat['files'] += len(files)
  121. stat['bytes'] += nbytes
  122. exts, bad = {}, []
  123. for f in files:
  124. e = f.suffix.lower()
  125. exts[e] = exts.get(e, 0) + 1
  126. want = EXPECT.get(d.name, (None, ''))[0]
  127. # ★2026-09-19: Access 的**锁文件/临时文件**(.laccdb/.ldb/~$*) 不是源数据 ——
  128. # 读/写 .mdb 时 Access 会在同目录留下它(实测: MDB 重建期间 scada_mdb 下出现 .laccdb),
  129. # 按"后缀不在约定内"报结构问题会把人引到错的地方。忽略并如实注明。
  130. if e in ACCESS_TMP_SUFFIX or f.name.startswith('~$'):
  131. notes.append(('i', P.rel(f), 'Access 锁/临时文件 (读写 .mdb 时产生), 不算源件', RC_OK))
  132. continue
  133. if want and e not in want and e not in ZIP_MOD:
  134. bad.append(f)
  135. if bad:
  136. sample = ', '.join(sorted({f.suffix.lower() for f in bad})[:4])
  137. res.append(('!', P.rel(d), f'{len(bad)} 个文件后缀不在该源类的约定内 ({sample}); '
  138. f'约定: {sorted(EXPECT[d.name][0])}', RC_STRUCT))
  139. if not files:
  140. notes.append(('i', P.rel(d), '目录是空的 (放了数据才有用)', RC_OK))
  141. # 各类专项
  142. if d.name in ('scada_10min', 'scada_1min'):
  143. names = {f.name for f in files}
  144. nested = [f for f in files if len(f.relative_to(d).parts) > 1]
  145. if nested:
  146. res.append(('!', P.rel(d), f'{len(nested)} 个文件在子目录里 —— 摄入按 <源类>/<机组>.csv 平铺读, '
  147. f'多一层就读不到', RC_STRUCT))
  148. try:
  149. want = set(C.farm(None)['turbines'])
  150. n_want = len(want)
  151. except BaseException:
  152. want, n_want = set(), 0
  153. got = {f.stem for f in files if f.parent == d}
  154. # ★ 两类命名都对: scada_10min 用显示机组号 (WTG01.csv); scada_1min 的现场导出用的是
  155. # **内部台号** (01E.csv…37B.csv, 见 place_raw_data.py 的映射说明) —— 后者按"台数一致"判完整,
  156. # 不再按名字判 (2026-09-17 实测: 按 WTG 名字判会把 38/38 齐全的 1min 目录误报成"缺 38 台")。
  157. if d.name == 'scada_10min':
  158. miss = sorted(want - got) if want else []
  159. extra = sorted(got - want) if want else []
  160. stat['units'].append((d.name, len(got)))
  161. if miss:
  162. res.append(('!', P.rel(d), f'缺 {len(miss)} 台机组的数据: {miss[:6]}'
  163. f'{" …" if len(miss) > 6 else ""} (页面按台取数, 缺台就是空白)', RC_GAP))
  164. if extra:
  165. res.append(('!', P.rel(d), f'{len(extra)} 个文件名不是本场机组: {extra[:6]}', RC_STRUCT))
  166. else:
  167. stat['units'].append((d.name, len(got)))
  168. if n_want and len(got) != n_want:
  169. res.append(('!', P.rel(d), f'台数不符: 现有 {len(got)} 个 <台号>.csv, 场配置是 {n_want} 台'
  170. f' (1min 导出用内部台号 01E/37B 这类, 按台数判完整)', RC_GAP))
  171. # R6 时间空洞 (文件名带年月时才判)
  172. months = sorted({m.group(0) for m in (DATE_RE.search(f.stem) for f in files) if m})
  173. if months:
  174. stat['months'] = months
  175. if d.name == '油样报告':
  176. flat = [f for f in files if len(f.relative_to(d).parts) < 2]
  177. if flat:
  178. res.append(('!', P.rel(d), f'{len(flat)} 个 PDF 直接躺在源类目录下 —— 约定是 <台号>/<部件>/*.pdf '
  179. f'(维护页按此分组)', RC_STRUCT))
  180. if d.name == 'windcms':
  181. js = [f for f in files if f.suffix.lower() == '.json']
  182. dec = [f for f in js if f.name.endswith('_decode.json')] # TCM 导出本体
  183. if not dec:
  184. res.append(('!', P.rel(d), '没有 *_decode.json (CMS 原始导出) —— 振动侧就没有可重算的源件', RC_GAP))
  185. if deep and dec:
  186. # 抽样核对"摄入脚本真正要的结构" —— 依据 scripts/rudong_tcm_index.py:
  187. # doc['body']['body'][<ISO时间戳>] = [ {Record: {...}}, … ], 机组取 Record.Location / Record.LocationName
  188. # (顶层还有 method/controller/buildId 等信封字段, 别拿它们当判据 —— 2026-09-17 实测踩过这个坑)
  189. import random as _rnd
  190. _rnd.seed(0)
  191. sample = _rnd.sample(dec, min(12, len(dec)))
  192. ok_n, bad_n, shapes = 0, 0, {}
  193. for f in sample:
  194. try:
  195. o = json.loads(f.read_text(encoding='utf-8', errors='replace'))
  196. body = ((o or {}).get('body') or {}).get('body') if isinstance(o, dict) else None
  197. if not isinstance(body, dict) or not body:
  198. shapes[str(sorted(o)[:4]) if isinstance(o, dict) else type(o).__name__] = \
  199. shapes.get(str(sorted(o)[:4]) if isinstance(o, dict) else type(o).__name__, 0) + 1
  200. bad_n += 1
  201. continue
  202. recs = next((v for v in body.values() if isinstance(v, list) and v), None)
  203. keys = sorted((recs[0] or {}).get('Record', {}) or {}) if recs else []
  204. if recs and (('Location' in keys) or ('LocationName' in keys)):
  205. ok_n += 1
  206. else:
  207. shapes[f'Record 缺 Location/LocationName: {keys[:6]}'] = \
  208. shapes.get(f'Record 缺 Location/LocationName: {keys[:6]}', 0) + 1
  209. bad_n += 1
  210. except Exception as e:
  211. shapes[f'解析失败 {type(e).__name__}'] = shapes.get(f'解析失败 {type(e).__name__}', 0) + 1
  212. bad_n += 1
  213. stat['tcm_sample'] = dict(ok=ok_n, bad=bad_n, shapes=shapes)
  214. if bad_n:
  215. res.append(('!', P.rel(d), f'TCM 抽样 {len(sample)} 件里 {bad_n} 件结构对不上摄入脚本的取法 '
  216. f'(body.body[时间戳]=[{{Record:…}}], 机组取 Record.Location/LocationName): '
  217. f'{list(shapes.items())[:3]}', RC_STRUCT))
  218. else:
  219. stat['tcm_ok'] = True
  220. if d.name == 'm5_cms_tcm':
  221. hand = [f for f in files if f.name in ('handoff_vibration_v2.json', 'component_history.json')]
  222. if not hand:
  223. notes.append(('i', P.rel(d), '缺 handoff 正本 (handoff_vibration_v2.json / component_history.json) —— '
  224. '融合面判级与 /cms/ 只能用随包快照 (docs §7 的已知缺口)', RC_OK))
  225. # R4 可读性抽样 (CSV 编码/表头)
  226. if deep:
  227. for name in ('scada_10min', 'scada_1min'):
  228. d = st / name
  229. if d.is_dir():
  230. f = next(iter(sorted(x for x in d.glob('*.csv'))), None)
  231. if f:
  232. head = None
  233. for enc in ('utf-8-sig', 'gbk'):
  234. try:
  235. with f.open(encoding=enc, errors='strict') as fh:
  236. head = fh.readline().strip()
  237. stat[f'{name}_enc'] = enc
  238. break
  239. except Exception:
  240. continue
  241. if not head:
  242. res.append(('!', P.rel(f), 'CSV 表头读不出来 (utf-8/gbk 都不行)', RC_STRUCT))
  243. elif len(head.split(',')) < 5:
  244. res.append(('!', P.rel(f), f'表头列数异常 ({len(head.split(","))} 列): {head[:60]}', RC_STRUCT))
  245. # R7 体量
  246. if stat['bytes'] > BIG_DIR:
  247. notes.append(('?', P.rel(st), f'源件合计 {stat["bytes"]/2**30:.1f} GB —— 摄入与重算会很慢, 建议分批', RC_NOTE))
  248. return res, notes, stat
  249. def increment_findings(plan: dict, src_name: str = '') -> list:
  250. """把增量三清单里的**冲突**变成一条可拦人的问题 (供本模块与 place_raw_data 共用)。"""
  251. out = []
  252. if plan.get('clash'):
  253. c = plan['clash'][0]
  254. out.append(('!', f'--src {src_name}'.strip(), f'{len(plan["clash"])} 件**同名不同大小**: 直接放会覆盖已摄入的源件 '
  255. f'(示例 {c["path"]}: 现有 {c.get("now", 0):,} B → 源包 '
  256. f'{c["size"]:,} B) —— 先确认哪一份是对的, 再决定 --force 覆盖还是改名并存',
  257. RC_CLASH))
  258. return out
  259. def increment_src(src: pathlib.Path, st: pathlib.Path) -> tuple[list, dict]:
  260. """R5 增量三清单: 遍历源包 zip 成员/散件, 算出它们会落到哪, 与现有树对比。"""
  261. res, plan = [], dict(new=[], same=[], clash=[], unmapped=[])
  262. import importlib.util
  263. spec = importlib.util.spec_from_file_location('_place', ROOT / 'scripts' / 'place_raw_data.py')
  264. place = importlib.util.module_from_spec(spec)
  265. try:
  266. spec.loader.exec_module(place)
  267. except SystemExit:
  268. pass
  269. rules = list(getattr(place, 'RULES', [])) + list(getattr(place, 'RULES_MECH', [])) + list(getattr(place, 'RULES_VIB', []))
  270. loose = list(getattr(place, 'RULES_VIB_LOOSE', []))
  271. have = sizes_map(st)
  272. have_root = sizes_map(P.RAW_ROOT)
  273. def classify(rel_path: str, size: int, base: pathlib.Path):
  274. target = base / rel_path
  275. cur = have.get(rel_path) if base == st else have_root.get(rel_path)
  276. item = dict(path=rel_path, size=size)
  277. if cur is None:
  278. plan['new'].append(item)
  279. elif cur == size:
  280. plan['same'].append(item)
  281. else:
  282. item['now'] = cur
  283. plan['clash'].append(item)
  284. for zname, prefix, target, strip, root_kind in rules:
  285. zp = src / zname if src.is_dir() else None
  286. if not zp or not zp.is_file():
  287. continue
  288. base = st if root_kind == 'station' else P.RAW_ROOT
  289. try:
  290. with zipfile.ZipFile(zp) as z:
  291. for info in z.infolist():
  292. if info.is_dir():
  293. continue
  294. nm = place.gbk_name(info)
  295. if prefix and not nm.startswith(prefix):
  296. continue
  297. tail = nm[len(prefix):] if prefix else nm
  298. if strip:
  299. tail = tail.split('/', 1)[1] if '/' in tail else tail
  300. rel = f'{target}/{tail}'
  301. if prefix and nm == prefix.rstrip('/'):
  302. continue
  303. classify(rel, info.file_size, base)
  304. except zipfile.BadZipFile:
  305. res.append(('!', zp.name, '不是合法 zip (现场包常被截断; 重新拷一份)', RC_STRUCT))
  306. for fn, target in loose:
  307. fp = src / fn
  308. if fp.is_file():
  309. classify(f'{target}/{fn}', fp.stat().st_size, st)
  310. return res, plan
  311. def guidance(plan: dict, stat: dict) -> list[str]:
  312. """R8 放置指导: 按新增源类给出该跑的重算命令与预期变化。"""
  313. out = []
  314. new = [x['path'] for x in plan.get('new', [])]
  315. kinds = {p.split('/', 1)[0] for p in new}
  316. if not new:
  317. out.append('没有新增源件: 不必重算 (若刚换过数据, 用 scripts/rebuild_all.py --dry-run 看计划)')
  318. return out
  319. out.append(f'新增 {len(new)} 件, 涉及源类: {sorted(kinds)}')
  320. if 'scada_10min' in kinds or 'scada_1min' in kinds:
  321. out.append('· SCADA 侧变了 → `python scripts/rebuild_all.py` (含 10 个构建器, 约 15 分钟); '
  322. '台账/报警/油样一起换时同一条命令即可')
  323. if kinds & {'故障报警', '风机故障记录', '油样报告'}:
  324. out.append('· 台账类变了 → `python scripts/rebuild_all.py --skip-scada` (约 2 分钟)')
  325. if kinds & {'windcms', 'm5_cms_tcm'}:
  326. out.append('· 振动侧变了 → 重算链第 ④b 步会自动摄入 (`scripts/vib_raw_build.py`); '
  327. '注意它只做索引/谱, 不会重生成报告 (本包缺六层链, 见 docs §7)')
  328. out.append('· 放完先自检: `python scripts/raw_data_check.py --deep` → 再看 `/detail/` 左栏「系统维护」两屏')
  329. out.append('· 重算后核对锚点: 报警 39211 · 工单 5876 · 油样 404 · temp_monthly 19494 · 本体 9702 对象')
  330. return out
  331. def main() -> int:
  332. ap = argparse.ArgumentParser()
  333. ap.add_argument('--farm', default=None)
  334. ap.add_argument('--src', default=None, help='现场数据包目录 (给了就做增量体检: 新增/相同/冲突三清单)')
  335. ap.add_argument('--deep', action='store_true', help='加可读性抽样 (CSV 编码/表头, TCM json 字段)')
  336. ap.add_argument('--json', action='store_true')
  337. ap.add_argument('--write-doc', action='store_true', help='把结论写进 docs/输入数据放置指导_v0.1.md')
  338. a = ap.parse_args()
  339. st = station_dir(a.farm)
  340. res, notes, stat = check_tree(st, deep=a.deep)
  341. plan = {}
  342. if a.src:
  343. src = pathlib.Path(a.src)
  344. if not src.is_dir():
  345. print(f'[X] --src 不是目录: {src}')
  346. return RC_STRUCT
  347. r2, plan = increment_src(src, st)
  348. res += r2 + increment_findings(plan, src.name)
  349. guide = guidance(plan, stat)
  350. if a.json:
  351. print(json.dumps(dict(station=P.rel(st), results=[dict(level=x[0], item=x[1], note=x[2], rc=x[3]) for x in res],
  352. notes=[dict(level=x[0], item=x[1], note=x[2]) for x in notes], stat=stat, plan=plan,
  353. guidance=guide), ensure_ascii=False, indent=1, default=str))
  354. else:
  355. print(f'== 输入数据放置体检 · {P.rel(st)} ==')
  356. if stat.get('exists'):
  357. print(f' {stat["files"]:,} 件 / {stat["bytes"]/2**30:.2f} GB')
  358. for k, v in sorted(stat['subdirs'].items()):
  359. print(f' {k:14s} {v["files"]:7,d} 件 {v["bytes"]/2**20:9.1f} MB')
  360. for level, item, note, rc in res:
  361. print(f' [{level}] {item}: {note}')
  362. seen = set()
  363. for level, item, note, rc in notes:
  364. if note in seen:
  365. continue
  366. seen.add(note)
  367. print(f' [{level}] {item}: {note}')
  368. if plan:
  369. print('\n== 增量放置三清单 (源包 → data/raw) ==')
  370. print(f' 新增 {len(plan["new"]):,} 件 (将拷入) · 相同 {len(plan["same"]):,} 件 (同尺寸, 跳过) · '
  371. f'冲突 {len(plan["clash"]):,} 件 (同名不同大小, **需人确认**)')
  372. for k, label in (('new', '新增'), ('clash', '冲突')):
  373. for x in plan[k][:5]:
  374. extra = f' (现有 {x.get("now"):,} B)' if k == 'clash' else ''
  375. print(f' [{label}] {x["path"]} {x["size"]:,} B{extra}')
  376. if len(plan[k]) > 5:
  377. print(f' … 另有 {len(plan[k]) - 5} 件{label}')
  378. print('\n== 放置指导 ==')
  379. for g in guide:
  380. print(f' {g}')
  381. rc_map = {x[3] for x in res if x[0] in ('X', '!') and x[3]}
  382. rc = max(rc_map) if rc_map else (RC_NOTE if notes else RC_OK)
  383. if not a.json:
  384. print(f'\n结论: {"合规" if rc == 0 else "见上"}; 退出码 {rc}')
  385. if a.write_doc:
  386. write_doc(res, notes, stat, plan, guide)
  387. return rc
  388. DOC = P.ROOT / 'docs' / '输入数据放置指导_v0.1.md'
  389. BEGIN, END = '<!-- RAWCHECK:BEGIN -->', '<!-- RAWCHECK:END -->'
  390. def write_doc(res, notes, stat, plan, guide) -> None:
  391. body = [BEGIN, '### 体检结论(自动生成,勿手改)', '',
  392. f'· 源件目录 `{P.rel(station_dir())}`:{stat.get("files", 0):,} 件 / {stat.get("bytes", 0)/2**30:.2f} GB']
  393. for k, v in sorted((stat.get('subdirs') or {}).items()):
  394. body.append(f' - `{k}/`:{v["files"]:,} 件 / {v["bytes"]/2**20:.1f} MB')
  395. if res:
  396. body += ['', '| 级别 | 项 | 说明 |', '|---|---|---|']
  397. body += [f'| `{x[0]}` | `{x[1]}` | {x[2]} |' for x in res[:40]]
  398. else:
  399. body += ['', '· 结构与命名检查:**无不一致**(提示见下)']
  400. for x in notes[:12]:
  401. body.append(f'· `{x[0]}` {x[1]}:{x[2]}')
  402. if plan:
  403. body += ['', f'· 增量(源包 → data/raw):新增 {len(plan["new"]):,} · 相同 {len(plan["same"]):,} · '
  404. f'冲突 {len(plan["clash"]):,}']
  405. body += ['', '**放置后该做什么**:'] + [f'· {g}' for g in guide] + ['', END]
  406. block = '\n'.join(body)
  407. if DOC.is_file():
  408. t = DOC.read_text(encoding='utf-8')
  409. if BEGIN in t and END in t:
  410. t = re.sub(re.escape(BEGIN) + r'.*?' + re.escape(END), lambda m: block, t, flags=re.S)
  411. else:
  412. t = t.rstrip() + '\n\n' + block + '\n'
  413. else:
  414. t = '# 输入数据放置指导\n\n' + block + '\n'
  415. DOC.write_text(t, encoding='utf-8')
  416. print(f' 已写入 {P.rel(DOC)}')
  417. if __name__ == '__main__':
  418. from src import console
  419. console.soft()
  420. sys.exit(main())