201 lines
7.9 KiB
Python
201 lines
7.9 KiB
Python
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.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: Handeling 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]
|
|
extension: Optional[str]
|
|
title: Optional[str]
|
|
description: Optional[str]
|
|
|
|
lu_file_purpose_id: Optional[int]
|
|
file_purpose: Optional[str]
|
|
|
|
#### NOTE ***** NOTE
|
|
internal_os: Optional[str]
|
|
public_use: Optional[bool] = False
|
|
public: Optional[bool]
|
|
approve: Optional[bool]
|
|
#### internal_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'
|
|
)
|
|
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'
|
|
)
|
|
|
|
# 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. Handeling 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 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() ###
|