Files
OSIT-AE-API-FastAPI/app/models/site_model.py

116 lines
3.3 KiB
Python

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 ..lib_general import *
from ..log import *
from .common_field_schema import base_fields, default_num_bytes
class Site_Base(BaseModel):
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
log.debug(locals())
id_random: Optional[str] = Field(
**base_fields['site_id_random'],
alias='site_id_random',
default_factory=lambda:secrets.token_urlsafe(default_num_bytes),
)
id: Optional[int] = Field(
#alias='site_id'
)
account_id_random: Optional[str]
account_id: Optional[int]
code: Optional[str]
# fqdn_1: Optional[str]
# fqdn_2: Optional[str]
# fqdn_3: Optional[str]
name: Optional[str]
description: Optional[str]
restrict_access: Optional[bool]
access_key: Optional[str]
enable: Optional[bool]
enable_from: Optional[datetime.datetime] = None
enable_to: Optional[datetime.datetime] = None
path_logo: Optional[str]
logo_bg_color: Optional[str]
path_banner_image: Optional[str]
banner_bg_color: Optional[str]
path_background_image: Optional[str]
background_bg_color: Optional[str]
path_html_menu: Optional[str]
title: Optional[str]
html_menu: Optional[str]
html_header: Optional[str]
html_header_h1: Optional[str]
html_header_h2: Optional[str]
html_banner: Optional[str]
html_root_body: Optional[str]
html_tagline: Optional[str]
logo_filename: Optional[str]
banner_image_filename: Optional[str]
banner_html: Optional[str]
site_title: Optional[str]
site_menu_html_path: Optional[str]
site_header_h1: Optional[str]
site_header_h2: Optional[str]
site_body: Optional[str]
site_tagline: Optional[str]
style_href: Optional[str]
script_src: Optional[str]
google_tracking_id: 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('site_id_random', always=True)
def site_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 site_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='site')
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
class Config:
underscore_attrs_are_private = True
fields = base_fields
Site_Base.update_forward_refs()