OSIT-AE-API-FastAPI/app/models/event_session_models.py

from __future__ import annotations
import datetime, hashlib, logging, os, pytz, redis, secrets

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

from app.db_sql import redis_lookup_id_random
from app.lib_general import log, logging

from app.models.common_field_schema import base_fields, default_num_bytes
# from app.models.event_models import Event_Base
from app.models.event_location_models import Event_Location_Base
from app.models.event_person_models import Event_Person_Base # NOTE: Using this will probably create an import loop
from app.models.event_presentation_models import Event_Presentation_Base
from app.models.event_track_models import Event_Track_Base
from app.models.person_models import Person_Base


class Event_Session_Base(BaseModel):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    id_random: Optional[str] = Field(
        **base_fields['event_session_id_random'],
        alias = 'event_session_id_random',
        default_factory = lambda:secrets.token_urlsafe(default_num_bytes),
    )
    id: Optional[int] = Field(
        alias = 'event_session_id'
    )

    code: Optional[str] = Field(
        alias = 'event_session_code'
    )

    event_id_random: Optional[str]
    event_id: Optional[int]

    event_location_id_random: Optional[str]
    event_location_id: Optional[int]

    event_track_id_random: Optional[str]
    event_track_id: Optional[int]

    poc_event_person_id_random: Optional[str]
    poc_event_person_id: Optional[int]

    poc_person_id_random: Optional[str]
    poc_person_id: Optional[int]

    session_type_id_random: Optional[str]
    session_type_id: Optional[int]
    session_type_code: Optional[str]

    name: Optional[str]
    description: Optional[str]

    start_datetime: Optional[datetime.datetime]
    end_datetime: Optional[datetime.datetime]

    attend_url: Optional[str]
    attend_url_text: Optional[str]
    attend_url_passcode: Optional[str]
    attend_phone: Optional[str]
    attend_phone_passcode: Optional[str]
    attend_text: Optional[str]

    rehearsal_start_datetime: Optional[datetime.datetime]
    rehearsal_end_datetime: Optional[datetime.datetime]
    rehearsal_url: Optional[str]
    rehearsal_url_passcode: Optional[str]
    rehearsal_phone: Optional[str]
    rehearsal_phone_passcode: Optional[str]
    rehearsal_text: Optional[str]

    image_path: Optional[str]
    presentation_file_path: Optional[str]
    presentation_file_size: Optional[str]

    enable_event_file_approval_option: Optional[bool]

    file_count: Optional[int]

    status: Optional[int]
    review: Optional[bool]
    approve: Optional[bool]
    ready: Optional[bool]

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

    hide: Optional[bool]
    public: Optional[bool]
    public_hide: Optional[bool]
    hide_event_launcher: Optional[bool]

    priority: Optional[bool]
    sort: Optional[int]
    group: Optional[str]

    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.
    location_name: Optional[str] = Field(
        alias = 'event_location_name'
    )
    track_name: Optional[str] = Field(
        alias = 'event_track_name'
    )
    # Maybe add xxx in the future?

    # Including other related objects
    # event: Optional[Event_Base]
    event_abstract_list: Optional[list] # Optional[Event_Abstract_Base]
    event_badge_list: Optional[list] # Optional[Event_Abstract_Base]
    event_device_list: Optional[list] # Optional[Event_Device_Base]
    event_file_list: Optional[list] # Optional[Event_File_Base]
    event_location: Optional[Union[Event_Location_Base, None]]
    event_person_list: Optional[list]
    event_presenter_cat: Optional[Union[str, None]]
    event_presentation_list: Optional[list[Event_Presentation_Base]] # Optional[Event_Presentation_Base]
    event_presenter_list: Optional[list] # Optional[Event_Presenter_Base]
    event_track: Optional[Event_Track_Base]
    poc_event_person: Optional[Event_Person_Base] # NOTE: Using thi will probably create an import loop

    proposal_json: Optional[Union[Json, None]]

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

    #@validator('event_session_id_random', always=True)
    def event_session_id_random_copy(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values['id_random']:
            return values['id_random']
        return None

    @validator('id', always=True)
    def event_session_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values['id_random']:
            log.debug(values['id_random'])
            return redis_lookup_id_random(record_id_random=values['id_random'], table_name='event_session')
        return None

    @validator('event_id', always=True)
    def event_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values['event_id_random']:
            return redis_lookup_id_random(record_id_random=values['event_id_random'], table_name='event')
        return None

    @validator('event_location_id', always=True)
    def event_location_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values.get('event_location_id_random', None):
            return redis_lookup_id_random(record_id_random=values['event_location_id_random'], table_name='event_location')
        return None

    @validator('event_track_id', always=True)
    def event_track_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values['event_track_id_random']:
            return redis_lookup_id_random(record_id_random=values['event_track_id_random'], table_name='event_track')
        return None

    class Config:
        underscore_attrs_are_private = True
        allow_population_by_field_name = True
        fields = base_fields

#Event_Session_Base.update_forward_refs()