Files
OSIT-AE-API-FastAPI/app/models/archive_content_models.py
2022-08-10 17:33:46 -04:00

141 lines
4.5 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 redis_lookup_id_random
from app.lib_general import log, logging
from app.models.common_field_schema import base_fields, default_num_bytes
# ### BEGIN ### API Archive Content Models ### Archive_Content_Base() ###
class Archive_Content_Base(BaseModel):
log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
log.debug(locals())
def testing(test_var=None):
log.debug(test_var)
return test_var
id_random: Optional[str] = Field(
# **base_fields['archive_content_id_random'],
alias = 'archive_content_id_random',
)
id: Optional[int] = Field(
alias = 'archive_content_id'
)
# account_id_random: Optional[str] # Is this field really needed?
# account_id: Optional[int] # Is this field really needed?
archive_id_random: Optional[str]
archive_id: Optional[int]
archive_content_type_id: Optional[int]
archive_content_type: Optional[str]
lu_media_type_id: Optional[int]
lu_media_type: Optional[str]
name: Optional[str]
description: Optional[str]
content_html: Optional[str]
content_json: Optional[str]
url: Optional[str]
url_text: Optional[str]
hosted_file_id_random: Optional[str]
hosted_file_id: Optional[int]
file_path: Optional[str]
filename: Optional[str]
file_extension: Optional[str]
# xxxx_red: str = Field(default='xxx')
# xxxx_blue: str = Field(default_factory=testing)
hosted_file_path: str = None # '/testing/test-test'
wtf: str = 'HELLO???'
original_datetime: Optional[datetime.datetime]
original_datetime_timezone: Optional[str]
original_location: Optional[str]
original_address_id: Optional[int]
original_url: Optional[str]
original_url_text: Optional[str]
meta_data: Optional[str]
access_key: Optional[str]
enable: Optional[bool]
enable_from: Optional[datetime.datetime]
enable_to: Optional[datetime.datetime]
priority: Optional[bool]
sort: Optional[int]
group: Optional[str]
notes: Optional[str]
created_on: Optional[datetime.datetime]
updated_on: Optional[datetime.datetime]
_processed_at: datetime.datetime = PrivateAttr(default_factory=datetime.datetime.now)
@validator('id', always=True)
def archive_content_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='archive_content')
return None
@validator('archive_id', always=True)
def archive_id_lookup(cls, v, values, **kwargs):
if isinstance(v, int) and v > 0: return v
elif id_random := values.get('archive_id_random'):
return redis_lookup_id_random(record_id_random=id_random, table_name='archive')
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('hosted_file_path', always=True)
def hosted_file_path_lookup(cls, v, values, **kwargs):
log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
log.debug(v)
log.debug(values)
if hosted_file_id_random := values.get('hosted_file_id_random'):
log.debug('Found hosted_file_id_random...')
path_str = f'/hosted_file/download/{hosted_file_id_random}'
if filename := values.get('filename'):
path_str = f'{path_str}?filename={filename}'
log.debug(path_str)
return path_str
log.debug('NOT Found hosted_file_id_random...')
return v
# @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
class Config:
underscore_attrs_are_private = True
allow_population_by_field_name = True
fields = base_fields
# ### END ### API Archive Content Models ### Archive_Content_Base() ###