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

119 lines
3.5 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
from .site_domain_model import Site_Domain_Base as Site_Domain_Base_2
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] = None
html_menu: Optional[str] = None
html_header: Optional[str] = None
html_header_h1: Optional[str] = None
html_header_h2: Optional[str] = None
html_banner: Optional[str] = None
html_root_body: Optional[str] = None
html_tagline: Optional[str] = None
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
site_domain: Optional[Site_Domain_Base_2] = Site_Domain_Base_2()
_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()