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

Add verbose mode and flag for project prefix in env vars

Browse source
This commit is contained in:
conor 2019-07-17 10:15:21 +02:00
parent 8bff7443d5
commit b6d8d623df
1 changed files with 29 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
atmos.py
View file

@ -9,6 +9,8 @@ def main(argv):
parser.add_argument("-e", help="Gather shared-creds from environment variables (Dont use this flag if you dont want your ~/.aws/credentials replaced. This is for CI/CD", action='store_true', default=False) parser.add_argument("-e", help="Gather shared-creds from environment variables (Dont use this flag if you dont want your ~/.aws/credentials replaced. This is for CI/CD", action='store_true', default=False)
parser.add_argument("-m", help="Prevents workspace from changing with git branches automatically", action='store_true', default=False) parser.add_argument("-m", help="Prevents workspace from changing with git branches automatically", action='store_true', default=False)
parser.add_argument("-n", help="Atmos will not add -var-file or -var args to terraform", action='store_true', default=False) parser.add_argument("-n", help="Atmos will not add -var-file or -var args to terraform", action='store_true', default=False)
parser.add_argument("-p", "--project", help="Add a project prefix for env vars", nargs='?', default="")
parser.add_argument("-v", "--verbose", help="Debug mode", 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)
@ -16,22 +18,30 @@ def main(argv):
def determine_actions(args, params): def determine_actions(args, params):
aws_creds_file = "$HOME/.aws/credentials" aws_creds_file = "$HOME/.aws/credentials"
if (is_git_directory()) and not (args.m): if (is_git_directory()) and not (args.m):
aws_creds_file = aws_creds_file + "-atmos" if (args.e):
aws_creds_file = aws_creds_file + "-atmos"
workspace_manager() workspace_manager()
if (args.project) and (args.verbose):
print("Project: " + args.project)
workspace = get_env() 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)
if (args.command in env_actions) and not (args.n): # Append with env context if (args.command in env_actions) and not (args.n): # Append with env context
cmd = cmd + ' -var-file=vars/{env}.tfvars -var "workspace={env}" -var "shared_credentials_file={aws_creds_file}"'.format(env=workspace, aws_creds_file=aws_creds_file) cmd = cmd + ' -var-file=vars/{env}.tfvars -var "workspace={env}"'.format(env=workspace)
cmd = cmd + ' -var "shared_credentials_file={aws_creds_file}"'.format(aws_creds_file=aws_creds_file)
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.e): if (args.e):
generate_creds() generate_creds(args)
if (args.verbose):
print("Atmos will run: " + cmd)
print('Terraform {args} using env vars in {env}'.format(args=args.command, env=workspace)) 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
@ -52,18 +62,29 @@ def workspace_manager():
subprocess.call(["terraform", "workspace", "new", branch], stderr=subprocess.STDOUT, stdout=open(os.devnull, 'w')) subprocess.call(["terraform", "workspace", "new", branch], stderr=subprocess.STDOUT, stdout=open(os.devnull, 'w'))
subprocess.call(["terraform", "workspace", "select", branch], stderr=subprocess.STDOUT, stdout=open(os.devnull, 'w')) subprocess.call(["terraform", "workspace", "select", branch], stderr=subprocess.STDOUT, stdout=open(os.devnull, 'w'))
def generate_creds(): def generate_creds(args):
current_workspace = get_env() current_workspace = get_env()
workspaces = ['default'] workspaces = ['default']
if current_workspace != 'default': if current_workspace != 'default':
workspaces.append(current_workspace) workspaces.append(current_workspace)
project_name = ""
if (args.project):
project_name = args.project.upper() + "_"
contents = "" contents = ""
for workspace in workspaces: for workspace in workspaces:
access_key_name = project_name + workspace.upper() + '_ACCESS_KEY_ID'
secret_key_name = project_name + workspace.upper() + '_SECRET_ACCESS_KEY'
if (args.verbose):
print(access_key_name)
print(secret_key_name)
contents = contents + "[{workspace}]\n".format(workspace=workspace) contents = contents + "[{workspace}]\n".format(workspace=workspace)
contents = contents + "aws_access_key_id=" + os.environ.get(workspace.upper() + '_ACCESS_KEY_ID') + "\n" contents = contents + "aws_access_key_id=" + os.environ.get(access_key_name) + "\n"
contents = contents + "aws_secret_access_key=" + os.environ.get(workspace.upper() + '_SECRET_ACCESS_KEY') + "\n" contents = contents + "aws_secret_access_key=" + os.environ.get(secret_key_name) + "\n"
with open(os.path.expanduser('~/.aws/credentials-atmos'), 'w+') as f: with open(os.path.expanduser('~/.aws/credentials-atmos'), 'w+') as f:
f.write(contents) f.write(contents)