Vereto/atmos
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add -t flag to use environment vars to generate credetials file.

Browse source
This commit is contained in:
Conor.McManus 2019-04-15 15:15:39 +02:00
parent e528db59d0
commit 34b7449141
1 changed files with 26 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
atmos.py
View file

@ -1,29 +1,51 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse, subprocess, shlex, sys, os, glob import argparse, subprocess, shlex, sys, os, glob
from jinja2 import Environment, FileSystemLoader
def main(argv): def main(argv):
parser = argparse.ArgumentParser(description='Control Terraform Workspaces.') parser = argparse.ArgumentParser(description='Control Terraform Workspaces.')
g = parser.add_mutually_exclusive_group() g = parser.add_mutually_exclusive_group()
g.add_argument("command", help="Send commands to terraform with workspace variable context", nargs='?', default=False) g.add_argument("command", help="Send commands to terraform with workspace variable context", nargs='?', default=False)
parser.add_argument("-t", help="Template mode, gather shared-creds from environment variables", action='store_true', default=False)
args, params = parser.parse_known_args() args, params = parser.parse_known_args()
if args.command: if args.command:
determine_actions(args, params) determine_actions(args, params)
def determine_actions(args, params): def determine_actions(args, params):
workspace = get_env()
env_actions = ["plan", "apply", "destroy"] # Commands that require env context env_actions = ["plan", "apply", "destroy"] # Commands that require env context
cmd = 'terraform {args}'.format(args=args.command) cmd = 'terraform {args}'.format(args=args.command)
for param in params: # Pass terraform params directly through for param in params: # Pass terraform params directly through
cmd = cmd + ' ' + param cmd = cmd + ' ' + param
if (args.command in env_actions) and (get_env() != "master"): # Append with env context if (args.command in env_actions) and (workspace != "default"): # Append with env context
cmd = cmd + ' -var-file=vars/{env}.tfvars -var "workspace={env}"'.format(env=get_env()) cmd = cmd + ' -var-file=vars/{env}.tfvars -var "workspace={env}"'.format(env=workspace)
print('Terraform {args} using env vars in {env}'.format(args=args.command, env=get_env())) if (args.t):
generate_creds()
print('Terraform {args} using env vars in {env}'.format(args=args.command, env=workspace))
with subprocess.Popen(shlex.split(cmd)) as proc: with subprocess.Popen(shlex.split(cmd)) as proc:
exit # Start process but kill py program exit # Start process but kill py program
def generate_creds():
current_workspace = get_env()
workspaces = ['default']
if current_workspace != 'default':
workspaces.append(current_workspace)
contents = ""
for workspace in workspaces:
contents = contents + "[{workspace}]\n".format(workspace=workspace)
contents = contents + "access_key_id=" + os.environ.get(workspace.upper() + '_ACCESS_KEY_ID') + "\n"
contents = contents + "secret_access_key=" + os.environ.get(workspace.upper() + '_SECRET_ACCESS_KEY') + "\n"
print(contents)
with open(os.path.expanduser('~/.aws/credentials'), 'w+') as f:
f.write(contents)
def get_valid_envs(): def get_valid_envs():
try: try:
# Use var files when present, otherwise default to qa # Use var files when present, otherwise default to qa
@ -35,7 +57,7 @@ def get_env():
try: try:
tf_env = open('.terraform/environment', 'r').read() tf_env = open('.terraform/environment', 'r').read()
except: except:
return("master") return("default")
if str(tf_env) in get_valid_envs(): if str(tf_env) in get_valid_envs():
return(tf_env) return(tf_env)
else: else: