import datetime from typing import Dict, List, Optional, Set, Union from pydantic import BaseModel, EmailStr, Field, PrivateAttr, ValidationError, validator 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, 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, limit: int = 1000, offset: int = 0, by_alias: bool = True, exclude_unset: bool = True, model_as_dict: bool = False, enabled: str = 'enabled', # enabled, disabled, all ) -> Event_Badge_Template_Base_Out|dict|bool: # log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(locals()) if event_badge_template_id := redis_lookup_id_random(record_id_random=event_badge_template_id, table_name='event_badge_template'): pass else: return False if event_badge_template_rec := sql_select(table_name='event_badge_template', record_id=event_badge_template_id): pass else: return False try: 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()) return False if model_as_dict: return event_badge_template_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset) # pylint: disable=no-member else: return event_badge_template_obj # ### END ### API Event Badge Template Methods ### load_event_badge_template_obj() ### # ### BEGIN ### API Event Badge Template Methods ### get_event_badge_template_rec_list() ### # Updated 2022-07-20 @logger_reset def get_event_badge_template_rec_list( event_id: str, enabled: str = 'enabled', # enabled, disabled, all limit: int = 25, offset: int = 0, ) -> list|bool: log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(locals()) if event_id := redis_lookup_id_random(record_id_random=event_id, table_name='event'): pass else: return False data = {} data['event_id'] = event_id sql_enabled, data['enable'] = sql_enable_part(table_name='event_badge_template', enabled=enabled) # Reasonably safe return str and bool sql_limit = sql_limit_offset_part(limit=limit, offset=offset) # Reasonably safe return str sql = f""" SELECT `event_badge_template`.id AS 'event_badge_template_id', `event_badge_template`.id_random AS 'event_badge_template_id_random' FROM `v_event_badge_template` AS `event_badge_template` WHERE `event_badge_template`.event_id = :event_id {sql_enabled} ORDER BY event_badge_template.name ASC, `event_badge_template`.created_on DESC, `event_badge_template`.updated_on DESC {sql_limit}; """ log.debug(sql) if event_badge_template_rec_li_result := sql_select(data=data, sql=sql, as_list=True): event_badge_template_rec_li = event_badge_template_rec_li_result else: # [] or False event_badge_template_rec_li = event_badge_template_rec_li_result log.debug(event_badge_template_rec_li_result) 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() ###