-
-
Notifications
You must be signed in to change notification settings - Fork 4.1k
/
set-more-info-link.py
145 lines (119 loc) · 4.31 KB
/
set-more-info-link.py
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
import argparse
import os
import re
import subprocess
import sys
labels = {
'en': 'More information:',
'bs': 'Više informacija:',
'da': 'Mere information:',
'de': 'Mehr Informationen:',
'es': 'Más información:',
'fa': 'اطلاعات بیشتر:',
'fr': 'Plus d\'informations\xa0:',
'sh': 'Više informacija:',
'hi': 'अधिक जानकारी:',
'id': 'Informasi lebih lanjut:',
'it': 'Maggiori informazioni:',
'ja': '詳しくはこちら:',
'ko': '더 많은 정보:',
'ml': 'കൂടുതൽ വിവരങ്ങൾ:',
'nl': 'Meer informatie:',
'no': 'Mer informasjon:',
'pl': 'Więcej informacji:',
'pt_BR': 'Mais informações:',
'pt_PT': 'Mais informações:',
'ru': 'Больше информации:',
'sv': 'Mer information:',
'ta': 'மேலும் தகவல்:',
'th': 'ดูเพิ่มเติม:',
'tr': 'Daha fazla bilgi için:',
'zh_TW': '更多資訊:',
'zh': '更多信息:',
}
IGNORE_FILES = (
'.DS_Store',
)
def get_tldr_root():
# if this script is running from tldr/scripts, the parent's parent is the root
f = os.path.normpath(__file__)
if f.endswith('tldr/scripts/set-more-info-link.py'):
return os.path.dirname(os.path.dirname(f))
if 'TLDR_ROOT' in os.environ:
return os.environ['TLDR_ROOT']
else:
print(
'\x1b[31mPlease set TLDR_ROOT to the location of a clone of https://github.com/tldr-pages/tldr.')
sys.exit(1)
def set_link(file, link):
with open(file) as f:
lines = f.readlines()
desc_start = 0
desc_end = 0
# find start and end of description
for i, line in enumerate(lines):
if line.startswith('>') and desc_start == 0:
desc_start = i
if not lines[i + 1].startswith('>') and desc_start != 0:
desc_end = i
break
# compute locale
pages_dir = os.path.basename(os.path.dirname(os.path.dirname(file)))
if '.' in pages_dir:
_, locale = pages_dir.split('.')
else:
locale = 'en'
# build new line
new_line = f'> {labels[locale]} <{link}>.\n'
if lines[desc_end] == new_line:
# return empty status to indicate that no changes were made
return ''
if re.search(r'^>.*<.+>', lines[desc_end]):
# overwrite last line
lines[desc_end] = new_line
status = '\x1b[34mlink updated'
else:
# add new line
lines.insert(desc_end + 1, new_line)
status = '\x1b[36mlink added'
with open(file, 'w') as f:
f.writelines(lines)
return status
def main():
parser = argparse.ArgumentParser(
description='Sets the "More information" link for all translations of a page')
parser.add_argument('-p', '--page', type=str, required=True,
help='page name in the format "platform/command.md"')
parser.add_argument('-s', '--stage', action='store_true', default=False,
help='stage modified pages (requires `git` to be on $PATH and TLDR_ROOT to be a Git repository)')
parser.add_argument('link', type=str)
args = parser.parse_args()
root = get_tldr_root()
pages_dirs = [d for d in os.listdir(root) if d.startswith('pages')]
target_paths = []
rel_paths = []
if not args.page.lower().endswith('.md'):
args.page = f'{args.page}.md'
for pages_dir in pages_dirs:
pages_dir_path = os.path.join(root, pages_dir)
platforms = [i for i in os.listdir(
pages_dir_path) if i not in IGNORE_FILES]
for platform in platforms:
platform_path = os.path.join(pages_dir_path, platform)
pages = os.listdir(platform_path)
commands = [
f'{platform}/{p}' for p in pages if p not in IGNORE_FILES]
if args.page in commands:
path = os.path.join(pages_dir_path, args.page)
target_paths.append(path)
target_paths.sort()
for path in target_paths:
rel_path = path.replace(f'{root}/', '')
rel_paths.append(rel_path)
status = set_link(path, args.link)
if status != '':
print(f'\x1b[32m{rel_path} {status}\x1b[0m')
if args.stage:
subprocess.call(['git', 'add', *rel_paths], cwd=root)
if __name__ == '__main__':
main()