-
Notifications
You must be signed in to change notification settings - Fork 0
/
zscaler-config-builder.py
74 lines (62 loc) · 2.32 KB
/
zscaler-config-builder.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
#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import os.path
import csv
import argparse
import traceback
from jinja2 import Template
def main():
parser = argparse.ArgumentParser()
parser.prog = 'zscaler-config.bulder.py'
parser.description = "Script builds configuraiton templates for Zscaler services."
parser.epilog = "Please visit http://github.com/zscaler/zscaler-config-builder/ for updates."
parser.add_argument('-C',
dest='confpath',
required=False,
default=' ',
help='Configuration Path')
parser.add_argument('-I',
dest='input',
required=True,
default='template.csv',
help='CSV file with input variables')
parser.add_argument('-T',
dest='template',
required=True,
default='template.j2',
help='Jinja template file')
parser.add_argument('-V', '--version',
action='version',
version='%(prog)s 1.0')
if len(sys.argv)==1:
parser.print_help()
sys.exit(1)
else:
args = parser.parse_args()
if not os.path.exists(args.input):
parser.error("The file %s does not exist! Please recheck the filename specfied" % args.input)
if not os.path.exists(args.template):
parser.error("The file %s does not exist! Please recheck the filename specfied" % args.template)
cvs_variables = csv.DictReader(open(args.input))
#conf_path = args.confpath
conf_path = ""
for row in cvs_variables:
vars = row
configfile = "{}{}{}".format(conf_path, row["hostname"], ".conf")
with open(args.template) as template_fh:
template_format = template_fh.read()
template = Template(template_format)
fout = open(configfile, 'w')
print fout
fout.write((template.render(vars)))
fout.close()
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
print "Ending...\n\n"
except Exception:
traceback.print_exc(file=sys.stdout)
print "Ending..."
sys.exit(0)