1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
# SPDX-License-Identifier: NONE
# 2020 by David Lamparter, placed in the public domain.
import sys
import os
import re
import argparse
wrap_res = [
(re.compile(r'(?<!\\n)"\s*\n\s*"', re.M), r""),
]
pri_res = [
(re.compile(r'(PRI[udx][0-9]+)\s*\n\s*"', re.M), r'\1"'),
(re.compile(r'"\s*PRI([udx])32\s*"'), r"\1"),
(re.compile(r'"\s*PRI([udx])32'), r'\1"'),
(re.compile(r'"\s*PRI([udx])16\s*"'), r"h\1"),
(re.compile(r'"\s*PRI([udx])16'), r'h\1"'),
(re.compile(r'"\s*PRI([udx])8\s*"'), r"hh\1"),
(re.compile(r'"\s*PRI([udx])8'), r'hh\1"'),
]
def main():
argp = argparse.ArgumentParser(description="C string mangler")
argp.add_argument("--unwrap", action="store_const", const=True)
argp.add_argument("--pri8-16-32", action="store_const", const=True)
argp.add_argument("files", type=str, nargs="+")
args = argp.parse_args()
regexes = []
if args.unwrap:
regexes.extend(wrap_res)
if args.pri8_16_32:
regexes.extend(pri_res)
if len(regexes) == 0:
sys.stderr.write("no action selected to execute\n")
sys.exit(1)
l = 0
for fn in args.files:
sys.stderr.write(fn + "\033[K\r")
with open(fn, "r") as ifd:
data = ifd.read()
newdata = data
n = 0
for regex, repl in regexes:
newdata, m = regex.subn(repl, newdata)
n += m
if n > 0:
sys.stderr.write("changed: %s\n" % fn)
with open(fn + ".new", "w") as ofd:
ofd.write(newdata)
os.rename(fn + ".new", fn)
l += 1
sys.stderr.write("%d files changed.\n" % (l))
main()
|
