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
|
#!/usr/bin/env python3
import argparse
import os
import sys
VARIANTS = ["amazon", "arch", "centos", "debian", "fedora", "freebsd", "rhel",
"suse", "ubuntu", "unknown"]
if "avoid-pep8-E402-import-not-top-of-file":
_tdir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.insert(0, _tdir)
from cloudinit import templater
from cloudinit import util
from cloudinit.atomic_helper import write_file
def main():
parser = argparse.ArgumentParser()
platform = util.system_info()
parser.add_argument(
"--variant", default=platform['variant'], action="store",
help="define the variant.", choices=VARIANTS)
parser.add_argument(
"template", nargs="?", action="store",
default='./config/cloud.cfg.tmpl',
help="Path to the cloud.cfg template")
parser.add_argument(
"output", nargs="?", action="store", default="-",
help="Output file. Use '-' to write to stdout")
args = parser.parse_args()
with open(args.template, 'r') as fh:
contents = fh.read()
tpl_params = {'variant': args.variant}
contents = (templater.render_string(contents, tpl_params)).rstrip() + "\n"
util.load_yaml(contents)
if args.output == "-":
sys.stdout.write(contents)
else:
write_file(args.output, contents, omode="w")
if __name__ == '__main__':
main()
|