From 3f3463b47bcfe97dbadfc757ac8f82529e1c0b4f Mon Sep 17 00:00:00 2001
From: abhigyanghosh30 <abhigyanghosh30@gmail.com>
Date: Thu, 7 Mar 2024 04:03:03 +0530
Subject: [PATCH] fix: your exams and scheduling

---
 templates/credentials/schedule.html |   2 +-
 webapp/shop/api/ua_contracts/api.py |   5 +-
 webapp/shop/cred/views.py           | 132 ++++++++++++++++------------
 3 files changed, 82 insertions(+), 57 deletions(-)

diff --git a/templates/credentials/schedule.html b/templates/credentials/schedule.html
index 5c164ede1d4..073ccc50fcb 100644
--- a/templates/credentials/schedule.html
+++ b/templates/credentials/schedule.html
@@ -52,7 +52,7 @@ <h1 class="p-heading--2"><strong>Schedule your exam</strong></h1>
 </section>
 
 <script>
-  {% if timezone %}
+  {% if timezone and timezone!="" %}
   const timezone = "{{ timezone }}";
   {% else %}
   const timezone = Intl.DateTimeFormat().resolvedOptions().timeZone;
diff --git a/webapp/shop/api/ua_contracts/api.py b/webapp/shop/api/ua_contracts/api.py
index e7dd9e6bccf..03d1986c96b 100644
--- a/webapp/shop/api/ua_contracts/api.py
+++ b/webapp/shop/api/ua_contracts/api.py
@@ -361,7 +361,7 @@ def post_assessment_reservation(
         starts_at,
         country_code,
     ) -> dict:
