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 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
from app.models.site_domain_models import Site_Domain_Base


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

    # Including other related objects
    site_domain_list: Optional[list] # Site_Domain_Base()

    _processed_at: datetime.datetime = PrivateAttr(default_factory=datetime.datetime.now)

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