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

229 lines
8.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 PoolMeta
from trytond.pyson import Bool, Eval, If
from trytond.tools import cached_property
class Line(metaclass=PoolMeta):
__name__ = 'sale.line'
product_customer = fields.Many2One(
'sale.product_customer', "Customer's Product",
ondelete='RESTRICT',
domain=[
If(Bool(Eval('product')),
['OR',
[
('template.products', '=', Eval('product', -1)),
('product', '=', None),
],
('product', '=', Eval('product', -1)),
],
[]),
('party', '=', Eval('customer', -1)),
],
states={
'invisible': Eval('type') != 'line',
'readonly': Eval('sale_state') != 'draft',
},
depends={'sale'})
@fields.depends('sale', '_parent_sale.party')
def _get_product_customer_pattern(self):
return {
'party': (
self.sale.party.id if self.sale and self.sale.party else -1),
}
@fields.depends('product', 'product_customer',
methods=['_get_product_customer_pattern'])
def on_change_product(self):
super().on_change_product()
if not self.product:
return
product_customers = list(self.product.product_customer_used(
**self._get_product_customer_pattern()))
if len(product_customers) == 1:
self.product_customer, = product_customers
elif (self.product_customer
and self.product_customer not in product_customers):
self.product_customer = None
@fields.depends('product', 'product_customer',
methods=['on_change_product'])
def on_change_product_customer(self):
if self.product_customer:
if self.product_customer.product:
self.product = self.product_customer.product
elif not self.product:
if len(self.product_customer.template.products) == 1:
self.product, = self.product_customer.template.products
self.on_change_product()
@cached_property
def product_name(self):
name = super().product_name
if self.product_customer:
name = self.product_customer.rec_name
return name
class AmendmentLine(metaclass=PoolMeta):
__name__ = 'sale.amendment.line'
product_customer = fields.Many2One(
'sale.product_customer', "Customer's Product",
domain=[
['OR',
[
('template.products', '=', Eval('product', -1)),
('product', '=', None),
],
('product', '=', Eval('product', -1)),
],
('party', '=', Eval('party', -1)),
],
states={
'invisible': Eval('action') != 'line',
})
@fields.depends('line')
def on_change_line(self):
super().on_change_line()
if self.line:
self.product_customer = self.line.product_customer
@fields.depends('product', 'product_customer', 'line')
def on_change_product(self):
try:
super().on_change_product()
except AttributeError:
pass
if not self.product or not self.line:
return
product_customers = list(self.product.product_customer_used(
**self.line._get_product_customer_pattern()))
if len(product_customers) == 1:
self.product_customer, = product_customers
elif (self.product_customer
and self.product_customer not in product_customers):
self.product_customer = None
@fields.depends('product', 'product_customer',
methods=['on_change_product'])
def on_change_product_customer(self):
if self.product_customer:
if self.product_customer.product:
self.product = self.product_customer.product
elif not self.product:
if len(self.product_customer.template.products) == 1:
self.product, = self.product_customer.template.products
self.on_change_product()
def _apply_line(self, sale, sale_line):
super()._apply_line(sale, sale_line)
sale_line.product_customer = self.product_customer
class LineBlanketAgreement(metaclass=PoolMeta):
__name__ = 'sale.line'
@classmethod
def _domain_blanket_agreemnt_line_product(cls):
return [
super()._domain_blanket_agreemnt_line_product(),
If(Eval('product_customer'),
['OR',
('product_customer', '=', Eval('product_customer')),
('product_customer', '=', None),
],
[]),
]
@fields.depends(
'blanket_agreement_line', 'product_customer',
'_parent_blanket_agreement_line.product_customer')
def on_change_blanket_agreement_line(self):
if self.blanket_agreement_line:
line = self.blanket_agreement_line
if not self.product_customer:
self.product_customer = line.product_customer
super().on_change_blanket_agreement_line()
@fields.depends(
'product_customer', '_parent_blanket_agreement_line.product_customer')
def is_valid_product_for_blanket_agreement(self):
return (super().is_valid_product_for_blanket_agreement()
and (
(self.product_customer
== self.blanket_agreement_line.product_customer)
or not self.product_customer))
class BlanketAgreementLine(metaclass=PoolMeta):
__name__ = 'sale.blanket_agreement.line'
product_customer = fields.Many2One(
'sale.product_customer', "Customer's Product", ondelete='RESTRICT',
domain=[
If(Bool(Eval('product')),
['OR',
[
('template.products', '=', Eval('product')),
('product', '=', None),
],
('product', '=', Eval('product')),
],
[]),
('party', '=', Eval('_parent_blanket_agreement', {}).get(
'customer', -1)),
],
states={
'readonly': Eval('agreement_state') != 'draft'
})
@fields.depends('blanket_agreement', '_parent_blanket_agreement.customer')
def _get_product_customer_pattern(self):
return {
'party': (
self.blanket_agreement.customer.id
if self.blanket_agreement and self.blanket_agreement.customer
else -1),
}
@fields.depends('product', 'product_customer')
def on_change_product(self):
try:
super().on_change_product()
except AttributeError:
pass
if not self.product:
return
product_customers = list(self.product.product_customer_used(
**self._get_product_customer_pattern()))
if len(product_customers) == 1:
self.product_customer, = product_customers
elif (self.product_customer
and self.product_customer not in product_customers):
self.product_customer = None
@fields.depends('product', 'product_customer',
methods=['on_change_product'])
def on_change_product_customer(self):
if self.product_customer:
if self.product_customer.product:
self.product = self.product_customer.product
elif not self.product:
if len(self.product_customer.template.products) == 1:
self.product, = self.product_customer.template.products
self.on_change_product()
def get_sale_line(self, sale):
sale_line = super().get_sale_line(sale)
if self.product_customer:
sale_line.product_customer = self.product_customer
return sale_line