Simplify pulling credential parts from AWS CLI JSON response body

magnetikonline · Jan 25, 2023 · 5b0a176 · 5b0a176
1 parent 1f9969b
commit 5b0a176
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/main.py b/main.py
@@ -161,10 +161,7 @@ def aws_sts_assume_role(
 
     # pull out generated session credentials
     def credential_part(key: str) -> str:
-        if "Credentials" not in assume_data:
-            return ""
-
-        return assume_data["Credentials"].get(key, "")
+        return assume_data.get("Credentials", {}).get(key, "")
 
     access_key_id = credential_part("AccessKeyId")
     secret_access_key = credential_part("SecretAccessKey")