From 9ba81d3f3bd82c34b8c0fe1842ecba6e11a34dbe Mon Sep 17 00:00:00 2001
From: Christoph Goessmann <44035999+gochristoph@users.noreply.github.com>
Date: Tue, 9 Jan 2024 02:10:59 +0100
Subject: [PATCH] Remove unused AAD token to support other identity providers
 (#459)

---
 backend/auth/auth_utils.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/backend/auth/auth_utils.py b/backend/auth/auth_utils.py
index d84aaa60fd..59dd02ea3b 100644
--- a/backend/auth/auth_utils.py
+++ b/backend/auth/auth_utils.py
@@ -10,11 +10,11 @@ def get_authenticated_user_details(request_headers):
         ## if it is, get the user details from the EasyAuth headers
         raw_user_object = {k:v for k,v in request_headers.items()}
 
-    user_object['user_principal_id'] = raw_user_object['X-Ms-Client-Principal-Id']
-    user_object['user_name'] = raw_user_object['X-Ms-Client-Principal-Name']
-    user_object['auth_provider'] = raw_user_object['X-Ms-Client-Principal-Idp']
-    user_object['auth_token'] = raw_user_object['X-Ms-Token-Aad-Id-Token']
-    user_object['client_principal_b64'] = raw_user_object['X-Ms-Client-Principal']
-    user_object['aad_id_token'] = raw_user_object["X-Ms-Token-Aad-Id-Token"]
+    user_object['user_principal_id'] = raw_user_object.get('X-Ms-Client-Principal-Id')
+    user_object['user_name'] = raw_user_object.get('X-Ms-Client-Principal-Name')
+    user_object['auth_provider'] = raw_user_object.get('X-Ms-Client-Principal-Idp')
+    user_object['auth_token'] = raw_user_object.get('X-Ms-Token-Aad-Id-Token')
+    user_object['client_principal_b64'] = raw_user_object.get('X-Ms-Client-Principal')
+    user_object['aad_id_token'] = raw_user_object.get('X-Ms-Token-Aad-Id-Token')
 
     return user_object
\ No newline at end of file