-
Notifications
You must be signed in to change notification settings - Fork 0
/
Name_replacer.py
45 lines (26 loc) · 1.09 KB
/
Name_replacer.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
import sys
import re
def get_dict_of_names(file_of_names):
'''
:param file_of_names: The file name where field1 is the term to search and field2 the term to replace to
:return: a dictionary
'''
with open(file_of_names, 'r') as file_obj:
return {line[0]: line[1] for line in (line.strip().split("|") for line in file_obj if line.strip())}
def replace_occurrences(dict_of_names, file_name):
with open(file_name, 'r') as f:
for line in f:
for name in dict_of_names.keys():
if name in line:
line = line.replace(name, dict_of_names[name])
print(line.strip())
def main():
if (len(sys.argv) != 3) or sys.argv[1] == "-h":
print('Usage\npython ' + sys.argv[0] + "<alternative_file_of_names> <publication_file_names>\n\n ")
sys.exit()
alternative_file_of_names = sys.argv[1]
publication_file_names = sys.argv[2]
dict_of_names = get_dict_of_names(alternative_file_of_names)
replace_occurrences(dict_of_names, publication_file_names)
if __name__ == '__main__':
main()