Work on badge model and other

This commit is contained in:
Scott Idem
2022-07-25 19:47:11 -04:00
parent fccfefe24b
commit 4db2c8cbcb
3 changed files with 165 additions and 17 deletions

View File

@@ -6,10 +6,12 @@ from pydantic import BaseModel, EmailStr, Field, PrivateAttr, ValidationError, v
from app.db_sql import redis_lookup_id_random, sql_enable_part, sql_insert, sql_limit_offset_part, sql_select, sql_update
from app.lib_general import log, logging, logger_reset
from app.models.event_badge_template_models import Event_Badge_Template_Base
from app.models.event_badge_template_models import Event_Badge_Template_Base, Event_Badge_Template_Base_In, Event_Badge_Template_Base_Out
# ### BEGIN ### API Event Badge Template Methods ### load_event_badge_template_obj() ###
# Updated 2022-07-25
@logger_reset
def load_event_badge_template_obj(
event_badge_template_id: int|str,
inc_event_badge_list: bool = False,
@@ -19,7 +21,7 @@ def load_event_badge_template_obj(
exclude_unset: bool = True,
model_as_dict: bool = False,
enabled: str = 'enabled', # enabled, disabled, all
) -> Event_Badge_Template_Base|dict|bool:
) -> Event_Badge_Template_Base_Out|dict|bool:
# log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
log.debug(locals())
@@ -30,7 +32,7 @@ def load_event_badge_template_obj(
else: return False
try:
event_badge_template_obj = Event_Badge_Template_Base(**event_badge_template_rec)
event_badge_template_obj = Event_Badge_Template_Base_Out(**event_badge_template_rec)
log.debug(event_badge_template_obj)
except ValidationError as e:
log.error(e.json())
@@ -84,3 +86,86 @@ def get_event_badge_template_rec_list(
return event_badge_template_rec_li
# ### END ### API Event Badge Template Methods ### get_event_badge_template_rec_list() ###
# ### BEGIN ### API Event Badge Template Methods ### create_update_event_badge_template_obj_v4() ###
# Updated 2022-07-25
@logger_reset
def create_update_event_badge_template_obj_v4(
event_badge_template_dict_obj: Event_Badge_Template_Base_In|dict,
event_badge_template_id: int|str = None,
event_id: int|str = None,
return_outline: bool = False,
) -> int|bool:
log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
log.debug(locals())
# ### SECTION ### Secondary data validation
log.info('Checking requirements...')
if event_badge_template_id:
log.info(f'Event Badge Template ID passed. Update existing Event Badge Template. Event Badge Template ID: {event_badge_template_id}')
if event_badge_template_id := redis_lookup_id_random(record_id_random=event_badge_template_id, table_name='event_badge_template'): pass
else:
log.error('Event Badge Template ID passed but is invalid. Failed requirement.')
return False
else:
log.info('No Event Badge Template ID passed. Create new Event Badge Template. Required: Event ID')
if event_id := redis_lookup_id_random(record_id_random=event_id, table_name='event'): pass
else:
log.error('Missing or invalid Event ID passed. Failed requirement.')
log.info(f'Event ID: {event_id}')
return False
log.info('Create dictionary or Pydantic object')
log.debug(type(event_badge_template_dict_obj))
if isinstance(event_badge_template_dict_obj, dict):
event_badge_template_dict = event_badge_template_dict_obj
if event_badge_template_id:
event_badge_template_dict['event_badge_template_id'] = event_badge_template_id
try:
event_badge_template_obj = Event_Badge_Template_Base_In(**event_badge_template_dict)
except ValidationError as e:
log.error(e.json())
return False
else:
event_badge_template_obj = event_badge_template_dict_obj
if event_badge_template_id:
# NOTE: Can't update the ID alias if it was never set.
event_badge_template_obj.id = event_badge_template_id
log.debug(event_badge_template_obj)
event_badge_template_dict = event_badge_template_obj.dict(by_alias=False, exclude_defaults=False, exclude_unset=True, exclude={'event_badge_list', 'event_id_random', 'created_on', 'updated_on'})
# ### SECTION ### Process data
if event_id: pass
elif event_id := event_badge_template_obj.event_id: pass
if event_badge_template_id:
if event_badge_template_dict_up_result := sql_update(data=event_badge_template_dict, table_name='event_badge_template', rm_id_random=True): pass
else:
log.warning(f'Event Badge Template not updated. Event Badge Template ID: {event_badge_template_id}')
log.debug(event_badge_template_dict_up_result)
return False
log.debug(event_badge_template_dict_up_result)
else:
if event_badge_template_dict_in_result := sql_insert(data=event_badge_template_dict, table_name='event_badge_template', rm_id_random=True): pass
else:
log.warning(f'Event Badge Template not created.')
log.debug(event_badge_template_dict_in_result)
return False
log.debug(event_badge_template_dict_in_result)
event_badge_template_id = event_badge_template_dict_in_result
event_badge_template_outline = {}
event_badge_template_outline['event_badge_template_id'] = event_badge_template_id
if return_outline:
log.debug(f'Returning the Event Badge Outline: {event_badge_template_outline}')
return event_badge_template_outline
else:
log.debug(f'Returning the Event Badge ID: {event_badge_template_id}')
return event_badge_template_id
# ### END ### API Event Badge Methods ### create_update_event_badge_template_obj_v4() ###