gabriel / muse public
revert_json_alias.py python
46 lines 1.2 KB
Raw
sha256:d11a87833d5fad6059b7662844bf5448a8911a17cce7a51811f71ad394f248eb bump to v0.2.0rc13 Human patch 7 days ago
1 """Remove all --json store_const aliases inserted by add_json_alias.py.
2
3 Strips every block of the form:
4 <var>.add_argument(
5 "--json", action="store_const", const="json", dest="fmt",
6 help="Shorthand for --format json."
7 )
8
9 Safe to run multiple times.
10 """
11 from __future__ import annotations
12
13 import pathlib
14 import re
15
16 _COMMANDS_DIR = pathlib.Path(__file__).parent.parent / "muse" / "cli" / "commands"
17
18 # Match the exact 4-line block the insertion script emits (any indentation).
19 _BLOCK_RE = re.compile(
20 r'\n[ \t]*\w+\.add_argument\(\n'
21 r'[ \t]+"--json", action="store_const", const="json", dest="fmt",\n'
22 r'[ \t]+help="Shorthand for --format json\."\n'
23 r'[ \t]+\)',
24 re.MULTILINE,
25 )
26
27
28 def revert_file(path: pathlib.Path) -> bool:
29 original = path.read_text(encoding="utf-8")
30 cleaned, count = _BLOCK_RE.subn("", original)
31 if count:
32 path.write_text(cleaned, encoding="utf-8")
33 print(f" reverted {count} block(s) in {path.relative_to(_COMMANDS_DIR.parent.parent.parent)}")
34 return bool(count)
35
36
37 def main() -> None:
38 total = 0
39 for f in sorted(_COMMANDS_DIR.rglob("*.py")):
40 if revert_file(f):
41 total += 1
42 print(f"\n{total} file(s) reverted.")
43
44
45 if __name__ == "__main__":
46 main()
File History 1 commit
sha256:d11a87833d5fad6059b7662844bf5448a8911a17cce7a51811f71ad394f248eb bump to v0.2.0rc13 Human patch 7 days ago