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 log, logging

from .common_field_schema import base_fields, default_num_bytes
#from .account_models import Account_Base


class Address_Base(BaseModel):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    id_random: Optional[str] = Field(
        **base_fields['address_id_random'],
        alias='address_id_random',
        default_factory=lambda:secrets.token_urlsafe(default_num_bytes),
    )
    id: Optional[int] = Field(
        #alias='address_id'
    )
    account_id_random: Optional[str]
    account_id: Optional[int]

    for_type: Optional[str]
    for_id_random: Optional[str]
    for_id: Optional[int]

    #organization: Optional[Organization_Base] = Organization_Base()

    name: Optional[str]
    attention_to: Optional[str]
    organization_name: Optional[str]

    line_1: Optional[str]
    line_2: Optional[str]
    line_3: Optional[str]
    city: Optional[str]
    country_subdivision_code: Optional[str]
    state_province: Optional[str]
    postal_code: Optional[str]
    country_alpha_2_code: Optional[str]
    country: Optional[str]

    lu_time_zone_id: Optional[str]
    timezone: Optional[str]

    latitude: Optional[str]
    longitude: Optional[str]

    map_url: Optional[str]

    congressional_district: Optional[str]

    #priority: Optional[int]
    #sort: Optional[int]
    #group: Optional[str]

    created_on: Optional[datetime.datetime] = None
    updated_on: Optional[datetime.datetime] = None

    #account: Optional[Account_Base] = Account_Base()

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

    #@validator('address_id_random', always=True)
    def address_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 address_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='address')
        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

    #@validator('organization_id', always=True)
    #def organization_id_lookup(cls, v, values, **kwargs):
        #log.setLevel(logging.WARNING)
        #log.debug(locals())

        #if values['organization_id']:
            #return redis_lookup_id_random(record_id_random=values['organization_id'], table_name='organization')
        #return None

    @validator('for_id', always=True)
    def for_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values['for_id_random'] and values['for_type']:
            return redis_lookup_id_random(record_id_random=values['for_id_random'], table_name=values['for_type'])
        return None

    class Config:
        underscore_attrs_are_private = True
        fields = base_fields

#Address_Base.update_forward_refs()