fix mapper

acm-uiuc · Oct 19, 2023 · 9ea3092 · 9ea3092
1 parent e6e7c03
commit 9ea3092
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/code/mapper.py b/code/mapper.py
@@ -41,15 +41,18 @@ def getPaidMembership(context, queryParams) -> dict:
                 'key': 'access_token'
             }
         )
-        aad_secret = response['Item']
+        aad_secret = response['Item']['value']
     except:
         # We don't have a cached token, got TTL'ed out.
         aad_secret = json.loads(client.get_secret_value(SecretId=AAD_SECRET_ID)['SecretString'])
         table.put_item(
             Item={
                 'key': {
-                    'M': aad_secret 
+                    'S': 'access_token' 
                 },
+                'value': {
+                    'M': aad_secret
+                }
                 'TimeToLive': {
                     'N': int(time.time()) + TOKEN_VALIDITY_SECONDS
                 }