import datetime, pytz, time
from fastapi import APIRouter, Body, Depends, Header, HTTPException, Path, Query, Response, 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.organization_methods import create_update_organization_obj, load_organization_obj

from app.models.organization_models import Organization_Base
from app.models.response_models import Resp_Body_Base, mk_resp


router = APIRouter()


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

    obj_type = 'organization'
    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_organization_obj(
        obj: Organization_Base,
        obj_id: str = Path(min_length=11, max_length=22),
        x_account_id: Optional[str] = Header(..., ),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'organization'
    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


# ### BEGIN ### API Organization ### post_organization_json() ###
# Updated 2021-06-25
@router.post('/json', response_model=Resp_Body_Base)
async def post_organization_json(
        organization_obj: Organization_Base,
        # organization_id: str = Query(..., min_length=1, max_length=22),
        # create_sub_obj: bool = False,
        process_contact: bool = False,
        x_account_id: Optional[str] = Header(..., ),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        include: Optional[list] = [],
        exclude: Optional[list] = [],
        exclude_unset: Optional[bool] = True,
        exclude_none: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    # if organization_id := redis_lookup_id_random(record_id_random=organization_id, table_name='organization'): pass
    # else: return mk_resp(data=None, status_code=404)

    if organization_obj_in_result := create_update_organization_obj(
            organization_id = None,
            organization_obj = organization_obj,
            process_contact = process_contact,
            ):
        log.debug(organization_obj_in_result)
        if return_obj:
            organization_obj = load_organization_obj(organization_id=organization_obj_in_result)
            organization_dict = organization_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset)
            return mk_resp(data=organization_dict)
        else:
            return mk_resp(data=organization_obj_in_result)
    else:
        return mk_resp(data=False, status_code=400, response=response) # Bad Request
# ### END ### API Organization ### post_organization_json() ###


# ### BEGIN ### API Organization ### patch_organization_json() ###
# Updated 2021-06-25
@router.patch('/{organization_id}/json', response_model=Resp_Body_Base)
async def patch_organization_json(
        organization_obj: Organization_Base,
        organization_id: str = Path(min_length=11, max_length=22),
        process_contact: bool = False,
        x_account_id: Optional[str] = Header(..., ),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        include: Optional[list] = [],
        exclude: Optional[list] = [],
        exclude_unset: Optional[bool] = True,
        exclude_none: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

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

    if organization_obj_up_result := create_update_organization_obj(
            organization_id = organization_id,
            organization_obj = organization_obj,
            process_contact = process_contact,
            ):
        log.debug(organization_obj_up_result)
        if return_obj:
            organization_obj = load_organization_obj(organization_id=organization_id)
            organization_dict = organization_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset)
            return mk_resp(data=organization_dict)
        else:
            return mk_resp(data=organization_obj_up_result)
    else:
        return mk_resp(data=False, status_code=400, response=response) # Bad Request
# ### END ### API Organization ### patch_organization_json() ###


@router.get('/list', response_model=Resp_Body_Base)
async def get_organization_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,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'organization'
    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


@router.get('/{organization_id}', response_model=Resp_Body_Base)
async def get_organization_obj(
        organization_id: str = Path(min_length=11, max_length=22),
        inc_address: bool = False,
        inc_contact: bool = False,
        x_account_id: str = Header(...),
        by_alias: bool = True,
        exclude_unset: bool = True,
        response: Response = Response,
        ):
    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    organization_obj = load_organization_obj(
        organization_id=organization_id,
        inc_contact=inc_contact,
        inc_address=inc_address,
        ).dict(by_alias=by_alias, exclude_unset=exclude_unset)
    data = organization_obj
    return mk_resp(data=organization_obj)


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

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