From 3ddc0e286058ee1d57dc6107b7f1d681d56e494a Mon Sep 17 00:00:00 2001
From: Scott Idem <stidem@gmail.com>
Date: Thu, 21 Apr 2022 23:33:12 -0400
Subject: [PATCH] Work on exhibit tracking export

---
 app/methods/event_exhibit_tracking_methods.py |  2 +-
 app/routers/event_exhibit_tracking.py         | 45 +++++++++++++++++--
 2 files changed, 42 insertions(+), 5 deletions(-)

diff --git a/app/methods/event_exhibit_tracking_methods.py b/app/methods/event_exhibit_tracking_methods.py
index caf4ca1..17a9485 100644
--- a/app/methods/event_exhibit_tracking_methods.py
+++ b/app/methods/event_exhibit_tracking_methods.py
@@ -170,7 +170,7 @@ def load_event_exhibit_tracking_obj(
         inc_address: bool = False,
         inc_contact: bool = False,
         inc_event_badge: bool = False,
-        inc_event_exhibit: bool = False,
+        # inc_event_exhibit: bool = False,
         inc_event_person: bool = False,
         inc_event_person_profile: bool = False,
         ) -> Event_Exhibit_Tracking_Base|dict|bool:
diff --git a/app/routers/event_exhibit_tracking.py b/app/routers/event_exhibit_tracking.py
index 71ac140..5cc5546 100644
--- a/app/routers/event_exhibit_tracking.py
+++ b/app/routers/event_exhibit_tracking.py
@@ -99,7 +99,6 @@ async def post_event_exhibit_tracking_obj(
         event_exhibit_tracking_obj = load_event_exhibit_tracking_obj(
             event_exhibit_tracking_id = event_exhibit_tracking_id,
             inc_event_badge = inc_event_badge,
-            inc_event_exhibit = inc_event_exhibit,
             inc_event_person = inc_event_person,
             ).dict(by_alias=commons.by_alias, exclude_unset=commons.exclude_unset)
         data = event_exhibit_tracking_obj
@@ -150,7 +149,6 @@ async def patch_event_exhibit_tracking_obj(
         event_exhibit_tracking_obj = load_event_exhibit_tracking_obj(
             event_exhibit_tracking_id = event_exhibit_tracking_id,
             inc_event_badge = inc_event_badge,
-            inc_event_exhibit = inc_event_exhibit,
             inc_event_person = inc_event_person,
             ).dict(by_alias=commons.by_alias, exclude_unset=commons.exclude_unset)
         data = event_exhibit_tracking_obj
@@ -209,7 +207,6 @@ async def get_event_exhibit_tracking_obj(
     if event_exhibit_tracking_obj_result := load_event_exhibit_tracking_obj(
             event_exhibit_tracking_id = event_exhibit_tracking_id,
             inc_event_badge = inc_event_badge,
-            inc_event_exhibit = inc_event_exhibit,
             inc_event_person = inc_event_person,
             inc_event_person_profile = inc_event_person_profile,
             ):
@@ -224,6 +221,47 @@ async def get_event_exhibit_tracking_obj(
 # ### END ### API Event Exhibit Tracking ### get_event_exhibit_tracking_obj() ###
 
 
+# ### BEGIN ### API Event Exhibit Tracking ### get_event_exhibit_obj_tracking_export() ###
+# Updated 2022-04-21
+@router.get('/event/exhibit/{event_exhibit_id}/tracking/export', response_model=Resp_Body_Base)
+async def get_event_exhibit_obj_tracking_list(
+        event_exhibit_id: str = Query(..., min_length=11, max_length=22),
+
+        file_type: str = 'CSV', # CSV, Excel
+        hidden: str = 'not_hidden', # hidden, not_hidden, all
+
+        commons: Common_Route_Params = Depends(common_route_params),
+        ):
+    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
+    log.debug(locals())
+
+    if event_exhibit_id := redis_lookup_id_random(record_id_random=event_exhibit_id, table_name='event_exhibit'): pass
+    else: return mk_resp(data=None, status_code=404, response=commons.response)
+
+    if event_exhibit_tracking_rec_list_result := get_event_exhibit_tracking_rec_list(
+            event_exhibit_id = event_exhibit_id,
+            enabled = commons.enabled,
+            limit = commons.limit,
+            hidden = hidden,
+            ):
+        event_exhibit_tracking_result_list = []
+        for event_exhibit_tracking_rec in event_exhibit_tracking_rec_list_result:
+            if load_event_exhibit_tracking_result := load_event_exhibit_tracking_obj(
+                    event_exhibit_tracking_id = event_exhibit_tracking_rec.get('event_exhibit_tracking_id'),
+                    inc_event_badge = inc_event_badge,
+                    inc_event_person = inc_event_person,
+                    ):
+                event_exhibit_tracking_result_list.append(load_event_exhibit_tracking_result)
+            else:
+                event_exhibit_tracking_result_list.append(None)
+        return mk_resp(data=event_exhibit_tracking_result_list, response=commons.response) # Success
+    elif isinstance(event_exhibit_tracking_rec_list_result, list):
+        return mk_resp(data=False, status_code=404, response=commons.response) # Not Found
+    else:
+        return mk_resp(data=False, status_code=400, response=commons.response) # Bad Request
+# ### END ### API Event Exhibit Tracking ### get_event_exhibit_obj_tracking_list() ###
+
+
 # ### BEGIN ### API Event Exhibit Tracking ### get_event_exhibit_obj_tracking_list() ###
 # Updated 2022-02-15
 @router.get('/event/exhibit/{event_exhibit_id}/tracking/list', response_model=Resp_Body_Base)
@@ -258,7 +296,6 @@ async def get_event_exhibit_obj_tracking_list(
             if load_event_exhibit_tracking_result := load_event_exhibit_tracking_obj(
                     event_exhibit_tracking_id = event_exhibit_tracking_rec.get('event_exhibit_tracking_id'),
                     inc_event_badge = inc_event_badge,
-                    inc_event_exhibit = inc_event_exhibit,
                     inc_event_person = inc_event_person,
                     # by_alias = commons.by_alias,
                     # exclude_unset = commons.exclude_unset,