OSIT-AE-API-FastAPI/app/routers/order.py

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

from app.lib_general import *
from app.config import settings
from app.db_sql import *

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.order_methods import get_order_rec_list, load_order_obj, save_order_obj

from app.models.response_models import *
from app.models.order_models import Order_Base, Order_DB_Base


router = APIRouter()


# NOTE 2021-08-09: Use with rework of order_cart
# account_id is required, person_id is ideal
@router.post('/order', response_model=Resp_Body_Base)
async def post_order_obj(
        order_obj: Order_DB_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 = 'order'
    order_obj = order_obj.dict(by_alias=False, exclude_unset=False)
    result = post_obj_template(
        obj_type = obj_type,
        data = order_obj,
        return_obj = True,
        by_alias = True,
        exclude_unset = True,
        )
    return result


# NOTE 2021-08-09: Use with rework of order_cart
@router.patch('/order/{obj_id}', response_model=Resp_Body_Base)
async def patch_order_obj(
        obj_id: str = Query(..., min_length=1, max_length=22),
        obj: Order_Base = None,
        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 = 'order'
    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



# NOTE 2021-08-09: Use with rework of order_cart
# NOTE: The router needs to have the prefix (/order) removed.
@router.patch('/order/{order_id}/line/add', response_model=Resp_Body_Base)
async def patch_order_obj_line_add(
        order_id: str = Query(..., min_length=1, max_length=22),
        obj: Order_Base = None,
        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())



@router.get('/order/list', response_model=Resp_Body_Base)
async def get_order_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),
        order_status: Optional[str] = 'complete',
        from_datetime: Optional[datetime.datetime] = None,
        to_datetime: Optional[datetime.datetime] = None,
        limit: Optional[int] = None,
        inc_order_line_list: Optional[bool] = False,
        inc_order_cfg: Optional[bool] = False,
        inc_person_obj: Optional[bool] = False,
        inc_user_obj: Optional[bool] = False,
        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())

    data = {}
    if for_obj_type == 'account' and for_obj_id:
        data['account_id'] = redis_lookup_id_random(record_id_random=for_obj_id, table_name='account')
        sql_account_person_user = f"""`order`.account_id = :account_id"""
    elif for_obj_type == 'person' and for_obj_id:
        data['person_id'] = redis_lookup_id_random(record_id_random=for_obj_id, table_name='person')
        sql_account_person_user = f"""`order`.person_id = :person_id"""
    elif for_obj_type == 'user' and for_obj_id:
        data['user_id'] = redis_lookup_id_random(record_id_random=for_obj_id, table_name='user')
        sql_account_person_user = f"""`order`.user_id = :user_id"""
    else:
        sql_account_person_user = ''

    if order_status in ['canceled', 'complete', 'in progress', 'started']:
        data['status'] = order_status
        sql_status = f"""AND `order`.status = :status"""
    else:
        sql_status = ''

    if from_datetime and to_datetime:
        data['from_datetime'] = from_datetime
        data['to_datetime'] = to_datetime
        sql_from_to_datetime = f"""AND `order`.created_on >= :from_datetime AND `order`.created_on <= :to_datetime"""
    elif from_datetime:
        data['from_datetime'] = from_datetime
        sql_from_to_datetime = f"""AND `order`.created_on >= :from_datetime"""
    elif to_datetime:
        data['to_datetime'] = to_datetime
        sql_from_to_datetime = f"""AND `order`.created_on <= :to_datetime"""
    else:
        sql_from_to_datetime = ''

    if limit:
        data['limit'] = limit
        sql_limit = f"""LIMIT :limit"""
    else:
        sql_limit = ''

    sql = f"""
        SELECT `order`.id AS 'order_id', `order`.id_random AS 'order_id_random'
        FROM `order` AS `order`
        WHERE {sql_account_person_user}
        {sql_status}
        {sql_from_to_datetime}
        ORDER BY `order`.created_on DESC, `order`.updated_on DESC
        {sql_limit};
        """

    if order_obj_li_result := sql_select(data=data, sql=sql, as_list=True):
        order_obj_li = []
        for order_obj in order_obj_li_result:
            order_id = order_obj.get('order_id', None)
            if order_obj := load_order_obj(order_id=order_id, inc_order_line_list=inc_order_line_list, inc_order_cfg=inc_order_cfg, inc_person_obj=inc_person_obj):
                data = order_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset)
                order_obj_li.append(data)
        return mk_resp(data=order_obj_li)
    else:
        log.debug(order_obj_li_result)
        return mk_resp(data=False, status_code=404, response=response)


