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 redis_lookup_id_random, sql_insert, sql_select, sql_update
from app.lib_general import log, logging

# from app.methods.address_methods import load_address_obj
# from app.methods.contact_methods import load_contact_obj
from app.methods.event_person_methods import get_event_person_rec_list, load_event_person_obj
from app.methods.event_registration_cfg_methods import load_event_registration_cfg_obj
# from app.methods.person_methods import load_person_obj

from app.models.event_registration_models import Event_Registration_Base
from app.models.event_cfg_models import Event_Cfg_Base


# ### BEGIN ### API Event Registration Methods ### load_event_registration_obj() ###
def load_event_registration_obj(
        event_registration_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_address: bool = False, # Under contact
        inc_contact: bool = False,
        # inc_event_cfg: bool = False,
        inc_event_person_list: bool = False,
        inc_event_registration_cfg: bool = False,
        # inc_event_registration_list: bool = False,
        inc_person: bool = False,
        ) -> Event_Registration_Base|bool:
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if event_registration_id := redis_lookup_id_random(record_id_random=event_registration_id, table_name='event_registration'): pass
    else: return False

    if event_registration_rec := sql_select(table_name='v_event_registration', record_id=event_registration_id): pass
    else: return False

    print('******************************** HERE')

    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(event_registration_rec)

    try:
        log.info('Try to apply event registration record data to Event_Registration_Base...')
        event_registration_obj = Event_Registration_Base(**event_registration_rec)
        log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
        log.debug(event_registration_obj)
    except ValidationError as e:
        log.error(e.json())
        return False

    # Updated 2021-08-17
    if inc_event_registration_cfg:
        log.info('Need to include event registration configuration...')
        event_id = event_registration_rec.get('event_id', None)
        log.info(f'Need to include event registration config for event_id {event_id}...')
        if event_registration_cfg_result := load_event_registration_cfg_obj(
            event_id = event_id,
            by_alias = by_alias,
            exclude_unset = exclude_unset,
            model_as_dict = model_as_dict,
            ):
            event_registration_obj.cfg = event_registration_cfg_result
        else: event_registration_obj.event_registration_cfg = None

    # Updated 2021-08-17
    if inc_event_person_list:
        log.info('Need to include event person list...')
        if event_person_rec_list_result := get_event_person_rec_list(
                for_obj_type = 'event_registration',
                for_obj_id = event_registration_id,
                limit = limit,
                enabled = enabled,
                ):
            event_person_result_list = []
            for event_person_rec in event_person_rec_list_result:
                if load_event_person_result := load_event_person_obj(
                        event_person_id = event_person_rec.get('event_person_id', None),
                        limit = limit,
                        by_alias = by_alias,
                        exclude_unset = exclude_unset,
                        model_as_dict = model_as_dict,
                        enabled = enabled,
                        inc_person = inc_person,
                        # inc_user = inc_user,
                        ):
                    event_person_result_list.append(load_event_person_result)
                else: event_person_result_list.append(None)
            event_registration_obj.event_person_list = event_person_result_list
        else: event_registration_obj.event_person_list = []

    if model_as_dict:
        return event_registration_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset) # pylint: disable=no-member
    else:
        return event_registration_obj
# ### END ### API Event Registration Methods ### load_event_registration_obj() ###


# ### BEGIN ### API Event Registration Methods ### get_event_id_w_event_person_id() ###
# Updated 2021-08-23
def get_event_id_w_event_person_id(
        event_person_id: int|str,
        ) -> bool|int|None:
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if event_person_id := redis_lookup_id_random(record_id_random=event_person_id, table_name='event_person'): pass
    else: return False

    data = {}
    data['event_person_id'] = event_person_id

    sql = f"""
        SELECT `event_person`.id AS 'event_person_id', `event_person`.id_random AS 'event_person_id_random', `event_person`.event_id AS event_id
        FROM `event_person` AS `event_person`
        WHERE `event_person`.id = :event_person_id
        LIMIT 1;
        """

    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    if event_person_data_result := sql_select(data=data, sql=sql):
        log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
        log.debug(event_person_data_result)
        if event_id := event_person_data_result.get('event_id', None): return event_id
        else: return False
    else: return None
