79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
"""Приводит fig_*.puml к оформлению под ГОСТ: !include темы, без title и локальных skinparam."""
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from pathlib import Path
|
|
|
|
PUML_DIR = Path(__file__).resolve().parents[1] / "puml"
|
|
INCLUDE_LINE = "!include _gost-theme.inc.puml\n"
|
|
|
|
|
|
def strip_title_and_skinparam(text: str) -> str:
|
|
lines = text.splitlines(keepends=True)
|
|
out: list[str] = []
|
|
i = 0
|
|
while i < len(lines):
|
|
line = lines[i]
|
|
stripped = line.strip()
|
|
if stripped.startswith("skinparam"):
|
|
if "{" in stripped and "}" not in stripped:
|
|
i += 1
|
|
while i < len(lines) and "}" not in lines[i]:
|
|
i += 1
|
|
if i < len(lines):
|
|
i += 1
|
|
else:
|
|
i += 1
|
|
continue
|
|
if stripped.startswith("title"):
|
|
if stripped == "title":
|
|
i += 1
|
|
while i < len(lines) and lines[i].strip() != "end title":
|
|
i += 1
|
|
if i < len(lines):
|
|
i += 1
|
|
continue
|
|
i += 1
|
|
continue
|
|
if stripped in ("BackgroundColor", "BorderColor", "FontColor") or stripped.endswith(
|
|
("#F8F8F8", "#333333", "#E8F4FF", "#FFFDE7", "#F9A825")
|
|
):
|
|
i += 1
|
|
continue
|
|
if stripped == "}" and out and out[-1].strip().startswith("!include"):
|
|
i += 1
|
|
continue
|
|
out.append(line)
|
|
i += 1
|
|
return "".join(out)
|
|
|
|
|
|
def normalize_file(path: Path) -> bool:
|
|
text = path.read_text(encoding="utf-8")
|
|
original = text
|
|
text = strip_title_and_skinparam(text)
|
|
if INCLUDE_LINE.strip() not in text:
|
|
m = re.match(r"(@startuml\s+\S+\s*\n)", text)
|
|
if m:
|
|
text = text[: m.end()] + INCLUDE_LINE + text[m.end() :]
|
|
text = re.sub(r"\n{3,}", "\n\n", text)
|
|
if text != original:
|
|
path.write_text(text, encoding="utf-8")
|
|
return True
|
|
return False
|
|
|
|
|
|
def main() -> int:
|
|
changed = 0
|
|
for path in sorted(PUML_DIR.glob("fig_*.puml")):
|
|
if normalize_file(path):
|
|
changed += 1
|
|
print(f"updated: {path.name}")
|
|
print(f"Done. {changed} file(s) changed.")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|