# ### BEGIN ### API Order Routes ### get_order_obj() ###
# NOTE 2021-08-09: Use with rework of order_cart
# Updated 2021-08-07
@router.get('/order/{order_id}', response_model=Resp_Body_Base)
async def get_order_obj(
        order_id: str = Query(..., min_length=11, max_length=22),
        limit: int = 500,
        enabled: str = 'enabled',
        inc_order_line_list: bool = False,
        inc_order_cfg: bool = False,
        inc_person: bool = False,
        x_account_id: str = Header(...),
        by_alias: bool = True,
        exclude_unset: bool = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    if order_obj := load_order_obj(
            order_id = order_id,
            limit = limit,
            enabled = enabled,
            by_alias = by_alias,
            exclude_unset = exclude_unset,
            # model_as_dict = model_as_dict,
            inc_order_line_list = inc_order_line_list,
            inc_order_cfg = inc_order_cfg,
            inc_person = inc_person,
            ):
        log.debug(order_obj)
        data = order_obj.dict(by_alias=by_alias, exclude_unset=False) # NOTE NOTE NOTE NOTE exclude_unset is forced to False for now. Will return more fields than is ideal. Need to create another Order_Line_Base. Probably Order_Line_OUT_Base
        return mk_resp(data=data)
    else:
        return mk_resp(data=False, status_code=404, response=response) # Not Found
# ### END ### API Order Routes ### get_order_obj() ###


# ### BEGIN ### API Order ### get_account_obj_order_list() ###
# Updated 2021-11-19
@router.get('/account/{account_id}/order/list', response_model=Resp_Body_Base)
async def get_account_id_order_list(
        account_id: str = Query(..., min_length=11, max_length=22),
        limit: int = 50,
        enabled: str = 'enabled',
        from_datetime: datetime.datetime = None,
        to_datetime: datetime.datetime = None,
        inc_order_cfg: bool = False,
        inc_order_line_list: bool = False,
        # inc_order_w_prod_type: str = 'all', # all, membership, fundraising, event, etc
        status: str = 'closed', # open, locked, reopened?, closed (complete), canceled, other
        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())

    if account_id := redis_lookup_id_random(record_id_random=account_id, table_name='account'): pass
    else: return mk_resp(data=None, status_code=404, response=response)

    # Updated 2021-06-28
    if order_rec_list_result := get_order_rec_list(
            for_obj_type = 'account',
            for_obj_id = account_id,
            limit = limit,
            enabled = enabled,
            from_datetime = from_datetime,
            to_datetime = to_datetime,
            status = status,
            ):
        order_result_list = []
        for order_rec in order_rec_list_result:
            if load_order_result := load_order_obj(
                    order_id = order_rec.get('order_id', None),
                    limit = limit,
                    enabled = enabled,
                    by_alias = by_alias,
                    exclude_unset = exclude_unset,
                    # model_as_dict = model_as_dict,
                    inc_order_cfg = inc_order_cfg,
                    inc_order_line_list = inc_order_line_list,
                    ):
                order_result_list.append(load_order_result)
            else:
                order_result_list.append(None)
        response_data = order_result_list
    else:
        return mk_resp(data=False, status_code=400, response=response) # Bad Request

    return mk_resp(data=response_data)
# ### END ### API Order ### get_account_obj_order_list() ###


# ### BEGIN ### API Order ### get_person_obj_order_list() ###
# Updated 2021-11-19
@router.get('/person/{person_id}/order/list', response_model=Resp_Body_Base)
async def get_person_id_order_list(
        person_id: str = Query(..., min_length=11, max_length=22),
        limit: int = 50,
        enabled: str = 'enabled',
        from_datetime: datetime.datetime = None,
        to_datetime: datetime.datetime = None,
        inc_order_cfg: bool = False,
        inc_order_line_list: bool = False,
        # inc_order_w_prod_type: str = 'all', # all, membership, fundraising, event, etc
        status: str = 'closed', # open, locked, reopened?, closed (complete), canceled, other
        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())

    if person_id := redis_lookup_id_random(record_id_random=person_id, table_name='person'): pass
    else: return mk_resp(data=None, status_code=404, response=response)

    # Updated 2021-06-28
    if order_rec_list_result := get_order_rec_list(
            for_obj_type = 'person',
            for_obj_id = person_id,
            limit = limit,
            enabled = enabled,
            from_datetime = from_datetime,
            to_datetime = to_datetime,
            status = status,
            ):
        order_result_list = []
        for order_rec in order_rec_list_result:
            if load_order_result := load_order_obj(
                    order_id = order_rec.get('order_id', None),
                    limit = limit,
                    enabled = enabled,
                    by_alias = by_alias,
                    exclude_unset = exclude_unset,
                    # model_as_dict = model_as_dict,
                    inc_order_cfg = inc_order_cfg,
                    inc_order_line_list = inc_order_line_list,
                    ):
                order_result_list.append(load_order_result)
            else:
                order_result_list.append(None)
        response_data = order_result_list
    else:
        return mk_resp(data=False, status_code=400, response=response) # Bad Request

    return mk_resp(data=response_data)
# ### END ### API Order ### get_person_obj_order_list() ###


# ### BEGIN ### API Order ### get_person_id_order_cart() ###
# NOTE 2021-08-09: Use with rework of order_cart. The most recent (hopefully only one) "open" order for a person.
# Updated 2021-11-19
@router.get('/person/{person_id}/order/cart', response_model=Resp_Body_Base)
async def get_person_id_order_cart(
        person_id: str = Query(..., min_length=11, max_length=22),
        enabled: str = 'enabled',
        inc_order_line_list: bool = False,
        inc_order_cfg: bool = False,
        x_account_id: str = Header(...),
        by_alias: bool = True,
        exclude_unset: bool = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    # Query to get the one "open" order status for a person ID

    return False
# ### END ### API Order ### get_person_id_order_cart() ###


@router.delete('/order/{obj_id}', response_model=Resp_Body_Base)
async def delete_order_obj(
        obj_id: str = Query(..., min_length=1, 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 = 'order'
    result = delete_obj_template(
        obj_type=obj_type,
        obj_id=obj_id,
        )
    return result


# NOTE 2021-08-09: Use with rework of order_cart
# NOTE: The router needs to have the prefix (/order) removed.
@router.delete('/order/{order_id}/line/<order_line_id>/remove', response_model=Resp_Body_Base)
async def delete_order_line_obj_NOT_SURE(
        order_id: str = Query(..., min_length=1, max_length=22),
        obj: Order_Base = None,
        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())