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.account_models import Account_Base


class Address_Base(BaseModel):
    log.setLevel(logging.DEBUG) # 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]

    contact_id_random: Optional[str]
    contact_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]
    country_subdivision_name: Optional[str] # From country subdivision lookup table
    state_province: Optional[str] # Avoid using
    postal_code: Optional[str]
    country_alpha_2_code: Optional[str]
    country_name: Optional[str] # From country lookup table
    country: Optional[str] # Avoid using

    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

    _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.get('id_random', None): # 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('contact_id', always=True)
    def contact_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

        if values.get('contact_id_random', None): # values['contact_id_random']:
            return redis_lookup_id_random(record_id_random=values['contact_id_random'], table_name='contact')
        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
        allow_population_by_field_name = True
        fields = base_fields

#Address_Base.update_forward_refs()