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 ..db_sql import redis_lookup_id_random from ..lib_general import * from .common_field_schema import base_fields, default_num_bytes from .site_domain_models 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] 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 logo_path: Optional[str] logo_bg_color: Optional[str] banner_image_path: Optional[str] banner_bg_color: Optional[str] background_image_path: Optional[str] background_bg_color: Optional[str] html_menu_path: 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()