first commit

modules/sale_price_list/product.py

@@ -0,0 +1,93 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+from trytond.model import fields
+from trytond.pool import Pool, PoolMeta
+from trytond.transaction import Transaction
+
+
+class Product(metaclass=PoolMeta):
+    __name__ = 'product.product'
+
+    def _get_sale_unit_price(self, quantity=0):
+        pool = Pool()
+        Date = pool.get('ir.date')
+        PriceList = pool.get('product.price_list')
+        Uom = pool.get('product.uom')
+        Tax = pool.get('account.tax')
+        User = pool.get('res.user')
+        context = Transaction().context
+        today = Date.today()
+
+        unit_price = super()._get_sale_unit_price(quantity=quantity)
+
+        if context.get('price_list'):
+            price_list = PriceList(context['price_list'])
+            assert price_list.company == User(Transaction().user).company
+            context_uom = None
+            if context.get('uom'):
+                context_uom = Uom(context['uom'])
+            taxes = None
+            if context.get('taxes'):
+                taxes = Tax.browse(context.get('taxes'))
+            uom = context_uom or self.sale_uom
+            if uom.category != self.sale_uom.category:
+                uom = self.sale_uom
+            unit_price = price_list.compute(self, quantity, uom)
+            if price_list.tax_included and taxes and unit_price is not None:
+                unit_price = Tax.reverse_compute(
+                    unit_price, taxes, today)
+        return unit_price
+
+
+class PriceList(metaclass=PoolMeta):
+    __name__ = 'product.price_list'
+
+    @classmethod
+    def __setup__(cls):
+        super().__setup__()
+        cls.unit.selection.append(('product_sale', "Product Sale"))
+
+    def get_uom(self, product):
+        uom = super().get_uom(product)
+        if self.unit == 'product_sale' and product.sale_uom:
+            uom = product.sale_uom
+        return uom
+
+
+class SaleContext(metaclass=PoolMeta):
+    __name__ = 'product.sale.context'
+
+    price_list = fields.Many2One('product.price_list', "Price List")
+
+    @classmethod
+    def default_price_list(cls, **pattern):
+        pool = Pool()
+        Configuration = pool.get('sale.configuration')
+        context = Transaction().context
+        price_list = context.get('price_list')
+        if price_list is None:
+            config = Configuration(1)
+            price_list = config.get_multivalue('sale_price_list', **pattern)
+            if price_list:
+                price_list = price_list.id
+        return price_list
+
+    @fields.depends(methods=['on_change_customer'])
+    def on_change_company(self):
+        try:
+            super().on_change_company()
+        except AttributeError:
+            pass
+        self.on_change_customer()
+
+    @fields.depends('customer', 'company')
+    def on_change_customer(self):
+        try:
+            super().on_change_customer()
+        except AttributeError:
+            pass
+        if self.customer and self.customer.sale_price_list:
+            self.price_list = self.customer.sale_price_list
+        else:
+            self.price_list = self.default_price_list(
+                company=self.company.id if self.company else None)