-        return self._request(
+        req = self._request(
             method="get",
             path="v1/cue/schedule",
             json={
@@ -374,7 +374,8 @@ def post_assessment_reservation(
             },
             error_rules=["default"],
         ).json()
-
+        return req
+    
     def delete_assessment_reservation(self, contract_item_id) -> dict:
         self._request(
             method="delete",
diff --git a/webapp/shop/cred/views.py b/webapp/shop/cred/views.py
index 68bf17e3e0b..c60a433ccf9 100644
--- a/webapp/shop/cred/views.py
+++ b/webapp/shop/cred/views.py
@@ -43,6 +43,16 @@
     "scheduled": "Scheduled",
     "processed": "Complete",
     "canceled": "Cancelled",
+    "finalized": "Complete",
+    "provisioning": "In Progress",
+    "provisioned": "In Progress",
+    "in_progress": "In Progress",
+    "completed": "Complete",
+    "grading": "Complete",
+    "graded": "Complete",
+    "archiving": "Complete",
+    "archived": "Complete",
+    "canceling": "Cancelled"
 }
 
 
@@ -125,36 +135,34 @@ def cred_schedule(ua_contracts_api, trueability_api, **_):
         contract_item_id = data["contractItemID"]
         first_name, last_name = get_user_first_last_name()
         country_code = TIMEZONE_COUNTRIES[timezone]
-        try:
-            response = ua_contracts_api.post_assessment_reservation(
-                contract_item_id,
-                first_name,
-                last_name,
-                timezone,
-                starts_at.isoformat(),
-                country_code,
-            )
-
-            if response and "reservation" not in response:
-                error = response["message"]
-                return flask.render_template(
-                    "/credentials/schedule.html", error=error
-                )
-            else:
-                uuid = response.get("reservation", {}).get("IDs", [])[-1]
-                exam = {
-                    "name": "CUE: Linux",
-                    "date": starts_at.strftime("%d %b %Y"),
-                    "time": starts_at.strftime("%I:%M %p %Z"),
-                    "uuid": uuid,
-                    "contract_item_id": contract_item_id,
-                }
-                return flask.render_template(
-                    "/credentials/schedule-confirm.html", exam=exam
-                )
-        except Exception as e:
-            print("Exception:", e)
+        
+        response = ua_contracts_api.post_assessment_reservation(
+            contract_item_id,
+            first_name,
+            last_name,
+            timezone,
+            starts_at.isoformat(),
+            country_code,
+        )
 
+        if response and "reservation" not in response:
+            error = response["message"]
+            return flask.render_template(
+                "/credentials/schedule.html", error=error
+            )
+        else:
+            uuid = response.get("reservation", {}).get("IDs", [])[-1]
+            exam = {
+                "name": "CUE: Linux",
+                "date": starts_at.strftime("%d %b %Y"),
+                "time": starts_at.strftime("%I:%M %p ")+timezone,
+                "uuid": uuid,
+                "contract_item_id": contract_item_id,
+            }
+            return flask.render_template(
+                "/credentials/schedule-confirm.html", exam=exam
+            )
+        
     contract_item_id = flask.request.args.get("contractItemID")
     if contract_item_id is None:
         return flask.redirect("/credentials/your-exams")
@@ -242,7 +250,10 @@ def cred_your_exams(ua_contracts_api, trueability_api, **kwargs):
                 utc = pytz.timezone("UTC")
                 now = utc.localize(datetime.utcnow())
                 end = starts_at + timedelta(minutes=75)
-                state = RESERVATION_STATES.get(r["state"], r["state"])
+                if "assessment" in r and r["assessment"] is not None:
+                    state = RESERVATION_STATES.get(r["assessment"]["state"], r["state"])
+                else:
+                    state = RESERVATION_STATES.get(r["state"], r["state"])
 
                 if assessment_id and now > starts_at and now < end:
                     actions.extend(
@@ -267,7 +278,33 @@ def cred_your_exams(ua_contracts_api, trueability_api, **kwargs):
                             "actions": actions,
                         }
                     )
+                elif (
+                    assessment_id
+                    and now < starts_at
+                    and now > starts_at - timedelta(minutes=30)
+                ):
+                    actions.extend(
+                        [
+                            {
+                                "text": "Take exam",
+                                "href": "/credentials/exam?"
+                                f"id={ assessment_id }",
+                                "button_class": "p-button",
+                            }
+                        ]
+                    )
 
+                    exams_in_progress.append(
+                        {
+                            "name": name,
+                            "date": starts_at.strftime("%d %b %Y"),
+                            "time": starts_at.strftime("%I:%M %p %Z"),
+                            "timezone": timezone,
+                            "state": "In progress",
+                            "uuid": r["uuid"],
+                            "actions": actions,
+                        }
+                    )
                 elif state == "Scheduled" and starts_at > now + timedelta(
                     minutes=30
                 ):
@@ -295,30 +332,17 @@ def cred_your_exams(ua_contracts_api, trueability_api, **kwargs):
                         }
                     )
                 elif state == "Complete":
-                    if starts_at > now:
-                        exams_complete.append(
-                            {
-                                "name": name,
-                                "date": starts_at.strftime("%d %b %Y"),
-                                "time": starts_at.strftime("%I:%M %p %Z"),
-                                "timezone": timezone,
-                                "state": state,
-                                "uuid": r["uuid"],
-                                "actions": actions,
-                            }
-                        )
-                    else:
-                        exams_not_taken.append(
-                            {
-                                "name": name,
-                                "date": starts_at.strftime("%d %b %Y"),
-                                "time": starts_at.strftime("%I:%M %p %Z"),
-                                "timezone": timezone,
-                                "state": state,
-                                "uuid": r["uuid"],
-                                "actions": actions,
-                            }
-                        )
+                    exams_complete.append(
+                        {
+                            "name": name,
+                            "date": starts_at.strftime("%d %b %Y"),
+                            "time": starts_at.strftime("%I:%M %p %Z"),
+                            "timezone": timezone,
+                            "state": state,
+                            "uuid": r["uuid"],
+                            "actions": actions,
+                        }
+                    )
                 elif state == "Cancelled":
                     actions = [
                         {