blob: 50da73fd6346720b9e1c5360471bd8eb04dfa33c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from jinja2 import Template
import json
import os
import re
import sys
imposter_template = sys.argv[1]
responses_dir = sys.argv[2]
imposter_json = sys.argv[3]
with open(imposter_template, 'r') as imposter_file:
tm = Template(imposter_file.read())
responses = {}
for filename in os.listdir(responses_dir):
if filename.endswith('.json'):
with open(os.path.join(responses_dir, filename), 'r') as response_file:
responses[re.sub('\.json$', '', filename)] = response_file.read()
imposter = tm.render(responses)
with open(imposter_json, 'w') as imposter_file:
imposter_file.write(imposter)
|