-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathtrim.py
37 lines (29 loc) · 1.32 KB
/
trim.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
import json
import argparse
import sys
def strip_irrelevant_vacancy_info(raw_vacancy):
payment = raw_vacancy['payment'] or raw_vacancy['payment_from']
payment = payment or raw_vacancy['payment_to']
trimmed_vacancy = { 'profession': raw_vacancy['profession'],
'candidat': raw_vacancy['candidat'],
'payment': payment }
return trimmed_vacancy
def get_stripped_vacancy_list(raw_vacancies):
stripped_vacancy_list = []
for raw_vacancy in raw_vacancies:
stripped_vacancy_list.append(strip_irrelevant_vacancy_info(raw_vacancy))
return stripped_vacancy_list
def get_cli_arguments():
parser = argparse.ArgumentParser()
parser.add_argument('-i', '--infile', type=argparse.FileType('r'),
default=sys.stdin,
help='JSON file with vacancies from SuperJob, stdin by default')
parser.add_argument('-o', '--outfile', type=argparse.FileType('w'),
default=sys.stdout,
help='JSON file for trimmed vacancies, stdout by default')
return parser.parse_args()
if __name__ == '__main__':
args = get_cli_arguments()
raw_vacancies = json.load(args.infile)
stripped_vanacies = get_stripped_vacancy_list(raw_vacancies)
json.dump(stripped_vanacies, args.outfile)