208 lines
7.6 KiB
Python
208 lines
7.6 KiB
Python
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 *
|
|
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 load_order_obj, save_order_obj
|
|
|
|
from app.models.response_models import *
|
|
from app.models.order_models import Order_Base
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post('', response_model=Resp_Body_Base)
|
|
async def post_order_obj(
|
|
obj: Order_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 = 'order'
|
|
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_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,
|
|
):
|
|
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
|
|
|
|
|
|
@router.get('/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,
|
|
):
|
|
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
# obj_type = 'order'
|
|
# 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
|
|
|
|
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, inc_user_obj=inc_user_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)
|
|
|
|
|
|
# ### BEGIN ### API Order Routes ### get_order_obj() ###
|
|
# Updated 2021-08-07
|
|
@router.get('/{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,
|
|
x_account_id: str = Header(...),
|
|
by_alias: bool = True,
|
|
exclude_unset: bool = True,
|
|
):
|
|
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,
|
|
):
|
|
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) # Not Found
|
|
# ### END ### API Order Routes ### get_order_obj() ###
|
|
|
|
|
|
@router.delete('/{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(...),
|
|
):
|
|
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 |