273 lines
11 KiB
Python
273 lines
11 KiB
Python
from __future__ import annotations
|
|
import datetime
|
|
|
|
from typing import Dict, List, Optional, Set, Union
|
|
from pydantic import BaseModel, EmailStr, Field, PrivateAttr, ValidationError, validator
|
|
|
|
from app.db_sql import redis_lookup_id_random, sql_insert, sql_select, sql_update
|
|
from app.lib_general import log, logging
|
|
|
|
|
|
from app.models.order_line_models import Order_Line_Base, Order_Line_Full_Detail_Base
|
|
|
|
|
|
# ### BEGIN ### API Order Line Methods ### create_order_line_obj() ###
|
|
def create_order_line_obj(order_line_obj_new:Order_Line_Base):
|
|
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
if not order_line_obj_new:
|
|
return False
|
|
|
|
order_line_obj_data = order_line_obj_new.dict(by_alias=False, exclude_defaults=False, exclude_unset=True, exclude={'user', 'created_on', 'updated_on'})
|
|
|
|
if order_line_obj_in_result := sql_insert(data=order_line_obj_data, table_name='order_line', rm_id_random=True, id_random_length=8): pass
|
|
else:
|
|
return False
|
|
|
|
log.debug(order_line_obj_in_result)
|
|
|
|
order_line_id = order_line_obj_in_result
|
|
|
|
log.debug(f'Returning the new order_line_id: {order_line_id}')
|
|
return order_line_id
|
|
# ### END ### API Order Line Methods ### create_order_line_obj() ###
|
|
|
|
|
|
# ### BEGIN ### API Order Line Methods ### load_order_line_obj() ###
|
|
def load_order_line_obj(
|
|
order_line_id: int|str,
|
|
limit: int = 1000,
|
|
by_alias: bool = True,
|
|
exclude_unset: bool = True, # NOTE: Normally this is True
|
|
model_as_dict: bool = False, # NOTE: Normally this is False
|
|
) -> Order_Line_Base|dict|bool:
|
|
log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
if order_line_id := redis_lookup_id_random(record_id_random=order_line_id, table_name='order_line'): pass
|
|
else: return False
|
|
|
|
if order_line_rec := sql_select(table_name='v_order_line', record_id=order_line_id): pass
|
|
else: return False
|
|
|
|
log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(order_line_rec)
|
|
|
|
try:
|
|
order_line_obj = Order_Line_Base(**order_line_rec)
|
|
log.debug(order_line_obj)
|
|
except ValidationError as e:
|
|
log.error(e.json())
|
|
return False
|
|
|
|
if model_as_dict:
|
|
return order_line_obj.dict(by_alias=by_alias, exclude_unset=exclude_unset) # pylint: disable=no-member
|
|
else:
|
|
return order_line_obj
|
|
# ### END ### API Order Line Methods ### load_order_line_obj() ###
|
|
|
|
|
|
# ### BEGIN ### API Order Line Methods ### load_order_line_obj_full_detail() ###
|
|
# Updated 2021-11-22
|
|
def load_order_line_obj_full_detail(
|
|
order_line_rec: dict,
|
|
by_alias: bool = True,
|
|
exclude_unset: bool = True, # NOTE: Normally this is True
|
|
model_as_dict: bool = False, # NOTE: Normally this is False
|
|
) -> Order_Line_Full_Detail_Base|dict|bool:
|
|
log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
try:
|
|
order_line_obj_full_detail = Order_Line_Full_Detail_Base(**order_line_rec)
|
|
log.debug(order_line_obj_full_detail)
|
|
except ValidationError as e:
|
|
log.error(e.json())
|
|
return False
|
|
|
|
if model_as_dict:
|
|
return order_line_obj_full_detail.dict(by_alias=by_alias, exclude_unset=exclude_unset) # pylint: disable=no-member
|
|
else:
|
|
return order_line_obj_full_detail
|
|
# ### END ### API Order Line Methods ### load_order_line_obj_full_detail() ###
|
|
|
|
|
|
# ### BEGIN ### API Order Line Methods ### update_order_line_obj() ###
|
|
def update_order_line_obj(
|
|
order_line_id: int|str, # Ideally the int ID should be passed. This allows for updating of the id_random value.
|
|
order_line_obj_up: Order_Line_Base,
|
|
create_sub_obj: bool = False,
|
|
) -> bool:
|
|
log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
if order_line_id := redis_lookup_id_random(record_id_random=order_line_id, table_name='order_line'): pass
|
|
else: return False
|
|
|
|
order_line_obj_up.id = order_line_id
|
|
|
|
log.debug(order_line_obj_up)
|
|
# log.debug(order_line_obj_up.dict(by_alias=True, exclude_unset=True))
|
|
log.debug(order_line_obj_up.dict(by_alias=False, exclude_unset=True))
|
|
# log.debug(order_line_obj_up.dict(by_alias=False, exclude_unset=False))
|
|
|
|
#order_line_dict_up = order_line_obj_up.dict(by_alias=False, exclude_unset=True)
|
|
|
|
# if order_line_obj_up.person_id and order_line_obj_up.person:
|
|
# person_id = order_line_obj_up.person_id
|
|
# person_obj_up = order_line_obj_up.person
|
|
# log.debug(person_id)
|
|
# log.debug(person_obj_up)
|
|
# if person_obj_up_result := update_person_obj(
|
|
# person_id = person_id,
|
|
# person_obj_up = person_obj_up,
|
|
# create_sub_obj = create_sub_obj,
|
|
# ):
|
|
# log.debug(person_obj_up_result)
|
|
# else:
|
|
# log.debug(person_obj_up_result)
|
|
# return False
|
|
|
|
# if order_line_obj_up.user_id and order_line_obj_up.user:
|
|
# user_id = order_line_obj_up.user_id
|
|
# user_obj_up = order_line_obj_up.user
|
|
# log.debug(user_id)
|
|
# log.debug(user_obj_up)
|
|
# if user_obj_up_result := update_user_obj(
|
|
# user_id = user_id,
|
|
# user_dict_obj = user_obj_up,
|
|
# create_sub_obj = create_sub_obj,
|
|
# ):
|
|
# log.debug(user_obj_up_result)
|
|
# else:
|
|
# log.debug(user_obj_up_result)
|
|
# return False
|
|
|
|
order_line_dict_up = order_line_obj_up.dict(by_alias=False, exclude_unset=True, exclude={'user'})
|
|
log.debug(order_line_dict_up)
|
|
|
|
if order_line_obj_up_result := sql_update(data=order_line_dict_up, table_name='order_line', rm_id_random=True):
|
|
log.debug(order_line_obj_up_result)
|
|
return True
|
|
else:
|
|
log.debug(order_line_obj_up_result)
|
|
return False
|
|
# ### END ### API Order Line Methods ### update_order_line_obj() ###
|
|
|
|
|
|
# ### BEGIN ### API Order Line Methods ### get_order_line_rec_list() ###
|
|
def get_order_line_rec_list(
|
|
for_obj_type: str,
|
|
for_obj_id: str,
|
|
from_datetime: datetime.datetime = None, # For the order, not order_line
|
|
to_datetime: datetime.datetime = None, # For the order, not order_line
|
|
prod_type = 'all', # all, cont_edu_cert, event, fundraising, membership, etc
|
|
status: str = 'closed', # started, in progress, complete, all
|
|
full_detail: bool = False,
|
|
# enabled: str = 'enabled', # enabled, disabled, all
|
|
limit: int = 1000,
|
|
) -> list|bool:
|
|
log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
if for_obj_id := redis_lookup_id_random(record_id_random=for_obj_id, table_name=for_obj_type): pass
|
|
else: return False
|
|
|
|
data = {}
|
|
data[f'{for_obj_type}_id'] = for_obj_id
|
|
# data['for_obj_type'] = for_obj_type
|
|
sql_obj_type_id = f'`order_line`.{for_obj_type}_id = :{for_obj_type}_id'
|
|
|
|
allowed_prod_type_li = ['cont_edu_cert', 'event', 'fundraising', 'membership', 'other'] # TEMPORARY list...
|
|
sql_prod_type = ''
|
|
if prod_type in allowed_prod_type_li:
|
|
if prod_type == 'closed' or prod_type == 'complete':
|
|
data['prod_type'] = ['closed', 'complete']
|
|
sql_prod_type = f'AND `order_line`.product_type IN :prod_type'
|
|
elif prod_type == 'locked' or prod_type == 'in progress':
|
|
data['prod_type'] = ['locked', 'in progress']
|
|
sql_prod_type = f'AND `order_line`.product_type IN :prod_type'
|
|
else:
|
|
data['prod_type'] = prod_type
|
|
sql_prod_type = f'AND `order_line`.product_type = :prod_type'
|
|
elif prod_type == 'all':
|
|
sql_prod_type = f'AND `order_line`.product_type IS NOT NULL'
|
|
else:
|
|
log.warning('The prod_type value passed is not allowed. Returning None')
|
|
return False
|
|
|
|
allowed_status_li = ['open', 'locked', 'in progress', 'reopened', 'closed', 'complete', 'canceled', 'other'] # TEMPORARY list...
|
|
sql_status = ''
|
|
if status in allowed_status_li:
|
|
if status == 'closed' or status == 'complete':
|
|
data['status'] = ['closed', 'complete']
|
|
sql_status = f'AND `order_line`.order_status IN :status'
|
|
elif status == 'locked' or status == 'in progress':
|
|
data['status'] = ['locked', 'in progress']
|
|
sql_status = f'AND `order_line`.order_status IN :status'
|
|
else:
|
|
data['status'] = status
|
|
sql_status = f'AND `order_line`.order_status = :status'
|
|
elif status == 'all':
|
|
sql_status = f'AND `order_line`.order_status IS NOT NULL'
|
|
else:
|
|
log.warning('The status value passed is not allowed. Returning None')
|
|
return False
|
|
|
|
if from_datetime and to_datetime:
|
|
data['from_datetime'] = from_datetime
|
|
data['to_datetime'] = to_datetime
|
|
sql_from_to_datetime = f'AND `order_line`.order_created_on >= :from_datetime AND `order_line`.order_created_on <= :to_datetime'
|
|
elif from_datetime:
|
|
data['from_datetime'] = from_datetime
|
|
sql_from_to_datetime = f'AND `order_line`.order_created_on >= :from_datetime'
|
|
elif to_datetime:
|
|
data['to_datetime'] = to_datetime
|
|
sql_from_to_datetime = f'AND `order_line`.order_created_on <= :to_datetime'
|
|
else:
|
|
sql_from_to_datetime = ''
|
|
|
|
if limit:
|
|
data['limit'] = limit
|
|
sql_limit = f'LIMIT :limit'
|
|
else:
|
|
sql_limit = ''
|
|
|
|
log.debug(data)
|
|
|
|
if not full_detail:
|
|
sql = f"""
|
|
SELECT `order_line`.id AS 'order_line_id', `order_line`.id_random AS 'order_line_id_random'
|
|
FROM `v_order_line` AS `order_line`
|
|
WHERE
|
|
{sql_obj_type_id}
|
|
{sql_prod_type}
|
|
{sql_status}
|
|
{sql_from_to_datetime}
|
|
ORDER BY order_line.name, `order_line`.created_on DESC, `order_line`.updated_on DESC
|
|
{sql_limit};
|
|
"""
|
|
else:
|
|
sql = f"""
|
|
SELECT *
|
|
FROM `v_order_line_full_detail` AS `order_line`
|
|
WHERE
|
|
{sql_obj_type_id}
|
|
{sql_prod_type}
|
|
{sql_status}
|
|
{sql_from_to_datetime}
|
|
ORDER BY `order_line`.created_on DESC, `order_line`.updated_on DESC
|
|
{sql_limit};
|
|
"""
|
|
log.debug(sql)
|
|
|
|
if order_line_rec_li_result := sql_select(data=data, sql=sql, as_list=True):
|
|
order_line_rec_li = order_line_rec_li_result
|
|
else:
|
|
order_line_rec_li = []
|
|
log.debug(order_line_rec_li_result)
|
|
|
|
return order_line_rec_li
|
|
# ### END ### API Order Line Methods ### get_order_line_rec_list() ###
|