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() ###