statement_analytic.py

# 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.pool import Pool, PoolMeta
from trytond.model import fields
from trytond.pyson import Eval
from trytond.modules.analytic_account import AnalyticMixin


class Line(AnalyticMixin, metaclass=PoolMeta):
    __name__ = 'account.statement.line'

    @classmethod
    def __setup__(cls):
        super().__setup__()
        cls.analytic_accounts.domain = [
            ('company', '=', Eval('company', -1)),
            ]
        cls.analytic_accounts.states = {
            'readonly': Eval('statement_state') != 'draft',
            }

    def get_move_line(self):
        pool = Pool()
        AnalyticLine = pool.get('analytic_account.line')

        move_line = super().get_move_line()

        if not hasattr(self, 'analytic_accounts'):
            return move_line

        if move_line and self.analytic_accounts:
            for analytic_account in self.analytic_accounts:
                if analytic_account.account:
                    account = analytic_account.account
                    if move_line.account != self.account:
                        continue
                    analytic_line = AnalyticLine()
                    analytic_line.debit = move_line.debit
                    analytic_line.credit = move_line.credit
                    analytic_line.account = account
                    analytic_line.date = self.date
                    if not hasattr(move_line, 'analytic_lines'):
                        move_line.analytic_lines = (analytic_line,)
                    else:
                        move_line.analytic_lines += (analytic_line,)
        return move_line


class Origin(metaclass=PoolMeta):
    __name__ = 'account.statement.origin'

    def _get_suggested_values(self, parent, name, line, amount, related_to,
            similarity):
        pool = Pool()
        AnalyticAccountEntry = pool.get('analytic.account.entry')

        values = super()._get_suggested_values(parent, name, line, amount,
            related_to, similarity)
        if hasattr(line, 'analytic_accounts') and line.analytic_accounts:
            new_entry = AnalyticAccountEntry.copy(line.analytic_accounts,
                default={
                    'origin': None,
                    })
            values['analytic_accounts'] = [('add', [e.id for e in new_entry])]
        return values


class OriginSuggestedLine(AnalyticMixin, metaclass=PoolMeta):
    __name__ = 'account.statement.origin.suggested.line'

    @classmethod
    def __setup__(cls):
        super().__setup__()
        cls.analytic_accounts.domain = [
            ('company', '=', Eval('company', -1)),
            ]
        cls.analytic_accounts.states = {
            'readonly': Eval('origin.state') != 'registered',
            }

    @classmethod
    def get_suggested_values(cls, child, description=None):
        pool = Pool()
        AnalyticAccountEntry = pool.get('analytic.account.entry')

        values = super().get_suggested_values(child, description)
        if child.analytic_accounts:
            new_entry = AnalyticAccountEntry.copy(child.analytic_accounts,
                default={
                    'origin': None,
                    })
            values['analytic_accounts'] = [('add', [e.id for e in new_entry])]
        return values


class AnalyticAccountEntry(metaclass=PoolMeta):
    __name__ = 'analytic.account.entry'

    @classmethod
    def _get_origin(cls):
        origins = super()._get_origin()
        return origins + ['account.statement.line',
            'account.statement.origin.suggested.line']

    @fields.depends('origin')
    def on_change_with_company(self, name=None):
        pool = Pool()
        StatementLine = pool.get('account.statement.line')
        SuggestedLine = pool.get('account.statement.origin.suggested.line')

        company = super().on_change_with_company(name)
        if (isinstance(self.origin, StatementLine)
                or isinstance(self.origin, SuggestedLine)):
            company = self.origin.company.id
        return company

    @classmethod
    def search_company(cls, name, clause):
        domain = super().search_company(name, clause),
        return ['OR',
            domain,
            (('origin.company',) + tuple(clause[1:]) +
                ('account.statement.line',)),
            (('origin.company',) + tuple(clause[1:]) +
                ('account.statement.origin.suggested.line',)),
            ]