OSIT-AE-API-FastAPI/app/models/event_person_model.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 ..db_sql import redis_lookup_id_random
from ..lib_general import *

from .common_field_schema import base_fields, default_num_bytes


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

    id_random: Optional[str] = Field(
        **base_fields['event_person_id_random'],
        alias='event_person_id_random',
        default_factory=lambda:secrets.token_urlsafe(default_num_bytes),
    )
    id: Optional[int] = Field(
        #alias='event_person_id'
    )
    account_id_random: Optional[str]
    account_id: Optional[int]
    event_id_random: Optional[str]
    event_id: Optional[int]

    person_given_name: Optional[str]
    person_family_name: Optional[str]
    person_full_name: Optional[str]

    organization_name: Optional[str]

    email: Optional[str]

    #new_password: str = Field(default_factory=lambda:secrets.token_urlsafe(default_num_bytes))
    #password: Optional[str]
    new_password: Optional[str]

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

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

    #@validator('event_person_id_random', always=True)
    def event_person_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_person_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_person')
        return None

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

        if values['account_id_random']:
            return redis_lookup_id_random(record_id_random=values['account_id_random'], table_name='account')
        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('password', always=True)
    # def hash_new_password(cls, v, values, **kwargs):
    #     log.setLevel(logging.WARNING)
    #     log.debug(locals())

    #     if values['new_password']:
    #         return secure_hash_string(string=values['new_password'])
    #     return None

    class Config:
        underscore_attrs_are_private = True
        fields = base_fields


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

    id_random: Optional[str] = Field(
        **base_fields['event_person_id_random'],
        alias='event_person_id_random',
        default_factory=lambda:secrets.token_urlsafe(default_num_bytes),
    )
    id: Optional[int] = Field(
        #alias='event_person_id'
    )
    account_id_random: Optional[str]
    account_id: Optional[int]
    event_id_random: Optional[str]
    event_id: Optional[int]
    event_registration_id_random: Optional[str]
    event_registration_id: Optional[int]
    event_badge_id_random: Optional[str]
    event_badge_id: Optional[int]
    person_id_random: Optional[str]
    person_id: Optional[int]
    user_id_random: Optional[str]
    user_id: Optional[int]

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

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

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

    #@validator('event_person_id_random', always=True)
    def event_person_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_person_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_person')
        return None

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

        if values['account_id_random']:
            return redis_lookup_id_random(record_id_random=values['account_id_random'], table_name='account')
        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_registration_id', always=True)
    def event_registration_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values['event_registration_id_random']:
            return redis_lookup_id_random(record_id_random=values['event_registration_id_random'], table_name='event_registration')
        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 values['event_badge_id_random']:
            return redis_lookup_id_random(record_id_random=values['event_badge_id_random'], table_name='event_badge')
        return None

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

        if values['person_id_random']:
            return redis_lookup_id_random(record_id_random=values['person_id_random'], table_name='person')
        return None

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

        if values['user_id_random']:
            return redis_lookup_id_random(record_id_random=values['user_id_random'], table_name='user')
        return None

    class Config:
        underscore_attrs_are_private = True
        fields = base_fields