import datetime
from fastapi import APIRouter, Body, Depends, Header, HTTPException, Query, status
from pydantic import BaseModel, EmailStr, Field
from typing import Dict, List, Optional, Set, Union

from app.lib_general import log, logging
from app.config import settings
from app.db_sql import sql_insert, sql_update, sql_insert_or_update, sql_select, sql_delete, redis_lookup_id_random

from app.routers.api_crud import delete_obj_template, get_obj_template, get_obj_li_template, patch_obj_template, post_obj_template

from app.methods.membership_methods import load_membership_obj

from app.models.membership_models import Membership_Base
from app.models.response_models import Resp_Body_Base, mk_resp


router = APIRouter()


@router.post('', response_model=Resp_Body_Base)
async def post_membership_obj(
        obj: Membership_Base,
        x_account_id: str = Header(...),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'membership'
    obj_data_dict = obj.dict(by_alias=False, exclude_unset=True)
    result = post_obj_template(
        obj_type=obj_type,
        data=obj_data_dict,
        return_obj=True,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.patch('/{obj_id}', response_model=Resp_Body_Base)
async def patch_membership_obj(
        obj_id: str = Query(..., min_length=1, max_length=22),
        obj: Membership_Base = None,
        #x_account_id: Optional[str] = Header(..., ),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        ):
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'membership'
    obj_data_dict = obj.dict(by_alias=False, exclude_unset=True)
    obj_data_dict['id'] = redis_lookup_id_random(record_id_random=obj_id, table_name=obj_type)
    obj_data_dict['id_random'] = obj_id
    result = patch_obj_template(
        obj_type=obj_type,
        data=obj_data_dict,
        obj_id=obj_id,
        return_obj=True,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.get('/list', response_model=Resp_Body_Base)
async def get_membership_obj_li(
        for_obj_type: Optional[str] = Query(None, min_length=2, max_length=50),
        for_obj_id: Optional[str] = Query(None, min_length=1, max_length=22),
        x_account_id: str = Header(...),
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'membership'
    result = get_obj_li_template(
        obj_type=obj_type,
        for_obj_type=for_obj_type,
        for_obj_id=for_obj_id,
        by_alias=True,
        exclude_unset=True,
        )
    return result


# Look up is only for account, person, or user records
@router.get('/lookup', response_model=Resp_Body_Base)
async def lookup_membership_obj(
        for_obj_id: Union[int,str],
        for_obj_type: str = Query(..., min_length=2, max_length=50),
        x_account_id: str = Header(...),
        inc_membership_profile: bool = True,
        inc_membership_cfg: bool = True,
        inc_extended_profile: bool = True,
        inc_person: bool = True,
        inc_user: bool = True,
        by_alias: bool = True,
        exclude_unset: bool = True,
        ):
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'membership'
    base_name = Membership_Base

    if for_obj_id := redis_lookup_id_random(record_id_random=for_obj_id, table_name=for_obj_type): pass
    else: return mk_resp(data=False, status_code=404) # Not Found
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL

    data = {}
    as_list = False
    if for_obj_type == 'account' and for_obj_id:
        data['account_id'] = for_obj_id
        sql_where_for_obj_type = """`membership`.account_id = :account_id"""
        sql_limit = ''
        as_list = True
    elif for_obj_type == 'person' and for_obj_id:
        data['person_id'] = for_obj_id
        sql_where_for_obj_type = """`membership`.person_id = :person_id"""
        sql_limit = 'LIMIT 1'
    elif for_obj_type == 'user' and for_obj_id:
        data['user_id'] = for_obj_id
        sql_where_for_obj_type = """`membership`.user_id = :user_id"""
        sql_limit = 'LIMIT 1'
    elif for_obj_type == 'membership' and for_obj_id:
        data['membership_id'] = for_obj_id
        sql_where_for_obj_type = """`membership`.id = :membership_id"""
        sql_limit = 'LIMIT 1'
    else:
        log.debug(f'Object type={for_obj_type}; Object ID={for_obj_id}')
        return mk_resp(data=False, status_code=400) # Bad Request

    sql = f"""
        SELECT id AS 'membership_id', id_random AS 'membership_id_random'
        FROM `membership` AS `membership`
        WHERE {sql_where_for_obj_type}
        {sql_limit}
        """

    # This will return a list if selecting by account ID
    membership_obj_result = sql_select(data=data, sql=sql, as_list=as_list)
    if isinstance(membership_obj_result, dict):
        membership_id = membership_obj_result.get('membership_id', None)
        membership_obj = load_membership_obj(
            membership_id=membership_id,
            inc_membership_profile=inc_membership_profile,
            inc_membership_cfg=inc_membership_cfg,
            inc_extended_profile=inc_extended_profile,
            inc_person=inc_person,
            inc_user=inc_user,
            ).dict(by_alias=by_alias, exclude_unset=exclude_unset)
        data = membership_obj
    elif isinstance(membership_obj_result, list):
        membership_obj_li = []
        for membership_obj in membership_obj_result:
            membership_id = membership_obj.get('membership_id', None)
            membership_obj_li.append(
                load_membership_obj(
                    membership_id=membership_id,
                    inc_membership_profile=inc_membership_profile,
                    inc_membership_cfg=inc_membership_cfg,
                    inc_extended_profile=inc_extended_profile,
                    inc_person=inc_person,
                    inc_user=inc_user,
                    ).dict(by_alias=by_alias, exclude_unset=exclude_unset)
                )
        data = membership_obj_li
    else:
        log.debug(membership_obj_result)
        return mk_resp(data=None, status_code=404) # Not Found
    return mk_resp(data=data)



# ### BEGIN ### API Membership ### get_membership_obj() ###
# Working well as of 2021-06-04. Using as a template for other routes.
@router.get('/{membership_id}', response_model=Resp_Body_Base)
async def get_membership_obj(
        membership_id: str = Query(..., min_length=1, max_length=22),
        enabled: str = 'enabled', # For now this covers any included objects or object lists
        limit: int = 500, # For now this covers any included objects or object lists
        inc_address: bool = False, # Per member
        inc_contact: bool = False, # Per member
        inc_membership_cfg: bool = False,
        inc_membership_group_list: bool = False, # Per member
        inc_membership_member_list: bool = False,
        inc_membership_profile: bool = False, # Per member
        inc_organization: bool = False, # Per member
        inc_person: bool = False, # Per member
        inc_product: bool = False, # Per member
        inc_product_list: bool = False,
        inc_user: bool = False, # Per member
        x_account_id: str = Header(...),
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        ):
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if membership_id := redis_lookup_id_random(record_id_random=membership_id, table_name='membership'): pass
    else:
        return mk_resp(data=None, status_code=404)

    if membership_obj := load_membership_obj(
        membership_id = membership_id,
        enabled = enabled,
        # model_as_dict = model_as_dict,
        limit = limit,
        inc_address = inc_address,
        inc_contact = inc_contact,
        inc_membership_cfg = inc_membership_cfg,
        inc_membership_group_list = inc_membership_group_list,
        inc_membership_member_list = inc_membership_member_list,
        inc_membership_profile = inc_membership_profile,
        inc_organization = inc_organization,
        inc_person = inc_person,
        inc_product = inc_product,
        inc_product_list = inc_product_list,
        inc_user = inc_user,
        ):
        # membership_dict = membership_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset)
        pass
    else:
        return mk_resp(data=False, status_code=400) # Bad Request

    return mk_resp(data=membership_obj)
# ### END ### API Membership ### get_membership_obj() ###


@router.get('/{obj_id}', response_model=Resp_Body_Base)
async def get_membership_obj_old(
        obj_id: str = Query(..., min_length=1, max_length=22),
        x_account_id: str = Header(...),
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'membership'
    result = get_obj_template(
        obj_type=obj_type,
        obj_id=obj_id,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.delete('/{obj_id}', response_model=Resp_Body_Base)
async def delete_membership_obj(
        obj_id: str = Query(..., min_length=1, max_length=22),
        x_account_id: str = Header(...),
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'membership'
    result = delete_obj_template(
        obj_type=obj_type,
        obj_id=obj_id,
        )
    return result