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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# Copyright 2017 Lars Wirzenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# =*= License: GPL-3+ =*=
import logging
import sys
import cliapp
import vmdb
class Vmdb2(cliapp.Application):
def add_settings(self):
self.settings.string(
['image'],
'create image file FILE',
metavar='FILE')
self.settings.boolean(
['verbose', 'v'],
'verbose output')
def setup(self):
self.step_runners = vmdb.StepRunnerList()
def process_args(self, args):
if len(args) != 1:
sys.exit("No image specification was given on the command line.")
vmdb.set_verbose_progress(self.settings['verbose'])
spec = self.load_spec_file(args[0])
state = vmdb.State()
state.tags = vmdb.Tags()
params = self.create_template_vars(state)
steps = spec.get_steps(params)
# Check that we have step runners for each step
for step in steps:
self.step_runners.find(step)
steps_taken, core_meltdown = self.run_steps(steps, state)
if core_meltdown:
vmdb.progress('Something went wrong, cleaning up!')
else:
vmdb.progress('All went fine, cleaning up.')
self.run_teardowns(steps_taken, state)
if core_meltdown:
logging.error('An error occurred, exiting with non-zero exit code')
sys.exit(1)
def load_spec_file(self, filename):
vmdb.progress('Load spec file {}'.format(filename))
spec = vmdb.Spec()
spec.load_file(filename)
return spec
def run_steps(self, steps, state):
return self.run_steps_helper(
steps, state, 'Running step: %r', 'run', False)
def run_teardowns(self, steps, state):
return self.run_steps_helper(
list(reversed(steps)), state, 'Running teardown: %r', 'teardown', True)
def run_steps_helper(self, steps, state, msg, method_name, keep_going):
core_meltdown = False
steps_taken = []
even_if_skipped = method_name + '_even_if_skipped'
for step in steps:
try:
logging.info(msg, step)
steps_taken.append(step)
runner = self.step_runners.find(step)
if runner.skip(step, self.settings, state):
logging.info('Skipping as requested by unless')
method_names = [even_if_skipped]
else:
method_names = [method_name, even_if_skipped]
methods = [
getattr(runner, name)
for name in method_names
if hasattr(runner, name)
]
for method in methods:
logging.info('Calling {}'.format(method))
method(step, self.settings, state)
except BaseException as e:
vmdb.error(str(e))
core_meltdown = True
if not keep_going:
break
return steps_taken, core_meltdown
def create_template_vars(self, state):
vars = dict()
for key in self.settings:
vars[key] = self.settings[key]
vars.update(state.as_dict())
return vars
|