From 5ff6ae25ae60d953a0b0454907f7d2f251355959 Mon Sep 17 00:00:00 2001 From: 42B <42B@protonmail.com> Date: Fri, 26 Oct 2018 15:44:26 -0600 Subject: [PATCH] Use os.getenv's default values --- krampus.py | 25 +++++-------------------- 1 file changed, 5 insertions(+), 20 deletions(-) diff --git a/krampus.py b/krampus.py index 4997059..3600c63 100644 --- a/krampus.py +++ b/krampus.py @@ -52,26 +52,11 @@ def updateTaskList(self): def main(event, context): # collect our run information, otherwise just give it something I guess - if os.getenv("DEFAULT_REGION"): - region = os.getenv("DEFAULT_REGION") - else: - region = "us-east-1" - if os.getenv("KRAMPUS_BUCKET"): - krampus_bucket = os.getenv("KRAMPUS_BUCKET") - else: - krampus_bucket = "krampus-dev" - if os.getenv("TASKS_FILE_KEY"): - tasks_file_key = os.getenv("TASKS_FILE_KEY") - else: - tasks_file_key = "tasks.json" - if os.getenv("ARN_WHITELIST"): - whitelist = os.getenv("ARN_WHITELIST") - else: - whitelist = None - if os.getenv("KRAMPUS_ROLE_NAME"): - krampus_role = os.getenv("KRAMPUS_ROLE_NAME") - else: - krampus_role = "krampus" + region = os.getenv("DEFAULT_REGION", "us-east-1") + krampus_bucket = os.getenv("KRAMPUS_BUCKET", "krampus-dev") + tasks_file_key = os.getenv("TASKS_FILE_KEY", "tasks.json") + whitelist = os.getenv("ARN_WHITELIST") + krampus_role = os.getenv("KRAMPUS_ROLE_NAME", "krampus") # fire it all up k = Krampus(region, krampus_bucket, tasks_file_key, whitelist, krampus_role) k.getTasks()