Fix cart API

1 files changed, 6 insertions, 2 deletions

diff --git a/indoteknik_custom/models/website_user_cart.py b/indoteknik_custom/models/website_user_cart.py
index 47a695fe..2721d66f 100644
--- a/indoteknik_custom/models/website_user_cart.py
+++ b/indoteknik_custom/models/website_user_cart.py
@@ -55,11 +55,15 @@ class WebsiteUserCart(models.Model):
 
     def get_product_by_user(self, user_id, selected=False, source=False):
         user_id = int(user_id)
-        source = source if source else 'add_to_cart'
+        
+        if source == 'buy':
+            source = ['buy']
+        else:
+            source = ['add_to_cart', 'buy']
 
         parameters = [
             ('user_id', '=', user_id), 
-            ('source', '=', source)
+            ('source', 'in', source)
         ]
 
         if selected: