From 988ed7326d57226128acb5f44915cb5731b180ef Mon Sep 17 00:00:00 2001
From: aviadl <aviad@descope.com>
Date: Fri, 5 Aug 2022 01:11:13 +0300
Subject: [PATCH] Fix sample app, to place data on cookie propely

---
 samples/otp_web_sample_app.py | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/samples/otp_web_sample_app.py b/samples/otp_web_sample_app.py
index 1cf10da0b..1442baeee 100644
--- a/samples/otp_web_sample_app.py
+++ b/samples/otp_web_sample_app.py
@@ -23,8 +23,8 @@
 descope_client = DescopeClient(PROJECT_ID, skip_verify=True)
 
 
-def set_cookie_on_response(response: Response, data):
-    cookie_domain = data.get("cookieDomain", "")
+def set_cookie_on_response(response: Response, cookieAttributes, data):
+    cookie_domain = cookieAttributes.get("cookieDomain", "")
     if cookie_domain == "":
         cookie_domain = None
 
@@ -32,11 +32,11 @@ def set_cookie_on_response(response: Response, data):
     expire_time = current_time + datetime.timedelta(days=30)
 
     return response.set_cookie(
-        key=data.get("cookieName", ""),
+        key=data.get("drn", ""),
         value=data.get("jwt", ""),
-        max_age=data.get("cookieMaxAge", int(expire_time.timestamp())),
-        expires=data.get("cookieExpiration", expire_time),
-        path=data.get("cookiePath", ""),
+        max_age=cookieAttributes.get("cookieMaxAge", int(expire_time.timestamp())),
+        expires=cookieAttributes.get("cookieExpiration", expire_time),
+        path=cookieAttributes.get("cookiePath", ""),
         domain=cookie_domain,
         secure=False,  # True
         httponly=True,
@@ -116,14 +116,14 @@ def verify():
         jwt_response = descope_client.otp.verify_code(DeliveryMethod.EMAIL, email, code)
     except AuthException:
         return Response("Unauthorized", 401)
-
     data = jwt_response["jwts"]
     response = Response(
         f"This is Verify code API handling, info example: {json.dumps(jwt_response)}",
         200,
     )
-    for _, cookieData in data.items():
-        set_cookie_on_response(response, cookieData)
+
+    set_cookie_on_response(response, data, data.get("DS"))
+    set_cookie_on_response(response, data, data.get("DSR"))
 
     return response
 
@@ -142,8 +142,10 @@ def private():
     response = Response(
         "This is a private API and you must be authenticated to see this", 200
     )
-    for _, cookieData in data.items():
-        set_cookie_on_response(response, cookieData)
+
+    set_cookie_on_response(response, data, data.get("DS"))
+    set_cookie_on_response(response, data, data.get("DSR"))
+
     return response