Add source on get checkout and post checkout API

1 files changed, 6 insertions, 3 deletions

diff --git a/indoteknik_api/controllers/api_v1/sale_order.py b/indoteknik_api/controllers/api_v1/sale_order.py
index 208789af..a8b2fd11 100644
--- a/indoteknik_api/controllers/api_v1/sale_order.py
+++ b/indoteknik_api/controllers/api_v1/sale_order.py
@@ -237,8 +237,9 @@ class SaleOrder(controller.Controller):
     def get_user_checkout_so(self, user_id, **kw):
         cart = request.env['website.user.cart']
         voucher_code = kw.get('voucher')
+        source = kw.get('source')
         voucher = request.env['voucher'].search([('code', '=', voucher_code)], limit=1)
-        result = cart.get_user_checkout(user_id, voucher)
+        result = cart.get_user_checkout(user_id, voucher, source)
         return self.response(result)
     
     @http.route(PREFIX_PARTNER + 'sale_order/checkout', auth='public', method=['POST', 'OPTIONS'], csrf=False)
@@ -259,7 +260,8 @@ class SaleOrder(controller.Controller):
             'delivery_amount': ['number', 'default:0'],
             'carrier_id': [],
             'delivery_service_type': [],
-            'voucher': []
+            'voucher': [],
+            'source': []
         })
         
         if not params['valid']:
@@ -297,7 +299,8 @@ class SaleOrder(controller.Controller):
 
         user_id = params['value']['user_id']
         user_cart = request.env['website.user.cart']
-        products = user_cart.get_product_by_user(user_id=user_id, selected=True)
+        source = params['value']['source']
+        products = user_cart.get_product_by_user(user_id=user_id, selected=True, source=source)
         for product in products:
             total_qty = product['quantity']
             price_unit = product['price']['price']