# ### END ### API Event Registration Methods ### get_event_id_w_event_person_id() ###


# ### BEGIN ### API Event Registration Methods ### create_event_registration_obj() ###
# Event Registration can only update the primary event_person because they should already exist before registration is started. Chicken and egg problem...
# Event Registration can create or update the event_person_list.
# NOTE: Should there be a check before trying to update the primary event_person if they are also in the event_person_list??? Chicken and egg problem again?
# Updated 2021-08-25
def create_event_registration_obj(
        event_id: int|str,
        event_person_id: int|str,
        event_registration_obj_new: Event_Registration_Base,
        create_sub_obj: bool = False,
        fail_any: bool = False, # Fail if any thing goes wrong for sub objects
        ) -> bool:
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if event_person_id := redis_lookup_id_random(record_id_random=event_person_id, table_name='event_person'): pass
    else: return False # The event_person should be created first (before registration and similar)

    if event_id := redis_lookup_id_random(record_id_random=event_id, table_name='event'): pass
    else:
        if event_id := get_event_id_w_event_person_id(event_person_id): pass
        else:
            log.warning(f'The event_id was not found using the event_person_id. Event Person ID: {event_person_id}')
            return event_id # False or None

    log.debug(type(event_registration_obj_new))
    if isinstance(event_registration_obj_new, dict):
        try:
            event_registration_obj_new = Event_Registration_Base(**event_registration_obj_new)
            log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
            log.debug(event_registration_obj_new)
        except ValidationError as e:
            log.error(e.json())
            return False

    event_registration_obj_new.event_id = event_id
    event_registration_obj_new.event_person_id = event_person_id

    event_registration_obj_data = event_registration_obj_new.dict(by_alias=False, exclude_defaults=False, exclude_unset=True, exclude={'event_person', 'event_person_list', 'created_on', 'updated_on'})
    log.debug(event_registration_obj_data)

    if event_registration_obj_in_result := sql_insert(data=event_registration_obj_data, table_name='event_registration', rm_id_random=True, id_random_length=8): pass
    else:
        log.warning(f'Event Registration not created.')
        log.debug(event_registration_obj_in_result)
        return False

    event_registration_id = event_registration_obj_in_result

    return_dict = {}
    return_dict['event_registration_id'] = None
    return_dict['event_person_id'] = None
    return_dict['event_id'] = None
    return_dict['event_person_list'] = []

    # NOTE: Primary Event Person update check is not finished
    if event_registration_obj_new.event_person:
        # What happens here?
        # Update only... do not create a new event_person here.
        pass

    if event_registration_obj_new.event_person_list and isinstance(event_registration_obj_new.event_person_list, list):
        for event_person_obj_new in event_registration_obj_new.event_person_list:
            # NOTE: This does not account for an edge case where the person already exists. Possibly as part of another registration, which should *not* happen.
            if create_event_person_obj_result := create_event_person_obj(
                    event_registration_id = event_registration_id,
                    event_person_obj_new = event_person_obj_new,
                    create_sub_obj = create_sub_obj,
                    fail_any = fail_any,
                    ):
                if isinstance(create_event_person_obj_result, int):
                    event_person_id = create_event_person_obj_result
                    log.info(f'Event Person created. Event Person ID: {event_person_id}')
                else:
                    log.warning(f'Event Person not created. Event Registration ID: {event_registration_id}')
                    log.debug(create_event_person_obj_result)
                    event_person_id = None
                    if fail_any: return False
            else:
                log.warning(f'Event Person not created. Event Registration ID: {event_registration_id}')
                log.debug(create_event_person_obj_result)
                event_person_id = None
                if fail_any: return False
            return_dict['event_person_list'].append(event_person_id)
    else:
        log.info('Event Person List not found')
        pass

    log.info(f'The event registration has been created. Event Registration ID: {event_registration_id}')
    return event_registration_id
# ### END ### API Event Registration Methods ### create_event_registration_obj() ###


# ### BEGIN ### API Event Registration Methods ### update_event_registration_obj_v3() ###
# Event Registration can only update the primary event_person because they should already exist since registration is started and registration "requires" the primary event_person first. Chicken and egg problem...
# Event Registration can create or update the event_person_list.
# NOTE: Should there be a check before trying to update the primary event_person if they are also in the event_person_list??? Chicken and egg problem again?
# Updated 2021-08-25
def update_event_registration_obj_v3(
        event_registration_id: int|str,
        event_registration_obj_exist: Event_Registration_Base,
        create_sub_obj: bool = False,
        fail_any: bool = False, # Fail if any thing goes wrong for sub objects
        ) -> bool:
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if event_registration_id := redis_lookup_id_random(record_id_random=event_registration_id, table_name='event_registration'): pass
    else: return False

    # Can't update the event_registration_id alias if the .id was never set.
    # event_registration_obj_exist.event_registration_id = event_registration_id
    if not event_registration_obj_exist.id:
        event_registration_obj_exist.id = event_registration_id

    event_registration_obj_data = event_registration_obj_exist.dict(by_alias=False, exclude_defaults=False, exclude_unset=True, exclude={'event_person', 'event_person_list', 'created_on', 'updated_on'})
    log.debug(event_registration_obj_data)

    if event_registration_obj_up_result := sql_update(data=event_registration_obj_data, table_name='event_registration', rm_id_random=True): pass
    else:
        log.warning(f'Event Registration not updated.')
        log.debug(event_registration_obj_up_result)
        return False

    return_dict = {}
    return_dict['event_registration_id'] = None
    return_dict['event_person_id'] = None
    return_dict['event_id'] = None
    return_dict['event_person_list'] = []

    # NOTE: Primary Event Person update check is not finished
    if event_registration_obj_exist.event_person:
        # What happens here?
        # Update only... do not create a new event_person here.
        pass

    if event_registration_obj_exist.event_person_list and isinstance(event_registration_obj_exist.event_person_list, list):
        for event_person_obj_unknown in event_registration_obj_exist.event_person_list:
            log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
            log.debug(event_person_obj_unknown)
            if event_person_id := event_person_obj_unknown.get('event_person_id_random', None):
                if update_event_person_obj_result := update_event_person_obj_v3(
                        event_person_id = event_person_id,
                        event_person_obj_exist = event_person_obj_unknown,
                        create_sub_obj = create_sub_obj,
                        fail_any = fail_any,
                        ):
                    event_person_id = update_event_person_obj_result
                    log.info(f'Event Person updated. Event Person ID: {event_person_id}')
                else:
                    log.warning(f'Event Person not updated. Event Registration ID: {event_registration_id}')
                    log.debug(update_event_person_obj_result)
                    event_person_id = None
                    if fail_any: return False

                    if isinstance(update_event_person_obj_result, int):
                        event_person_id = update_event_person_obj_result
                        log.info(f'Event Person updated. Event Person ID: {event_person_id}')
                    else:
                        log.warning(f'Event Person not updated. Event Registration ID: {event_registration_id}')
                        log.debug(update_event_person_obj_result)
                        event_person_id = None
                        if fail_any: return False
            else:
                log.info(f'No Event Person ID found.')
                if create_event_person_obj_result := create_event_person_obj(
                        event_registration_id = event_registration_id,
                        event_person_obj_new = event_person_obj_unknown,
                        create_sub_obj = create_sub_obj,
                        fail_any = fail_any,
                        ):
                    if isinstance(create_event_person_obj_result, int):
                        event_person_id = create_event_person_obj_result
                        log.info(f'Event Person created. Event Person ID: {event_person_id}')
                    else:
                        log.warning(f'Event Person not created. Event Registration ID: {event_registration_id}')
                        log.debug(create_event_person_obj_result)
                        event_person_id = None
                        if fail_any: return False
                else:
                    log.warning(f'Event Person not created. Event Registration ID: {event_registration_id}')
                    log.debug(create_event_person_obj_result)
                    event_person_id = None
                    if fail_any: return False
            return_dict['event_person_list'].append(event_person_id)
    else:
        log.info('Event Person List not found or not in a list.')
        pass

    log.info(f'The event registration has been updated. Event Registration ID: {event_registration_id}')
    return True
# ### END ### API Event Registration Methods ### update_event_registration_obj_v3() ###