import datetime, pytz

from typing import Dict, List, Optional, Set, Union
from pydantic import BaseModel, EmailStr, Field, Json, PrivateAttr, ValidationError, validator

from app.db_sql import get_id_random, redis_lookup_id_random
# from app.lib_general import log, logging
from app.log import log, logging, logger_reset

from app.models.common_field_schema import base_fields, default_num_bytes
from app.models.hosted_file_models import Hosted_File_Base


# ### BEGIN ### API Event File Models ### Event_File_Base() ###
class Event_File_Base(BaseModel):
    log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    id_random: Optional[str] = Field(
        # **base_fields['event_file_id_random'],
        alias = 'event_file_id_random',
    )
    id: Optional[int] = Field(
        alias = 'event_file_id'
    )

    hosted_file_id_random: Optional[str]
    hosted_file_id: Optional[int]

    # NOTE: Handling this outside of the Pydantic model and model validation. See below as well. -STI 2021-09-10
    for_type: Optional[str]
    for_id: Optional[int] # NOTE: This is reversed with for_id_random
    for_id_random: Optional[str] # NOTE: This is reversed with for_id
    # for_id_random: Optional[str] = None # Need to override value from common_field_schema.py
    # for_id: Optional[int]

    event_id_random: Optional[str]
    event_id: Optional[int]
    event_exhibit_id_random: Optional[str]
    event_exhibit_id: Optional[int]
    event_location_id_random: Optional[str]
    event_location_id: Optional[int]
    event_presentation_id_random: Optional[str]
    event_presentation_id: Optional[int]
    event_presenter_id_random: Optional[str]
    event_presenter_id: Optional[int]
    event_session_id_random: Optional[str]
    event_session_id: Optional[int]
    event_track_id_random: Optional[str]
    event_track_id: Optional[int]

    filename: Optional[str]
    filename_no_ext: Optional[str] # Currently created with a view
    filename_w_ext: Optional[str] # Currently created with a view
    extension: Optional[str]
    title: Optional[str]
    description: Optional[str]

    lu_file_purpose_id: Optional[int]
    file_purpose: Optional[str]

    # New internal use fields to help with logistics and planning 2022-09-15
    internal_use: Optional[bool] # Will hide from moderators, presenters, non support people, etc

    #### NOTE ***** NOTE
    open_in_os: Optional[str]
    public_use: Optional[bool] = False
    public: Optional[bool]
    approve: Optional[bool]
    #### open_in_os needs to be set
    #### Are other important fields missing????
    publish_optout: Optional[bool]
    #### NOTE ^^^^^ NOTE

    enable: Optional[bool]
    enable_from: Optional[datetime.datetime] = None
    enable_to: Optional[datetime.datetime] = None

    hide: Optional[bool]
    priority: Optional[bool]
    sort: Optional[int]
    group: Optional[str] # Same or similar as file_purpose?

    notes: Optional[str]
    created_on: Optional[datetime.datetime] = None
    updated_on: Optional[datetime.datetime] = None

    # Including convenience data
    # This is only for convenience. Probably going to keep unless it causes a problem.
    hosted_file_hash_sha256: Optional[str] = Field(
        alias = 'hash_sha256'
    )
    hosted_file_subdirectory_path: Optional[str] = Field(  # NOTE: This will frequently only contain numbers, but it still needs to be a string
        alias = 'subdirectory_path',
        exclude = True
    )
    hosted_file_content_type: Optional[str] = Field(
        alias = 'content_type'
    )
    hosted_file_size: Optional[str] = Field(
        alias = 'file_size'
    )

    lu_event_file_purpose_name: Optional[str] = Field(
        alias = 'file_purpose_name'
    )

    event_name: Optional[str]
    event_code: Optional[str]
    event_start_datetime: Optional[datetime.datetime]
    event_end_datetime: Optional[datetime.datetime]
    event_location_code: Optional[str]
    event_location_name: Optional[str]
    event_presentation_code: Optional[str]
    event_presentation_type_code: Optional[str]
    event_presentation_name: Optional[str]
    event_presentation_start_datetime: Optional[datetime.datetime]
    event_presentation_end_datetime: Optional[datetime.datetime]
    event_presenter_code: Optional[str]
    event_presenter_given_name: Optional[str]
    event_presenter_family_name: Optional[str]
    event_presenter_full_name: Optional[str]
    event_presenter_email: Optional[str]
    event_session_code: Optional[str]
    event_session_type_code: Optional[str]
    event_session_name: Optional[str]
    event_session_start_datetime: Optional[datetime.datetime]
    event_session_end_datetime: Optional[datetime.datetime]
    event_track_code: Optional[str]
    event_track_name: Optional[str]

    # Including other related objects
    hosted_file: Optional[Union[Hosted_File_Base, None]]

    _processed_at: datetime.datetime = PrivateAttr(default_factory=datetime.datetime.now)

    #@validator('event_file_id_random', always=True)
    def event_file_id_random_copy(cls, v, values, **kwargs):
        if values['id_random']:
            return values['id_random']
        return None

    @validator('id', always=True)
    def event_file_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_file')
        return None

    @validator('hosted_file_id', always=True)
    def hosted_file_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('hosted_file_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='hosted_file')
        return None

    @validator('event_id', always=True)
    def event_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event')
        return None

    @validator('event_exhibit_id', always=True)
    def event_exhibit_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_exhibit_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_exhibit')
        return None

    @validator('event_location_id', always=True)
    def event_location_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_location_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_location')
        return None

    @validator('event_presentation_id', always=True)
    def event_presentation_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_presentation_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_presentation')
        return None

    @validator('event_presenter_id', always=True)
    def event_presenter_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_presenter_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_presenter')
        return None

    @validator('event_session_id', always=True)
    def event_session_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_session_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_session')
        return None

    @validator('event_track_id', always=True)
    def event_track_id_lookup(cls, v, values, **kwargs):
        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_track_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_track')
        return None

    # NOTE: I kind of give up on this. Handling this outside of Pydantic and before the data is even attempted to be loaded into the Event_File_Base model. -STI 2021-09-10
    # NOTE: This validator will try to find and "set" the for_id_random value. However, The value is not really "set" in Pydantic. To get this value, exclude_unset=True when returning a dict from the model.
    @validator('for_id_random', always=True)
    def for_id_random_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())
        if isinstance(v, str): return v
        elif values.get('for_id') and values['for_type']:
            return get_id_random(record_id=values['for_id'], table_name=values['for_type'])
        return None

    # @validator('for_id', always=True)
    # def for_id_lookup(cls, v, values, **kwargs):
    #     log.setLevel(logging.DEBUG)
    #     log.debug(locals())

    #     if values.get('for_id_random', None) and values['for_type']:
    #         return redis_lookup_id_random(record_id_random=values['for_id_random'], table_name=values['for_type'])
    #     # return None
    #     else: return v

    class Config:
        underscore_attrs_are_private = True
        allow_population_by_field_name = True
        fields = base_fields
# ### END ### API Event File Models ### Event_File_Base() ###