diff --git a/atmos.py b/atmos.py index 849e910..a886806 100755 --- a/atmos.py +++ b/atmos.py @@ -1,29 +1,51 @@ #!/usr/bin/env python3 import argparse, subprocess, shlex, sys, os, glob +from jinja2 import Environment, FileSystemLoader def main(argv): parser = argparse.ArgumentParser(description='Control Terraform Workspaces.') g = parser.add_mutually_exclusive_group() 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() if args.command: determine_actions(args, params) def determine_actions(args, params): + workspace = get_env() env_actions = ["plan", "apply", "destroy"] # Commands that require env context cmd = 'terraform {args}'.format(args=args.command) for param in params: # Pass terraform params directly through cmd = cmd + ' ' + param - if (args.command in env_actions) and (get_env() != "master"): # Append with env context - cmd = cmd + ' -var-file=vars/{env}.tfvars -var "workspace={env}"'.format(env=get_env()) + 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=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: 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(): try: # Use var files when present, otherwise default to qa @@ -35,7 +57,7 @@ def get_env(): try: tf_env = open('.terraform/environment', 'r').read() except: - return("master") + return("default") if str(tf_env) in get_valid_envs(): return(tf_env) else: