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 ..lib_general import *

from app.config import settings

from .common_field_schema import base_fields, default_num_bytes


# ### BEGIN ### API Response Model ### Resp_Body_Base() ###
# The pydantic BaseModel to help make consistent REST responses - STI 2021-03-05
class Resp_Body_Base(BaseModel):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    data: Union[dict, list]
    meta: Optional[dict]
# ### END ### API Response Model ### Resp_Body_Base() ###


# ### BEGIN ### API Response Model ### mk_resp() ###
# The make response function for REST - STI 2021-03-17
def mk_resp(data:None|bool|dict|list, dict_to_json:bool=False, status_code:int=200, status_message:str='', status_name:str='', success:bool=True, details:str='', by_alias:bool=True, exclude_unset:bool=True):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if data is None: data = { 'result': data }
    elif data == False: data = { 'result': data }
    elif data == True: data = { 'result': data }

    resp_body = {}
    resp_body['data'] = data
    resp_body['meta'] = {}
    resp_body['meta']['details'] = details
    resp_body['meta']['status_code'] = status_code
    resp_body['meta']['status_message'] = settings.HTTP_STATUS_LI[status_code]['message']
    resp_body['meta']['status_name'] = settings.HTTP_STATUS_LI[status_code]['name']
    resp_body['meta']['success'] = success

    if isinstance(data, bool):
        resp_body['meta']['data_type'] = 'bool'
    elif isinstance(data, int):
        resp_body['meta']['data_type'] = 'int'
    elif isinstance(data, str):
        resp_body['meta']['data_type'] = 'str'
    elif isinstance(data, dict):
        resp_body['meta']['data_type'] = 'dict'
    elif isinstance(data, list):
        resp_body['meta']['data_type'] = 'list'
        resp_body['meta']['data_list_count'] = len(data)

    log.debug(type(resp_body['data']))

    resp_body = Resp_Body_Base(**resp_body).dict(by_alias=by_alias, exclude_unset=exclude_unset)

    return resp_body
# ### END ### API Response Model ### mk_resp() ###