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.order_cfg_models import Order_Cfg_Base
from app.models.order_line_models import Order_Line_Base
from app.models.person_models import Person_Base


class Order_Base(BaseModel):
    log.setLevel(logging.INFO)
    log.debug(locals())

    id_random: Optional[str] = Field(
        **base_fields['order_id_random'],
        alias = 'order_id_random',
    )
    id: Optional[int] = Field(
        alias = 'order_id'
    )

    account_id_random: Optional[str]
    account_id: Optional[int] # NOTE: This is not really optional

    person_id_random: Optional[str]
    person_id: Optional[int]

    total_quantity: Optional[int] = Field(0, ge=0, lt=150)
    total_bill: Optional[int] = Field(0, ge=0, lt=1500000) # NOTE: This is total_amount in the order_cart
    total_paid: Optional[int] = Field(0, ge=0, lt=1500000)
    balance: Optional[int] = Field(0, ge=-1500000, lt=1500000) # Balance needs to be calculated
    status: Optional[str] # open, locked, reopened, closed, canceled, other
    # open = building or creating cart, payment failed and no partial payment already processed
    # locked = processing payment, partial payment processed, admin lock? (no changes allowed)
    # closed = successful payment(s) with no balance, complete (no changes allowed)
    # canceled = order canceled; something odd happened (no changes allowed)
    # other = other reasons (no changes allowed)
    checkout_status: Optional[str] # canceled, waiting, success, failed, unknown
    # none = no checkout attempted
    # canceled = checkout attempted but canceled for some reason
    # waiting = the payment is processing or awaiting some type of authorization
    # success = payment successful (their may still be a balance)
    # failed = the payment failed for some reason
    # unknown = unknown response...

    notes: Optional[str]

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

    # Including other related objects
    cfg: Optional[Order_Cfg_Base]
    order_line_list: Optional[list[Order_Line_Base]]
    person: Optional[Person_Base]

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

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

        if id_random := values.get('id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='order')
        elif v:
            return get_id_random(v, table_name='order')
        return None

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

        if id_random := values.get('id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='account')
        elif v:
            return get_id_random(v, table_name='account')
        return None


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

        if id_random := values.get('id_random'):
            return redis_lookup_id_random(record_id_random=id_random, table_name='person')
        elif v:
            return get_id_random(v, table_name='person')
        return None

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


class Order_DB_Base(BaseModel):
    log.setLevel(logging.INFO)
    log.debug(locals())

    account_id_random: Optional[str]
    account_id: Optional[int] # NOTE: This is not really optional

    person_id_random: Union[str, None] = None
    person_id: Union[int, None] = None

    user_id_random: Union[str, None] = None
    user_id: Union[int, None] = None

    total_quantity: Optional[int] = Field(0, ge=0, lt=150)
    total_bill: Optional[int] = Field(0, ge=0, lt=1500000) # NOTE: This is total_amount in the order_cart
    total_paid: Optional[int] = Field(0, ge=0, lt=1500000)
    balance: Optional[int] = Field(0, ge=-1500000, lt=1500000) # Balance needs to be calculated
    status: Optional[str] = 'open' # open, locked, reopened, closed, canceled, other
    # open = building or creating cart, payment failed and no partial payment already processed
    # locked = processing payment, partial payment processed, admin lock? (no changes allowed)
    # closed = successful payment(s) with no balance, complete (no changes allowed)
    # canceled = order canceled; something odd happened (no changes allowed)
    # other = other reasons (no changes allowed)
    checkout_status: Union[str, None] = None # canceled, waiting, success, failed, unknown
    # none = no checkout attempted
    # canceled = checkout attempted but canceled for some reason
    # waiting = the payment is processing or awaiting some type of authorization
    # success = payment successful (their may still be a balance)
    # failed = the payment failed for some reason
    # unknown = unknown response...

    notes: Optional[str]

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

    @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('person_id', always=True)
    def person_id_lookup(cls, v, values, **kwargs):
        log.setLevel(logging.WARNING)
        log.debug(locals())

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

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