From 9ea3092bf56a99e9e9f8650143ca5494d012a966 Mon Sep 17 00:00:00 2001
From: Dev Singh <dev@devksingh.com>
Date: Thu, 19 Oct 2023 11:32:55 -0500
Subject: [PATCH] fix mapper

---
 code/mapper.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/code/mapper.py b/code/mapper.py
index db12194..dcd99fe 100644
--- 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
                 }