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 .user_models import User_Base


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

    id_random: Optional[str] = Field(
        **base_fields['user_role_id_random'],
        alias='user_role_id_random',
        default_factory=lambda:secrets.token_urlsafe(default_num_bytes),
    )
    id: Optional[int] = Field(
        #alias='user_role_id'
    )
    user_id_random: Optional[str] # NOTE: This needs to be reviewed. Current a user_role record does not have an id_random
    user_id: Optional[int]
    #role_id_random: Optional[str]
    role_id: Optional[int]

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

    enable: Optional[bool]
    enable_from: Optional[datetime.datetime] = None
    enable_to: Optional[datetime.datetime] = None

    notes: 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('user_role_id_random', always=True)
    def user_role_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 user_role_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='user_role')
        return None

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

        if values['user_id_random']:
            return redis_lookup_id_random(record_id_random=values['user_id_random'], table_name='user')
        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

#User_Role_Base.update_forward_refs()