Refactor pricelist on product price

1 files changed, 5 insertions, 3 deletions

diff --git a/indoteknik_api/models/product_product.py b/indoteknik_api/models/product_product.py
index 9e218565..1740f992 100644
--- a/indoteknik_api/models/product_product.py
+++ b/indoteknik_api/models/product_product.py
@@ -27,7 +27,7 @@ class ProductProduct(models.Model):
         }
         return data
     
-    def v2_api_single_response(self, product_product, pricelist=False):
+    def v2_api_single_response(self, product_product):
         product_template = product_product.product_tmpl_id
         data = {
             'id': product_product.id,
@@ -38,7 +38,7 @@ class ProductProduct(models.Model):
             },
             'code': product_product.default_code or '',
             'name': product_product.display_name,
-            'price': product_product.calculate_website_price(pricelist),
+            'price': product_product.calculate_website_price(),
             'stock': product_product.qty_stock_vendor,
             'weight': product_product.weight,
             'attributes': [x.name for x in product_product.product_template_attribute_value_ids],
@@ -46,7 +46,9 @@ class ProductProduct(models.Model):
         }
         return data
     
-    def calculate_website_price(self, pricelist):
+    def calculate_website_price(self):
+        pricelist = self.env.user_pricelist
+
         config = self.env['ir.config_parameter']
         product_pricelist_tier1 = int(config.get_param('product.pricelist.tier1'))
         product_pricelist_tier2 = int(config.get_param('product.pricelist.tier2'))