diff options
Diffstat (limited to 'indoteknik_custom/models')
| -rw-r--r-- | indoteknik_custom/models/website_user_cart.py | 8 |
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: |