Work on Impexium, exhibit, and badge related

app/methods/event_badge_methods.py

@@ -91,3 +91,89 @@ def get_event_badge_template_id_w_event_id(
         else: return False
     else: return None
 # ### END ### API Event Badge Methods ### get_event_badge_template_id_w_event_id() ###
+
+
+# ### BEGIN ### API Event Badge Methods ### create_update_event_badge_obj_v4() ###
+# Updated 2022-02-23
+def create_update_event_badge_obj_v4(
+        event_badge_dict_obj: Event_Badge_Base|dict,
+        event_badge_id: int|str = None,
+        # account_id: int|str = None,
+        # event_id: int|str = None,
+        event_person_id: int|str = None,
+        create_sub_obj: bool = False,
+        fail_any: bool = False, # Fail if any thing goes wrong for sub objects
+        return_outline: bool = False,
+        ) -> int|bool:
+    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
+    log.debug(locals())
+
+    # ### SECTION ### Secondary data validation
+    log.info('Checking requirements...')
+    if event_badge_id:
+        log.info(f'Event Badge ID passed. Update existing Event Badge. Event Badge ID: {event_badge_id}')
+
+        if event_badge_id := redis_lookup_id_random(record_id_random=event_badge_id, table_name='event_badge'): pass
+        else:
+            log.error('Event Badge ID passed but is invalid. Failed requirement.')
+            return False
+    else:
+        log.info('No Event Badge ID passed. Create new Event Badge. Required: Event Person ID')
+
+        if event_person_id := redis_lookup_id_random(record_id_random=event_person_id, table_name='event_person'): pass
+        else:
+            log.error('Missing or invalid Event Person ID passed. Failed requirement.')
+            log.info(f'Event Person ID: {event_person_id}')
+            return False
+
+    log.info('Create dictionary or Pydantic object')
+    log.debug(type(event_badge_dict_obj))
+    if isinstance(event_badge_dict_obj, dict):
+        event_badge_dict = event_badge_dict_obj
+        if event_badge_id:
+            event_badge_dict['event_badge_id'] = event_badge_id
+        try:
+            event_badge_obj = Event_Badge_Base(**event_badge_dict)
+        except ValidationError as e:
+            log.error(e.json())
+            return False
+    else:
+        event_badge_obj = event_badge_dict_obj
+        if event_badge_id:
+            # NOTE: Can't update the ID alias if it was never set.
+            event_badge_obj.id = event_badge_id
+    log.debug(event_badge_obj)
+
+    event_badge_dict = event_badge_obj.dict(by_alias=False, exclude_defaults=False, exclude_unset=True, exclude={'event_badge_template', 'event_id', 'event_id_random', 'created_on', 'updated_on'})
+
+    # ### SECTION ### Process data
+    if event_person_id: pass
+    elif event_person_id := event_badge_obj.event_person_id: pass
+
+    if event_badge_id:
+        if event_badge_dict_up_result := sql_update(data=event_badge_dict, table_name='event_badge', rm_id_random=True): pass
+        else:
+            log.warning(f'Event Badge not updated. Event Badge ID: {event_badge_id}')
+            log.debug(event_badge_dict_up_result)
+            return False
+        log.debug(event_badge_dict_up_result)
+    else:
+        if event_badge_dict_in_result := sql_insert(data=event_badge_dict, table_name='event_badge', rm_id_random=True, id_random_length=8): pass
+        else:
+            log.warning(f'Event Badge not created.')
+            log.debug(event_badge_dict_in_result)
+            return False
+        log.debug(event_badge_dict_in_result)
+
+        event_badge_id = event_badge_dict_in_result
+
+    event_badge_outline = {}
+    event_badge_outline['event_badge_id'] = event_badge_id
+
+    if return_outline:
+        log.debug(f'Returning the Event Badge Outline: {event_badge_outline}')
+        return event_badge_outline
+    else:
+        log.debug(f'Returning the Event Badge ID: {event_badge_id}')
+        return event_badge_id
+# ### END ### API Event Badge Methods ### create_update_event_badge_obj_v4() ###