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.membership_cfg_models import Membership_Cfg_Base
# from app.models.membership_type_models import Membership_Type_Base


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

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

    # membership_type_id_random: Optional[str]
    # membership_type_id: Optional[int]

    parent_membership_group_id_random: Optional[str]
    parent_membership_group_id: Optional[int]

    product_id_random: Optional[str]
    product_id: Optional[int] # The product they purchased to get this membership

    name: Optional[str]
    overview: Optional[str]
    description: Optional[str]

    expire_in_days: Optional[int]

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

    hide: Optional[bool]
    priority: Optional[bool]
    sort: Optional[int]
    group: Optional[str]

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

    # Including other related objects
    membership_cfg: Optional[Membership_Cfg_Base] = Field(
        alias = 'cfg',
        )
    membership_person_group_list: Optional[list] # Membership_Person_Group_Base
    product_list: Optional[list] # Product_Base() List of products (for_type and for_id) that get this membership option

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

    #@validator('membership_group_id_random', always=True)
    def membership_group_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 membership_group_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='membership_group')
        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('membership_type_id', always=True)
    # def membership_type_lookup(cls, v, values, **kwargs):
    #     log.setLevel(logging.WARNING)
    #     log.debug(locals())

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

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

        if values['parent_membership_group_id_random']:
            return redis_lookup_id_random(record_id_random=values['parent_membership_group_id_random'], table_name='membership_group')
        return None

    class Config:
        underscore_attrs_are_private = True
        fields = base_fields
        allow_population_by_field_name = True

# Membership_Group_Base.update_forward_refs()