|
|
|
@ -11,13 +11,16 @@ class Terraform:
|
|
|
|
|
self.targets = [] if targets is None else targets
|
|
|
|
|
self.variables = dict() if variables is None else variables
|
|
|
|
|
|
|
|
|
|
self.state = state
|
|
|
|
|
self.state_data = None
|
|
|
|
|
self.state_filename = state
|
|
|
|
|
self.state_data = dict()
|
|
|
|
|
self.parallelism = 50
|
|
|
|
|
|
|
|
|
|
def apply(self, targets=None, variables=None):
|
|
|
|
|
def apply(self, targets=None, variables=None, **kargs):
|
|
|
|
|
"""
|
|
|
|
|
refer to https://terraform.io/docs/commands/apply.html
|
|
|
|
|
:param variables: variables in dict type
|
|
|
|
|
:param targets: targets in list
|
|
|
|
|
:returns return_code, stdout, stderr
|
|
|
|
|
"""
|
|
|
|
|
variables = self.variables if variables is None else variables
|
|
|
|
|
targets = self.targets if targets is None else targets
|
|
|
|
@ -25,13 +28,20 @@ class Terraform:
|
|
|
|
|
parameters = []
|
|
|
|
|
parameters += self._generate_targets(targets)
|
|
|
|
|
parameters += self._generate_var_string(variables)
|
|
|
|
|
parameters += ['-parallelism=%s' % self.parallelism]
|
|
|
|
|
parameters += self._gen_param_string(kargs)
|
|
|
|
|
|
|
|
|
|
parameters = \
|
|
|
|
|
['terraform', 'apply', '-state=%s' % self.state] + parameters
|
|
|
|
|
['terraform', 'apply', '-state=%s' % self.state_filename] + parameters
|
|
|
|
|
|
|
|
|
|
cmd = ' '.join(parameters)
|
|
|
|
|
return self._run_cmd(cmd)
|
|
|
|
|
|
|
|
|
|
def _gen_param_string(self, kargs):
|
|
|
|
|
params = []
|
|
|
|
|
for key, value in kargs.items():
|
|
|
|
|
params += ['%s=%s' % (key, value)]
|
|
|
|
|
return params
|
|
|
|
|
|
|
|
|
|
def _run_cmd(self, cmd):
|
|
|
|
|
log.debug('command: ' + cmd)
|
|
|
|
|
|
|
|
|
@ -43,10 +53,10 @@ class Terraform:
|
|
|
|
|
|
|
|
|
|
if ret_code == 0:
|
|
|
|
|
log.debug('error: ' + err)
|
|
|
|
|
self.read_state()
|
|
|
|
|
self.read_state_file()
|
|
|
|
|
return ret_code, out, err
|
|
|
|
|
|
|
|
|
|
def destroy(self, targets=None, variables=None):
|
|
|
|
|
def destroy(self, targets=None, variables=None, **kwargs):
|
|
|
|
|
variables = self.variables if variables is None else variables
|
|
|
|
|
targets = self.targets if targets is None else targets
|
|
|
|
|
|
|
|
|
@ -55,7 +65,7 @@ class Terraform:
|
|
|
|
|
parameters += self._generate_var_string(variables)
|
|
|
|
|
|
|
|
|
|
parameters = \
|
|
|
|
|
['terraform', 'destroy', '-force', '-state=%s' % self.state] + \
|
|
|
|
|
['terraform', 'destroy', '-force', '-state=%s' % self.state_filename] + \
|
|
|
|
|
parameters
|
|
|
|
|
cmd = ' '.join(parameters)
|
|
|
|
|
return self._run_cmd(cmd)
|
|
|
|
@ -68,15 +78,18 @@ class Terraform:
|
|
|
|
|
parameters += self._generate_targets(targets)
|
|
|
|
|
parameters += self._generate_var_string(variables)
|
|
|
|
|
parameters = \
|
|
|
|
|
['terraform', 'refresh', '-state=%s' % self.state] + \
|
|
|
|
|
['terraform', 'refresh', '-state=%s' % self.state_filename] + \
|
|
|
|
|
parameters
|
|
|
|
|
cmd = ' '.join(parameters)
|
|
|
|
|
return self._run_cmd(cmd)
|
|
|
|
|
|
|
|
|
|
def read_state(self, state=None):
|
|
|
|
|
state = self.state if state is None else state
|
|
|
|
|
if os.path.exists(state):
|
|
|
|
|
with open(state) as f:
|
|
|
|
|
def read_state_file(self):
|
|
|
|
|
"""
|
|
|
|
|
read .tfstate file
|
|
|
|
|
:return: states file in dict type
|
|
|
|
|
"""
|
|
|
|
|
if os.path.exists(self.state_filename):
|
|
|
|
|
with open(self.state_filename) as f:
|
|
|
|
|
json_data = json.load(f)
|
|
|
|
|
self.state_data = json_data
|
|
|
|
|
log.debug("state_data=%s" % str(self.state_data))
|
|
|
|
@ -86,11 +99,11 @@ class Terraform:
|
|
|
|
|
|
|
|
|
|
def is_any_aws_instance_alive(self):
|
|
|
|
|
self.refresh()
|
|
|
|
|
if not os.path.exists(self.state):
|
|
|
|
|
if not os.path.exists(self.state_filename):
|
|
|
|
|
log.debug("can't find %s " % self.state_data)
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
self.read_state()
|
|
|
|
|
self.read_state_file()
|
|
|
|
|
try:
|
|
|
|
|
main_module = self._get_main_module()
|
|
|
|
|
for resource_key, info in main_module['resources'].items():
|
|
|
|
@ -136,6 +149,11 @@ class Terraform:
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
def get_output_value(self, output_name):
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
:param output_name:
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
try:
|
|
|
|
|
main_module = self._get_main_module()
|
|
|
|
|
return main_module['outputs'][output_name]
|