Files
tryton/account.py
T
2019-03-24 18:00:01 +01:00

75 lines
2.3 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.pyson import If, Eval, Bool
def AccountTypeMixin(template=False):
class Mixin:
unearned_revenue = fields.Boolean(
"Unearned Revenue",
domain=[
If(Eval('statement') != 'balance',
('unearned_revenue', '=', False), ()),
],
states={
'invisible': ((Eval('statement') != 'balance')
| Eval('assets', True)),
},
depends=['statement', 'assets'])
if not template:
for fname in dir(Mixin):
field = getattr(Mixin, fname)
if not isinstance(field, fields.Field):
continue
field.states['readonly'] = (
Bool(Eval('template', -1)) & ~Eval('template_override', False))
return Mixin
class AccountTypeTemplate(AccountTypeMixin(template=True), metaclass=PoolMeta):
__name__ = 'account.account.type.template'
def _get_type_value(self, type=None):
values = super()._get_type_value(type=type)
if not type or type.unearned_revenue != self.unearned_revenue:
values['unearned_revenue'] = self.unearned_revenue
return values
class AccountType(AccountTypeMixin(), metaclass=PoolMeta):
__name__ = 'account.account.type'
class Invoice(metaclass=PoolMeta):
__name__ = 'account.invoice'
def get_sales(self, name):
pool = Pool()
AdvancePaymentCondition = pool.get('sale.advance_payment.condition')
sales = set(super(Invoice, self).get_sales(name))
for line in self.lines:
if isinstance(line.origin, AdvancePaymentCondition):
sales.add(line.origin.sale.id)
return list(sales)
class InvoiceLine(metaclass=PoolMeta):
__name__ = 'account.invoice.line'
@classmethod
def _account_domain(cls, type_):
domain = super()._account_domain(type_)
if type_ == 'out':
domain.append(('type.unearned_revenue', '=', True))
return domain
@classmethod
def _get_origin(cls):
return (super(InvoiceLine, cls)._get_origin()
+ ['sale.advance_payment.condition'])