From a7e03157b3d31ec1f1af976c01718e0a92acb994 Mon Sep 17 00:00:00 2001
From: conor <conor.mcmanus@gfk.com>
Date: Mon, 19 Aug 2019 15:39:47 +0200
Subject: [PATCH] Change default workspace to default instead of qa

---
 atmos.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/atmos.py b/atmos.py
index 67b3761..077d1be 100755
--- a/atmos.py
+++ b/atmos.py
@@ -57,7 +57,7 @@ def workspace_manager():
         branch = "default"
     else:
         if branch not in get_valid_envs():
-            branch = "qa"
+            branch = "default"
 
     if get_env() != branch:
         print("[INFO]: Terraform workspace & git branch have diverged. Changing workspace to git branch...")
@@ -100,20 +100,22 @@ def generate_creds(args):
 
 def get_valid_envs():
     try:
-        # Use var files when present, otherwise default to qa
+        # Use var files when present, otherwise default to default
         return [os.path.splitext(os.path.basename(x))[0] for x in glob.glob("vars/*.tfvars")]
     except FileNotFoundError:
         return False
 
 def get_env():
     try:
-        tf_env = open('.terraform/environment', 'r').read()
+        tf_env = ""
+        with open('.terraform/environment', 'r') as f:
+            tf_env = f.readline()
     except:
         return("default")
     if str(tf_env) in get_valid_envs():
         return(tf_env)
     else:
-        return("qa")
+        return("default")
 
 if __name__ == "__main__":
     main(sys.argv)
\ No newline at end of file