import datetime, json, pytz, secrets, time
import pandas, xlrd # qrcode
from fastapi import APIRouter, Body, Depends, Header, HTTPException, Query, Response, status
from pydantic import BaseModel, EmailStr, Field
from typing import Dict, List, Optional, Set, Union

from app.lib_general import log, logging, secure_hash_string, common_route_params, Common_Route_Params
from app.config import settings
from app.db_sql import sql_insert, sql_update, sql_insert_or_update, sql_select, sql_delete, redis_lookup_id_random

from app.routers.api_crud import delete_obj_template, get_obj_template, get_obj_li_template, patch_obj_template, post_obj_template

from app.methods.e_impexium_methods import get_custom_fields, get_events, get_event_registrants

from app.methods.event_person_methods import create_event_person_obj, create_update_event_person_obj_v4, get_event_person_rec_list, load_event_person_obj, update_event_person_obj, update_event_person_obj_v3

from app.models.event_badge_models import Event_Badge_Base
from app.models.response_models import Resp_Body_Base, mk_resp


router = APIRouter()


# ### BEGIN ### API Impexium ### event_import_reg() ###
# Updated 2021-10-07
@router.get('/event/{e_impexium_event_id}/import_reg', response_model=Resp_Body_Base)
async def event_import_reg(
        e_impexium_event_id: str = Query(..., min_length=11, max_length=22), # For ISHLT: 42_AM (2022-04); EX22_AM (2022-04); 41V_2 (2021-04)
        details: bool = True,
        page: int = 0, # 250 per page from Impexium
        event_id: str = Query(..., min_length=11, max_length=22), # For ISHLT: ZDzTBlevhZs (2022-04)
        # Account ID For ISHLT: d8TqXqf1EOg
        commons: Common_Route_Params = Depends(common_route_params),
        ):
    log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    log.info('Starting Impexium event registration import...')

    account_id = commons.x_account_id

    if event_id := redis_lookup_id_random(record_id_random=event_id, table_name='event'): pass
    else: return mk_resp(data=None, status_code=404, status_message=f'The Event ID was not found. Event ID: {event_id}', response=commons.response)

    event_registrant_li = get_event_registrants(event_code=e_impexium_event_id, details=details, page=page)
    log.debug(type(event_registrant_li))

    # return mk_resp(data=len(event_registrant_li), status_code=501, response=commons.response) # Not Implemented

    loop_count = 0
    loop_limit = 2500 # 200
    event_person_li = []
    for event_registrant in event_registrant_li:
        if loop_count > loop_limit: break
        log.info(f"Record Num: {event_registrant.get('recordNumber')}, Registration Num: {event_registrant.get('registrationNumber')}, Registrant Type Code: {event_registrant.get('registrantTypeCode')}")
        log.debug(event_registrant)

        # Creating an external ID from the Impexium record number and registration number. Very explicitly converting each number to a string before joining, just in case.
        external_id = str(event_registrant.get('recordNumber'))+':'+str(event_registrant.get('registrationNumber'))

        event_person_data = {}
        event_person_data['account_id'] = account_id
        event_person_data['event_id'] = event_id
        event_person_data['external_id'] = external_id

        email = None
        if details:
            if emails := event_registrant.get('emails'):
                if isinstance(emails, list) and len(emails):
                    email = emails[0].get('address')

        event_person_profile_data = {}
        event_person_profile_data['pronouns'] = event_registrant.get('gender')
        event_person_profile_data['informal_name'] = event_registrant.get('preferredFirstName')

        event_person_profile_data['title_names'] = event_registrant.get('prefix')
        event_person_profile_data['given_name'] = event_registrant.get('firstName')
        event_person_profile_data['middle_name'] = event_registrant.get('middleName')
        event_person_profile_data['family_name'] = event_registrant.get('lastName')
        event_person_profile_data['designations'] = event_registrant.get('suffix')

        event_person_profile_data['professional_title'] = event_registrant.get('title')

        event_person_profile_data['display_name'] = event_registrant.get('badgeName')

        event_person_profile_data['affiliations'] = event_registrant.get('badgeOrganization')

        if email: event_person_profile_data['email'] = email

        event_person_data['event_person_profile'] = {}
        event_person_data['event_person_profile'] = event_person_profile_data

        event_badge_data = {}
        event_badge_data['event_id'] = event_id
        event_badge_data['external_id'] = external_id

        if reg_type_code := event_registrant.get('registrantTypeCode'):
            event_badge_data['badge_type_code'] = event_registrant.get('registrantTypeCode') # Using this as the badge_type

            ishlt_m = ('AHMB', 'IMBR', 'MBR')
            ishlt_nm = ('ANHM', 'INMB', 'NMBR')
            ishlt_student_m = ('SMBR')
            ishlt_student_nm = ('SNMB')
            ishlt_ex_all = ('EX')
            ishlt_ex_booth = ('EX')
            ishlt_guest = ('GUEST')
            if reg_type_code in ishlt_m:
                event_badge_data['badge_type'] = 'Member'
            elif reg_type_code in ishlt_nm:
                event_badge_data['badge_type'] = 'Non-Member'
            elif reg_type_code in ishlt_student_m:
                event_badge_data['badge_type'] = 'Student/Trainee Member'
            elif reg_type_code in ishlt_student_nm:
                event_badge_data['badge_type'] = 'Student/Trainee Non-Member'
            elif reg_type_code in ishlt_ex_all:
                event_badge_data['badge_type'] = 'Exhibitor All Access'
            elif reg_type_code in ishlt_ex_booth:
                event_badge_data['badge_type'] = 'Exhibitor Booth Staff'
            elif reg_type_code in ishlt_guest:
                event_badge_data['badge_type'] = 'Guest'
            else:
                event_badge_data['badge_type'] = event_registrant.get('registrantTypeName')

            event_badge_data['event_badge_template_id'] = 3
        elif e_impexium_event_id == '42_AM':
            if not reg_type_code and event_registrant.get('guestOfRecordNumber'):
                event_badge_data['badge_type_code'] = 'GUEST'
                event_badge_data['badge_type'] = 'Guest'
            else:
                event_badge_data['badge_type_code'] = 'UNKNOWN'
                event_badge_data['badge_type'] = 'Unknown Type'
            event_badge_data['event_badge_template_id'] = 3
        elif e_impexium_event_id == 'EX22_AM':
            event_badge_data['badge_type_code'] = 'EX'
            event_badge_data['badge_type'] = 'Exhibit Staff'

            if custom_fields := event_registrant.get('itemizedCustomFields'):
                if isinstance(custom_fields, list):
                    log.info('Found list of custom fields for an exhibitor. Searching for "exhibitor_reg_type')
                    for field in custom_fields:
                        if field.get('name') == 'exhibitor_reg_type':
                            if field.get('value') == 'Exhibitor Booth Staff ':
                                event_badge_data['badge_type_code'] = 'EXO'
                                event_badge_data['badge_type'] = 'Exhibitor Booth Staff'
                            elif field.get('value') == 'Exhibitor All Access ':
                                event_badge_data['badge_type_code'] = 'EXALL'
                                event_badge_data['badge_type'] = 'Exhibitor All Access'
                            break
                else: log.info('Found custom fields section, but no list was found.')
            else:
                log.warning('The custom fields were not found at all. Are the "details" being included?')
            event_badge_data['event_badge_template_id'] = 3
        elif e_impexium_event_id == '2022HEART':
            if not reg_type_code and event_registrant.get('guestOfRecordNumber'):
                event_badge_data['badge_type_code'] = 'GUEST'
                event_badge_data['badge_type'] = 'Guest'
            else:
                event_badge_data['badge_type_code'] = 'HEART'
                event_badge_data['badge_type'] = 'HFTX Core'
            event_badge_data['event_badge_template_id'] = 5
        elif e_impexium_event_id == '2022LUNG':
            if not reg_type_code and event_registrant.get('guestOfRecordNumber'):
                event_badge_data['badge_type_code'] = 'GUEST'
                event_badge_data['badge_type'] = 'Guest'
            else:
                event_badge_data['badge_type_code'] = 'LUNG'
                event_badge_data['badge_type'] = 'LTX Core'
            event_badge_data['event_badge_template_id'] = 5
        else:
            event_badge_data['badge_type_code'] = 'UNKNOWN'
            event_badge_data['badge_type'] = 'Unknown Type'
            event_badge_data['event_badge_template_id'] = 3

        event_badge_data['member_type_code'] = event_registrant.get('customerType') # Using this as the member_type
        event_badge_data['registration_type_code'] = event_registrant.get('registrantTypeCode')
        event_badge_data['registration_type'] = event_registrant.get('registrantTypeName')

        event_badge_data['pronouns'] = event_registrant.get('gender')
        event_badge_data['informal_name'] = event_registrant.get('preferredFirstName')

        event_badge_data['title_names'] = event_registrant.get('prefix')
        event_badge_data['given_name'] = event_registrant.get('firstName')
        event_badge_data['middle_name'] = event_registrant.get('middleName')
        event_badge_data['family_name'] = event_registrant.get('lastName')
        event_badge_data['designations'] = event_registrant.get('suffix')

        event_badge_data['professional_title'] = event_registrant.get('title')

        event_badge_data['display_name'] = event_registrant.get('badgeName')

        event_badge_data['affiliations'] = event_registrant.get('badgeOrganization')

        if email: event_badge_data['email'] = email

        event_badge_data['city'] = event_registrant.get('badgeCity')
        event_badge_data['state_province'] = event_registrant.get('badgeState')

        event_person_data['event_badge'] = {}
        event_person_data['event_badge'] = event_badge_data

        sql_select_event_person = f"""
            SELECT id AS event_person_id, id_random AS event_person_id_random, external_id AS event_person_external_id, event_badge_id AS event_badge_id, event_person_profile_id AS event_person_profile_id
            FROM `event_person` AS `event_person`
            WHERE event_person.event_id = :event_id
                AND event_person.external_id = :external_id
            /*LIMIT 1*/;
            """

        if event_person_result := sql_select(sql=sql_select_event_person, data=event_person_data):
            if isinstance(event_person_result, list):
                log.error(f'Found more than one Event Person with the same External ID. Count: {len(event_person_result)}')
                break
                # return False
            else:
                event_person_id = event_person_result.get('event_person_id')
                event_badge_id = event_person_result.get('event_badge_id')
                event_person_profile_id = event_person_result.get('event_person_profile_id')
                log.info(f'Found Event Person. Updating existing... Event Person ID: {event_person_id}')
                if create_event_person_obj_result := create_update_event_person_obj_v4(
                        event_person_dict_obj = event_person_data,
                        event_person_id = event_person_id,
                        account_id = account_id,
                        event_id = event_id,
                        event_badge_id = event_badge_id,
                        event_person_profile_id = event_person_profile_id,
                        # create_sub_obj = create_sub_obj,
                        # fail_any = fail_any,
                        # return_outline = False,
                        ):
                    event_person_id = create_event_person_obj_result
                    log.warning(f'Event Person updated. Event Person ID: {event_person_id}')
                else:
                    log.warning(f'Event Person not updated. Event Person ID: {event_person_id}')
                    log.debug(event_badge_obj_in_result)
                    break
                    # return False
        else:
            log.info('No Event Person found. Creating new...')

            if create_event_person_obj_result := create_update_event_person_obj_v4(
                    event_person_dict_obj = event_person_data,
                    account_id = account_id,
                    event_id = event_id,
                    # create_sub_obj = create_sub_obj,
                    # fail_any = fail_any,
                    # return_outline = False,
                    ):
                event_person_id = create_event_person_obj_result
                log.warning(f'Event Person created. Event Person ID: {event_person_id}')
            else:
                log.warning(f'Event Person not created.')
                log.debug(event_badge_obj_in_result)
                break
                # return False

        # event_badge_data = {}
        # external_id = event_registrant.get('recordNumber')+':'+event_registrant.get('registrationNumber')
        # event_badge_data['external_id'] = external_id
        # # event_badge_data['reg_num'] = event_registrant.get('registrationNumber')
        # event_badge_data['badge_type_code'] = event_registrant.get('registrantTypeCode') # Using this as the badge_type
        # event_badge_data['badge_type'] = event_registrant.get('registrantTypeName')
        # event_badge_data['member_type_code'] = event_registrant.get('customerType') # Using this as the member_type
        # event_badge_data['registration_type_code'] = event_registrant.get('registrantTypeCode')
        # event_badge_data['registration_type'] = event_registrant.get('registrantTypeName')

        # event_badge_data['pronouns'] = event_registrant.get('gender')

        # event_badge_data['informal_name'] = event_registrant.get('preferredFirstName')
        # event_badge_data['title_names'] = event_registrant.get('prefix')
        # event_badge_data['given_name'] = event_registrant.get('firstName')
        # event_badge_data['middle_name'] = event_registrant.get('middleName')
        # event_badge_data['family_name'] = event_registrant.get('lastName')
        # # event_badge_data['not_sure'] = event_registrant.get('secondLastName')

        # event_badge_data['designations'] = event_registrant.get('suffix')

        # event_badge_data['professional_title'] = event_registrant.get('title')

        # event_badge_data['display_name'] = event_registrant.get('badgeName')

        # event_badge_data['affiliations'] = event_registrant.get('badgeOrganization')

        # event_badge_data['city'] = event_registrant.get('badgeCity')
        # event_badge_data['state_province'] = event_registrant.get('badgeState')

        # id, attendedDate, sessions, itemizedCustomFields, guestOfRecordNumber, registrantTypeName, boughtTogetherWith, primaryOrganization, customerType, showInDirectory, addresses, emails, phones

        event_person_li.append(event_person_data)
        loop_count = loop_count + 1

    return mk_resp(data=event_person_li, response=commons.response)
# ### END ### API Impexium ### event_import_reg() ###