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

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 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_badge_models import Event_Badge_Base
from app.models.event_person_models import Event_Person_Base


# ### BEGIN ### API Event Exhibit Tracking Models ### Event_Exhibit_Base() ###
class Event_Exhibit_Tracking_Base(BaseModel):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    id_random: Optional[str] = Field(
        **base_fields['event_exhibit_tracking_id_random'],
        alias = 'event_exhibit_tracking_id_random',
    )
    id: Optional[int] = Field(
        alias = 'event_exhibit_tracking_id'
    )

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

    event_exhibit_id_random: Optional[str]
    event_exhibit_id: Optional[int]

    event_person_id_random: Optional[str]
    event_person_id: Optional[int]

    event_badge_id_random: Optional[str]
    event_badge_id: Optional[int]

    exhibitor_notes: Optional[str]
    # json_data: Optional[str]
    json_data: Optional[Json]

    enable: Optional[bool]
    hide: Optional[bool]
    priority: Optional[bool] = False
    sort: Optional[Union[None, int]] = None
    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.
    event_badge_informal_name: Optional[str]
    event_badge_title_names: Optional[str]
    event_badge_given_name: Optional[str]
    event_badge_middle_name: Optional[str]
    event_badge_family_name: Optional[str]
    event_badge_full_name: Optional[str]
    event_badge_full_name_override: Optional[str]

    event_badge_designations: Optional[str]
    event_badge_professional_title: Optional[str]
    event_badge_professional_title_override: Optional[str]
    event_badge_affiliations: Optional[str]
    event_badge_affiliations_override: Optional[str]

    event_badge_email: Optional[str]

    # event_badge_address_line_1: Optional[str]
    # event_badge_address_line_2: Optional[str]
    # event_badge_address_line_3: Optional[str]

    # event_badge_city: Optional[str]

    # event_badge_country_subdivision_code: Optional[str]
    # event_badge_state_province_abb: Optional[str]
    # event_badge_state_province: Optional[str]

    # event_badge_postal_code: Optional[str]

    event_badge_country_alpha_2_code: Optional[str]
    event_badge_country: Optional[str]

    # event_badge_location: Optional[str]
    event_badge_location_override: Optional[str]

    event_person_informal_name: Optional[str]
    event_person_given_name: Optional[str]
    event_person_family_name: Optional[str]
    event_person_full_name: Optional[str]
    event_person_full_name_override: Optional[str]
    event_person_affiliations: Optional[str]
    event_person_email: Optional[str]

    event_exhibit_name: Optional[str]

    # Including other related objects
    # event_badge: Optional[Union[Event_Badge_Base, None]]
    event_person: Optional[Union[Event_Person_Base, None]]

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

    #@validator('event_exhibit_tracking_id_random', always=True)
    def event_exhibit_tracking_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_exhibit_tracking_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        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_exhibit_tracking')
        return None

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

        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):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        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_person_id', always=True)
    def event_person_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_person_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_person')
        return None

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

        if isinstance(v, int) and v > 0: return v
        elif id_random := values.get('event_badge_id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='event_badge')
        return None

    class Config:
        underscore_attrs_are_private = True
        allow_population_by_field_name = True
        fields = base_fields
# ### END ### API Event Exhibit Tracking Models ### Event_Exhibit_Base() ###