Update routes/purchase.py

routes/purchase.py

@@ -6,13 +6,13 @@ router = APIRouter()
 
 
 @router.post("/add-to-cart")
-def handle_add_to_cart(query: Query):
+def handle_add_to_cart(input: Input):
     customer_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'customer_id'), None)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'customer_id'), None)
     product_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'product_id'), None)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'product_id'), None)
     quantity = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'quantity'), 1)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'quantity'), 1)
     if customer_id and product_id:
         cart = add_to_cart(customer_id, product_id, int(quantity))
         return cart
@@ -30,11 +30,11 @@ def handle_view_cart(customer_id: str):
 
 
 @router.post("/remove-from-cart")
-def handle_remove_from_cart(query: Query):
+def handle_remove_from_cart(input: Input):
     customer_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'customer_id'), None)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'customer_id'), None)
     product_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'product_id'), None)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'product_id'), None)
     if customer_id and product_id:
         cart = remove_from_cart(customer_id, product_id)
         return cart
@@ -44,9 +44,9 @@ def handle_remove_from_cart(query: Query):
 
 
 @router.post("/checkout")
-def handle_checkout(query: Query):
+def handle_checkout(input: Input):
     customer_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'customer_id'), None)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'customer_id'), None)
     if customer_id:
         order = checkout(customer_id)
         return order