/usr/lib/python2.7/dist-packages/trytond/modules/product_price_list/price_list.py is in tryton-modules-product-price-list 4.2.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 | # 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 decimal import Decimal
from simpleeval import simple_eval
from trytond.model import ModelView, ModelSQL, MatchMixin, fields, \
sequence_ordered
from trytond.tools import decistmt
from trytond.pyson import If, Eval
from trytond.transaction import Transaction
from trytond.pool import Pool
from trytond import backend
__all__ = ['PriceList', 'PriceListLine']
class PriceList(ModelSQL, ModelView):
'Price List'
__name__ = 'product.price_list'
name = fields.Char('Name', required=True, translate=True)
company = fields.Many2One('company.company', 'Company', required=True,
select=True, domain=[
('id', If(Eval('context', {}).contains('company'), '=', '!='),
Eval('context', {}).get('company', -1)),
])
tax_included = fields.Boolean('Tax Included')
lines = fields.One2Many('product.price_list.line', 'price_list', 'Lines')
@staticmethod
def default_company():
return Transaction().context.get('company')
@staticmethod
def default_tax_included():
return False
def get_context_formula(self, party, product, unit_price, quantity, uom):
return {
'names': {
'unit_price': unit_price,
},
}
def compute(self, party, product, unit_price, quantity, uom,
pattern=None):
'Compute price based on price list of party'
Uom = Pool().get('product.uom')
if pattern is None:
pattern = {}
pattern = pattern.copy()
pattern['product'] = product and product.id or None
pattern['quantity'] = Uom.compute_qty(uom, quantity,
product.default_uom, round=False) if product else quantity
context = self.get_context_formula(
party, product, unit_price, quantity, uom)
for line in self.lines:
if line.match(pattern):
return line.get_unit_price(**context)
return unit_price
class PriceListLine(sequence_ordered(), ModelSQL, ModelView, MatchMixin):
'Price List Line'
__name__ = 'product.price_list.line'
price_list = fields.Many2One('product.price_list', 'Price List',
required=True, ondelete='CASCADE')
product = fields.Many2One('product.product', 'Product', ondelete='CASCADE')
quantity = fields.Float('Quantity', digits=(16, Eval('unit_digits', 2)),
depends=['unit_digits'])
unit_digits = fields.Function(fields.Integer('Unit Digits'),
'on_change_with_unit_digits')
formula = fields.Char('Formula', required=True,
help=('Python expression that will be evaluated with:\n'
'- unit_price: the original unit_price'))
@classmethod
def __setup__(cls):
super(PriceListLine, cls).__setup__()
cls._order.insert(1, ('price_list', 'ASC'))
cls._error_messages.update({
'invalid_formula': ('Invalid formula "%(formula)s" in price '
'list line "%(line)s" with exception "%(exception)s".'),
})
@classmethod
def __register__(cls, module_name):
TableHandler = backend.get('TableHandler')
table = TableHandler(cls, module_name)
super(PriceListLine, cls).__register__(module_name)
# Migration from 2.4: drop required on sequence
table.not_null_action('sequence', action='remove')
@staticmethod
def default_formula():
return 'unit_price'
@fields.depends('product')
def on_change_with_unit_digits(self, name=None):
if self.product:
return self.product.default_uom.digits
return 2
@classmethod
def validate(cls, lines):
super(PriceListLine, cls).validate(lines)
for line in lines:
line.check_formula()
def check_formula(self):
'''
Check formula
'''
context = self.price_list.get_context_formula(
None, None, Decimal('0'), 0, None)
try:
if not isinstance(self.get_unit_price(**context), Decimal):
raise ValueError
except Exception, exception:
self.raise_user_error('invalid_formula', {
'formula': self.formula,
'line': self.rec_name,
'exception': exception,
})
def match(self, pattern):
if 'quantity' in pattern:
pattern = pattern.copy()
quantity = pattern.pop('quantity')
if self.quantity is not None and self.quantity > quantity:
return False
return super(PriceListLine, self).match(pattern)
def get_unit_price(self, **context):
'Return unit price (as Decimal)'
context.setdefault('functions', {})['Decimal'] = Decimal
return simple_eval(decistmt(self.formula), **context)
|