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_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]

    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()