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
|
import sys
import re
import argparse
path_prefix = ""
def repl_normal(m):
if m.group('name').lower() == 'section':
return f"<a href=\"{path_prefix}{m.group('section')}\">{m.group('name')} ({m.group('section')})</a>"
else:
if m.group('section') not in ['lib', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10']:
pass
else:
return f"<a href=\"{path_prefix}{m.group('section')}/{m.group('name')}.html\">{m.group('name')} ({m.group('section')})</a>"
return f"{m.group('name')} ({m.group('section')})"
if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog='mklinks',
description='Makes links obviously',
epilog='God bless you')
parser.add_argument('-i', '--input')
parser.add_argument('-o', '--output')
parser.add_argument('-p', '--path-prefix')
args = parser.parse_args()
path_prefix = ""
if args.path_prefix is not None:
path_prefix = args.path_prefix
link_re = re.compile(r"(?P<name>[a-zA-Z0-9\-\_]+) \((?P<section>[0-9a-z]+)\)")
link_i_re = re.compile(r"<i>(?P<name>[a-zA-Z0-9\-\_]+)</i>\((?P<section>[0-9a-z]+)\)")
f = None
if args.input is None:
f = sys.stdin
else:
f = open(args.input, 'r')
text = f.read()
f.close()
text = link_re.sub(repl_normal, text)
text = link_i_re.sub(repl_normal, text)
if args.output is not None:
with open(args.output, 'w') as out:
out.write(text)
else:
print(text)
|
