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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#!/usr/bin/python
from cloudinit import handlers
from cloudinit.handlers import cloud_config as cc_part
from cloudinit import helpers
from cloudinit import log as logging
from cloudinit.settings import PER_INSTANCE
from cloudinit import user_data as ud
import argparse
import os
import shutil
import tempfile
def main():
parser = argparse.ArgumentParser(
description='test cloud-config merging')
parser.add_argument("--output", "-o", metavar="file",
help="specify output file", default="-")
parser.add_argument('--verbose', '-v', action='count', default=0)
parser.add_argument('files', nargs='+')
args = parser.parse_args()
if args.verbose:
level = (logging.WARN, logging.INFO,
logging.DEBUG)[min(args.verbose, 2)]
logging.setupBasicLogging(level)
outfile = args.output
if args.output == "-":
outfile = "/dev/stdout"
tempd = tempfile.mkdtemp()
handler_dir = os.path.join(tempd, "hdir")
data = None # the 'init' object
frequency = PER_INSTANCE
paths = helpers.Paths({})
# make a '#include <f1>' style
udproc = ud.UserDataProcessor(paths=paths)
user_data_msg = udproc.process("#include\n" +
'\n'.join([os.path.abspath(f) for f in args.files]))
ccph = cc_part.CloudConfigPartHandler(paths=paths)
ccph.cloud_fn = outfile
c_handlers = helpers.ContentHandlers()
c_handlers.register(ccph)
called = []
for (_ctype, mod) in c_handlers.items():
if mod in called:
continue
handlers.call_begin(mod, data, frequency)
called.append(mod)
# Walk the user data
part_data = {
'handlers': c_handlers,
# Any new handlers that are encountered get writen here
'handlerdir': handler_dir,
'data': data,
# The default frequency if handlers don't have one
'frequency': frequency,
# This will be used when new handlers are found
# to help write there contents to files with numbered
# names...
'handlercount': 0,
'excluded': [],
}
handlers.walk(user_data_msg, handlers.walker_callback, data=part_data)
# Give callbacks opportunity to finalize
called = []
for (_ctype, mod) in c_handlers.items():
if mod in called:
continue
handlers.call_end(mod, data, frequency)
called.append(mod)
shutil.rmtree(tempd)
if __name__ == "__main__":
main()
# vi: ts=4 expandtab
|
