78 lines
2.6 KiB
Python
78 lines
2.6 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, Pool
|
|
from trytond.pyson import Eval, If
|
|
|
|
from trytond.modules.analytic_account import AnalyticMixin
|
|
|
|
__all__ = ['InvoiceLine', 'AnalyticAccountEntry']
|
|
|
|
|
|
class InvoiceLine(AnalyticMixin):
|
|
__name__ = 'account.invoice.line'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(InvoiceLine, cls).__setup__()
|
|
cls.analytic_accounts.domain = [
|
|
('company', '=', If(~Eval('company'),
|
|
Eval('context', {}).get('company', -1),
|
|
Eval('company', -1))),
|
|
]
|
|
cls.analytic_accounts.depends.append('company')
|
|
|
|
def _credit(self):
|
|
pool = Pool()
|
|
AnalyticAccountEntry = pool.get('analytic.account.entry')
|
|
|
|
line = super(InvoiceLine, self)._credit()
|
|
if self.analytic_accounts:
|
|
new_entries = AnalyticAccountEntry.copy(self.analytic_accounts,
|
|
default={
|
|
'origin': None,
|
|
})
|
|
line.analytic_accounts = new_entries
|
|
return line
|
|
|
|
def get_move_lines(self):
|
|
lines = super(InvoiceLine, self).get_move_lines()
|
|
if self.analytic_accounts:
|
|
date = self.invoice.accounting_date or self.invoice.invoice_date
|
|
for line in lines:
|
|
analytic_lines = []
|
|
for entry in self.analytic_accounts:
|
|
analytic_lines.extend(
|
|
entry.get_analytic_lines(line, date))
|
|
line.analytic_lines = analytic_lines
|
|
return lines
|
|
|
|
|
|
class AnalyticAccountEntry:
|
|
__metaclass__ = PoolMeta
|
|
__name__ = 'analytic.account.entry'
|
|
|
|
@classmethod
|
|
def _get_origin(cls):
|
|
origins = super(AnalyticAccountEntry, cls)._get_origin()
|
|
return origins + ['account.invoice.line']
|
|
|
|
@fields.depends('origin')
|
|
def on_change_with_company(self, name=None):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
company = super(AnalyticAccountEntry, self).on_change_with_company(
|
|
name)
|
|
if isinstance(self.origin, InvoiceLine):
|
|
company = self.origin.company.id
|
|
return company
|
|
|
|
@classmethod
|
|
def search_company(cls, name, clause):
|
|
domain = super(AnalyticAccountEntry, cls).search_company(name, clause),
|
|
return ['OR',
|
|
domain,
|
|
(('origin.company',) + tuple(clause[1:]) +
|
|
('account.invoice.line',)),
|
|
]
|