Files
tradon/modules/sale_price_list/product.py
2026-03-14 09:42:12 +00:00

94 lines
3.1 KiB
Python

# 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)