orders.py 9.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252
  1. import time
  2. from logging import getLogger
  3. from threading import Thread
  4. from uuid import uuid4
  5. from fastapi import HTTPException, APIRouter, Depends
  6. from fastapi.responses import JSONResponse
  7. from fudo import fudo
  8. from models.sales import ItemWeb, OrderWeb
  9. from models.items import Item, Order
  10. from models.user import User
  11. from services.fudo_service import add_product_to_fudo
  12. from services.email_service import get_email_sender
  13. import services.print_service as ps
  14. from services.data_service import DataServiceFactory
  15. from config.mails import PRINTER_DISCONNECTED_MAIL
  16. from config.messages import ErrorResponse, SuccessResponse, UserResponse
  17. from config.settings import DEVELOPMENT
  18. from auth.security import get_current_user
  19. logger = getLogger(__name__)
  20. # Data services initialization
  21. user_data_service = DataServiceFactory.get_user_service()
  22. product_data_service = DataServiceFactory.get_product_service()
  23. sale_data_service = DataServiceFactory.get_sales_service()
  24. # Global variables
  25. printer_orders = []
  26. order_router = APIRouter()
  27. name_promo = "Shop"
  28. @order_router.post("/send")
  29. async def printer_order(order: OrderWeb, current_user: User = Depends(get_current_user)):
  30. """Process printer order"""
  31. logger.info(f"Printer order received from user {current_user.email} for table {order.table}")
  32. # Printer status validation
  33. if not DEVELOPMENT:
  34. try:
  35. printer_status = ps.get_status()
  36. if not printer_status:
  37. logger.error(f"Printer is not connected. Order from user {current_user.email} cannot be processed.")
  38. # Send notification email to admins
  39. email_thread = Thread(
  40. target=get_email_sender().send_email,
  41. args=(
  42. PRINTER_DISCONNECTED_MAIL["subject"],
  43. PRINTER_DISCONNECTED_MAIL["body"],
  44. ["erwinjacimino2003@gmail.com", "mompyn@gmail.com"]
  45. ),
  46. daemon=True
  47. )
  48. email_thread.start()
  49. return JSONResponse(status_code=424, content={"message": ErrorResponse.PRINTER_DISCONNECTED})
  50. except Exception as e:
  51. logger.error(f"Error checking printer status: {e}")
  52. return JSONResponse(status_code=500, content={"message": "Error interno del servidor"})
  53. # Extract order data
  54. items = order.items
  55. table = order.table
  56. # Input validation
  57. if not items or not table:
  58. logger.warning(f"Invalid order data from user {current_user.email}: missing items or table")
  59. return JSONResponse(status_code=400, content={"message": ErrorResponse.MISSING_FIELDS})
  60. if not isinstance(table, int):
  61. logger.warning(f"Invalid table type from user {current_user.email}: {type(table)}")
  62. return JSONResponse(status_code=400, content={"message": ErrorResponse.INVALID_TABLE_TYPE})
  63. logger.info(f"Processing order for table {table} with {len(items)} items")
  64. # Add products to Fudo
  65. product_errors = []
  66. # Get products data
  67. try:
  68. products = product_data_service.get_products([item.id for item in items])
  69. logger.info(f"Retrieved {len(products)} products from database")
  70. except Exception as e:
  71. error_msg = f"Error retrieving products: {e}"
  72. logger.error(error_msg)
  73. return JSONResponse(status_code=500, content={"message": "Error interno del servidor"})
  74. beers_for_promo = 0
  75. try:
  76. fudo.get_token()
  77. logger.info("Fudo token obtained successfully")
  78. for item, product in zip(items, products):
  79. try:
  80. # Si es dia de promo
  81. if time.localtime().tm_wday + 1 == product.promo_day and product.promo_id:
  82. logger.info(f"Applying promotion for product {product.id} on table {table}")
  83. fudo_product = add_product_to_fudo(product.promo_id, item.quantity, table)
  84. #en caso contrario
  85. else:
  86. if product.type == name_promo:
  87. beers_for_promo += item.quantity
  88. logger.debug(f"Added {item.quantity} beers for promotion calculation")
  89. fudo_product = add_product_to_fudo(item.id, item.quantity, table)
  90. logger.info(f"Added product {item.id} to table {table} with quantity {item.quantity} in fudo")
  91. if not fudo_product:
  92. error_msg = f"Error adding product {item.id} to table {table}."
  93. product_errors.append(error_msg)
  94. logger.error(error_msg)
  95. except Exception as e:
  96. error_msg = f"Error processing product {item.id}: {e}"
  97. logger.error(error_msg)
  98. product_errors.append(error_msg)
  99. except Exception as e:
  100. error_msg = f"Error with Fudo integration: {e}"
  101. logger.error(error_msg)
  102. return JSONResponse(status_code=500, content={"message": "Error interno del servidor"})
  103. if product_errors:
  104. logger.error(f"Product errors occurred: {product_errors}")
  105. return JSONResponse(
  106. status_code=424,
  107. content={"message": ErrorResponse.PRODUCT_ADD_ERROR, "errors": product_errors}
  108. )
  109. # User validation
  110. try:
  111. user = user_data_service.get_by_id(order.customerId)
  112. if not user:
  113. logger.warning(f"User not found: {order.customerId}")
  114. return JSONResponse(status_code=404, content={"message": UserResponse.USER_NOT_FOUND.format(user_id=order.customerId)})
  115. logger.info(f"Order customer validated: {user.email}")
  116. except Exception as e:
  117. error_msg = f"Error validating user {order.customerId}: {e}"
  118. logger.error(error_msg)
  119. return JSONResponse(status_code=500, content={"message": "Error interno del servidor"})
  120. # Get active sale
  121. try:
  122. active_sale_id = fudo.get_active_sale(fudo.get_table(table))
  123. if not active_sale_id:
  124. error_msg = f"No active sale found for table {table}"
  125. logger.error(error_msg)
  126. raise HTTPException(status_code=404, detail=error_msg)
  127. active_sale_id = active_sale_id['id']
  128. logger.info(f"Active sale found for table {table}: {active_sale_id}")
  129. except HTTPException:
  130. raise
  131. except Exception as e:
  132. error_msg = f"Error retrieving active sale for table {table}: {e}"
  133. logger.error(error_msg)
  134. raise HTTPException(status_code=500, detail="Error interno del servidor")
  135. # Update user reward progress
  136. try:
  137. new_progress = user.reward_progress + beers_for_promo * 10
  138. user_data_service.set_reward_progress(user.id, new_progress)
  139. logger.info(f"Updated reward progress for user {user.email}: {user.reward_progress} -> {new_progress}")
  140. except Exception as e:
  141. error_msg = f"Error updating reward progress for user {user.id}: {e}"
  142. logger.error(error_msg)
  143. # Don't fail the order for this, just log it
  144. new_progress = user.reward_progress
  145. # Create sale record
  146. try:
  147. sale = sale_data_service.create(
  148. order.customerId,
  149. active_sale_id or uuid4().hex,
  150. order.totalAmount,
  151. order.table,
  152. [item.id for item in items],
  153. [item.quantity for item in items]
  154. )
  155. if sale > 0:
  156. logger.info(f"Sale created successfully: ID {sale}")
  157. else:
  158. error_msg = "Failed to create sale record"
  159. logger.error(error_msg)
  160. return JSONResponse(status_code=500, content={"message": "Error interno del servidor"})
  161. except Exception as e:
  162. error_msg = f"Error creating sale record: {e}"
  163. logger.error(error_msg)
  164. return JSONResponse(status_code=500, content={"message": "Error interno del servidor"})
  165. # Print order
  166. try:
  167. order_for_print = Order(
  168. table=table,
  169. items=[Item(name=product.name, price=product.price or 0, quantity=item.quantity) for product, item in zip(products, items)],
  170. customerName=user.name,
  171. totalAmount=order.totalAmount,
  172. orderDate=order.orderDate
  173. )
  174. ps.print_order(order_for_print)
  175. logger.info(f"Order printed successfully for table {table}")
  176. except Exception as e:
  177. error_msg = f"Error printing order for table {table}: {e}"
  178. logger.error(error_msg)
  179. # Don't fail the order for print issues, just log it
  180. # Log order
  181. try:
  182. logger.info(f"Logging order for table {table} with sale ID {sale}, products= {[(product.name, item.quantity) for product, item in zip(products, items)]}")
  183. except Exception as e:
  184. logger.error(f"Error in legacy order logging: {e}")
  185. # Don't fail the order for logging issues
  186. logger.info(f"Order processing completed successfully for table {table}, sale ID: {sale}")
  187. return JSONResponse({"message": SuccessResponse.ORDER_SUCCESS, "new_progress": new_progress})