from __future__ import annotations import datetime from typing import Dict, List, Optional, Set, Union from pydantic import BaseModel, EmailStr, Field, PrivateAttr, ValidationError, validator from app.db_sql import get_account_id_w_for_type_id, redis_lookup_id_random, sql_insert, sql_select, sql_update from app.lib_general import log, logging from app.models.common_field_schema import default_num_bytes from app.models.event_person_tracking_models import Event_Person_Tracking_Base # ### BEGIN ### API Event Person Tracking Methods ### load_event_person_tracking_obj() ### # Updated 2021-09-21 def load_event_person_tracking_obj( event_person_tracking_id: int|str, limit: int = 1000, by_alias: bool = True, exclude_unset: bool = True, model_as_dict: bool = False, enabled: str = 'enabled', # enabled, disabled, all inc_event_session: bool = False, inc_event_person: bool = False, ) -> Event_Person_Tracking_Base|dict|bool: log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(locals()) if event_person_tracking_id := redis_lookup_id_random(record_id_random=event_person_tracking_id, table_name='event_person_tracking'): pass else: return False if event_person_tracking_rec := sql_select(table_name='v_event_person_tracking', record_id=event_person_tracking_id): pass else: return False try: event_person_tracking_obj = Event_Person_Tracking_Base(**event_person_tracking_rec) log.debug(event_person_tracking_obj) except ValidationError as e: log.error(e.json()) return False if inc_event_person: log.info('Need to include Event Person data...') event_person_id = event_person_tracking_rec.get('event_person_id', None) log.debug(event_person_id) from app.methods.event_person_methods import load_event_person_obj if event_person_result := load_event_person_obj( event_person_id = event_person_id, by_alias = by_alias, exclude_unset = exclude_unset, model_as_dict = model_as_dict, ): event_person_tracking_obj.event_person = event_person_result else: event_person_tracking_obj.event_person = None if inc_event_session: log.info('Need to include Event Session data...') event_session_id = event_person_tracking_rec.get('event_session_id', None) log.debug(event_session_id) from app.methods.event_session_methods import load_event_session_obj if event_session_result := load_event_session_obj( event_session_id = event_session_id, by_alias = by_alias, exclude_unset = exclude_unset, model_as_dict = model_as_dict, ): event_person_tracking_obj.event_session = event_session_result else: event_person_tracking_obj.event_session = None if model_as_dict: return event_person_tracking_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset) # pylint: disable=no-member else: return event_person_tracking_obj # ### END ### API Event Person Tracking Methods ### load_event_person_tracking_obj() ### # ### BEGIN ### API Event Person Tracking Methods ### get_event_person_tracking_rec_list() ### # for_obj_type: account, event, event_session, event_person # Updated 2021-09-21 def get_event_person_tracking_rec_list( account_id: int|str, event_id: int|str = None, event_person_id: int|str = None, event_session_id: int|str = None, in_out_type: str = None, # for_obj_type: str, # NOTE: This is not for_type because the field name generated based # for_obj_id: str, # NOTE: This is not for_id because the field name generated based enabled: str = 'enabled', # enabled, disabled, all limit: int = 1000, ) -> list|bool: log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(locals()) data = {} if account_id := redis_lookup_id_random(record_id_random=account_id, table_name='account'): pass else: return False data['account_id'] = account_id # sql_select_account_id = f'`event_person_tracking`.account_id = :account_id' sql_select_account_id = '1=1' sql_select_event_id = '' sql_select_event_person_id = '' sql_select_event_session_id = '' if event_id: if event_id := redis_lookup_id_random(record_id_random=event_id, table_name='event'): pass else: return False data['event_id'] = event_id sql_select_event_id = f'AND `event_person_tracking`.event_id = :event_id' if event_person_id: if event_person_id := redis_lookup_id_random(record_id_random=event_person_id, table_name='event_person'): pass else: return False data['event_person_id'] = event_person_id sql_select_event_person_id = f'AND `event_person_tracking`.event_person_id = :event_person_id' if event_session_id: if event_session_id := redis_lookup_id_random(record_id_random=event_session_id, table_name='event_session'): pass else: return False data['event_session_id'] = event_session_id sql_select_event_session_id = f'AND `event_person_tracking`.event_session_id = :event_session_id' if event_id or event_person_id or event_session_id: pass else: return False if in_out_type: if in_out_type == 'check': sql_select_in_out_type = f'AND `event_person_tracking`.check_in_out = 1' elif in_out_type == 'break': sql_select_in_out_type = f'AND `event_person_tracking`.break_in_out = 1' else: data['other_type'] = in_out_type sql_select_in_out_type = f'AND `event_person_tracking`.other_in_out = 1 AND `event_person_tracking`.other_type = :other_type' else: sql_select_in_out_type = f'' # if for_obj_id := redis_lookup_id_random(record_id_random=for_obj_id, table_name=for_obj_type): pass # else: return False # data[f'{for_obj_type}_id'] = for_obj_id # sql_obj_type_id = f'`event_person_tracking`.{for_obj_type}_id = :{for_obj_type}_id' # if enabled in ['enabled', 'disabled', 'all']: # if enabled == 'enabled': # data['enable'] = True # sql_enabled = f'AND `event_person_tracking`.enable = :enable' # elif enabled == 'disabled': # data['enable'] = False # sql_enabled = f'AND `event_person_tracking`.enable = :enable' # elif enabled == 'all': # sql_enabled = '' sql_enabled = '' if limit: data['limit'] = limit sql_limit = f'LIMIT :limit' else: sql_limit = '' sql = f""" SELECT `event_person_tracking`.id AS 'event_person_tracking_id', `event_person_tracking`.id_random AS 'event_person_tracking_id_random' FROM `event_person_tracking` AS `event_person_tracking` WHERE {sql_select_account_id} {sql_select_event_id} {sql_select_event_person_id} {sql_select_event_session_id} {sql_enabled} {sql_select_in_out_type} ORDER BY `event_person_tracking`.created_on DESC, `event_person_tracking`.updated_on DESC {sql_limit}; """ if event_person_rec_li_result := sql_select(data=data, sql=sql, as_list=True): event_person_rec_li = event_person_rec_li_result else: event_person_rec_li = [] log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(event_person_rec_li_result) return event_person_rec_li # ### END ### API Event Person Tracking Methods ### get_event_person_tracking_rec_list() ###