Fix cart API

1 files changed, 4 insertions, 3 deletions

diff --git a/indoteknik_api/controllers/api_v1/cart.py b/indoteknik_api/controllers/api_v1/cart.py
index 88fa9f88..8ef2c1c1 100644
--- a/indoteknik_api/controllers/api_v1/cart.py
+++ b/indoteknik_api/controllers/api_v1/cart.py
@@ -14,8 +14,9 @@ class Cart(controller.Controller):
         user_id = int(user_id)
         limit = int(kw.get('limit', 0))
         offset = int(kw.get('offset', 0))
-        query = [('user_id', '=', user_id), ('source', '=', 'add_to_cart')]
+        query = [('user_id', '=', user_id)]
         carts = user_cart.search(query, limit=limit, offset=offset, order='create_date desc')
+        carts.write({'source': 'add_to_cart'})
         data = {
             'product_total': user_cart.search_count(query),
             'products': carts.with_context(price_for="web").get_products()
@@ -26,7 +27,7 @@ class Cart(controller.Controller):
     @controller.Controller.must_authorized()
     def get_cart_count_by_user_id(self, user_id, **kw):
         user_id = int(user_id)
-        query = [('user_id', '=', user_id), ('source', '=', 'add_to_cart')]
+        query = [('user_id', '=', user_id)]
         carts = request.env['website.user.cart'].search_count(query)
         return self.response(carts)
     
@@ -77,7 +78,7 @@ class Cart(controller.Controller):
             data_to_update['source'] = source
 
         result = {}
-        if cart and source in (None, 'add_to_cart'):
+        if cart:
             # Update existing cart entry
             cart.write(data_to_update)
             result['id'] = cart.id