Files
OSIT-AE-API-FastAPI/app/models/archive_models.py
2023-09-07 16:57:58 -04:00

88 lines
2.7 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 Models ### Archive_Base() ###
class Archive_Base(BaseModel):
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
log.debug(locals())
id_random: Optional[str] = Field(
**base_fields['archive_id_random'],
alias = 'archive_id_random',
)
id: Optional[int] = Field(
alias = 'archive_id'
)
account_id_random: Optional[str]
account_id: Optional[int]
archive_type_id: Optional[int]
archive_type: Optional[str]
name: Optional[str]
description: Optional[str]
content_html: Optional[str]
content_json: Optional[str]
content_url: Optional[str]
content_url_text: Optional[str]
original_datetime: Optional[datetime.datetime] = None
original_datetime_timezone: Optional[str] = None
original_location: Optional[str] = None
original_address_id: Optional[int] = None
original_url: Optional[str]
original_url_text: Optional[str]
meta_data: Optional[str]
access_key: Optional[str]
sort_by: Optional[str]
sort_by_desc: Optional[bool]
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]
notes: Optional[str]
created_on: Optional[datetime.datetime] = None
updated_on: Optional[datetime.datetime] = None
# Including other related objects
archive_content_list: Optional[list] # Archive_Content_Base()
_processed_at: datetime.datetime = PrivateAttr(default_factory=datetime.datetime.now)
@validator('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('id_random'):
return redis_lookup_id_random(record_id_random=id_random, table_name='archive')
return None
@validator('account_id', always=True)
def account_id_lookup(cls, v, values, **kwargs):
if isinstance(v, int) and v > 0: return v
elif id_random := values.get('account_id_random'):
return redis_lookup_id_random(record_id_random=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 Models ### Archive_Base() ###