#This file is part of Tryton. The COPYRIGHT file at the top level of #this repository contains the full copyright notices and license terms. "Statement" from trytond.model import ModelWorkflow, ModelView, ModelSQL, fields from decimal import Decimal _STATES = {'readonly': 'state != "draft"'} class Statement(ModelWorkflow, ModelSQL, ModelView): 'Account Statement' _name = 'account.statement' _description = __doc__ journal = fields.Many2One('account.statement.journal', 'Journal', required=True, states={ 'readonly': "(state != 'draft') or (bool(lines))", }, on_change=['journal'], select=1) currency_digits = fields.Function('get_currency_digits', type='integer', string='Currency Digits', on_change_with=['journal']) date = fields.Date('Date', required=True, states=_STATES, select=1) start_balance = fields.Numeric('Start Balance', digits="(16, currency_digits)", states=_STATES, depends=['currency_digits']) end_balance = fields.Numeric('End Balance', digits="(16, currency_digits)", states=_STATES, depends=['currency_digits']) lines = fields.One2Many('account.statement.line', 'statement', 'Transactions', states={ 'readonly': "(state != 'draft') or (not bool(journal))", }, on_change=['lines', 'journal']) state = fields.Selection([ ('draft', 'Draft'), ('validated', 'Validated'), ('cancel', 'Canceled'), ('posted', 'Posted'), ], 'State', readonly=True, select=1) move_lines = fields.Function('get_move_lines', type='one2many', relation='account.move.line', string='Move Lines') def __init__(self): super(Statement, self).__init__() self._rpc.update({ 'draft_workflow': True, }) self._order[0] = ('id', 'DESC') self._error_messages.update({ 'wrong_end_balance': 'End Balance must be %s!', }) def default_state(self, cursor, user, context=None): return 'draft' def default_date(self, cursor, user, context=None): date_obj = self.pool.get('ir.date') return date_obj.today(cursor, user, context=context) def default_currency_digits(self, cursor, user, context=None): company_obj = self.pool.get('company.company') if context is None: context = {} if context.get('company'): company = company_obj.browse(cursor, user, context['company'], context=context) return company.currency.digits return 2 def on_change_journal(self, cursor, user, ids, value, context=None): res = {} if not value.get('journal'): return res statement_ids = self.search(cursor, user, [ ('journal', '=', value['journal']), ], order=[ ('date', 'DESC'), ], limit=1, context=context) if not statement_ids: return res statement = self.browse(cursor, user, statement_ids[0], context=context) res['start_balance'] = statement.end_balance return res def on_change_with_currency_digits(self, cursor, user, ids, vals, context=None): journal_obj = self.pool.get('account.statement.journal') if vals.get('journal'): journal = journal_obj.browse(cursor, user, vals['journal'], context=context) return journal.currency.digits return 2 def get_currency_digits(self, cursor, user, ids, name, arg, context=None): res = {} for statement in self.browse(cursor, user, ids, context=context): res[statement.id] = statement.journal.currency.digits return res def get_rec_name(self, cursor, user, ids, name, arg, context=None): lang_obj = self.pool.get('ir.lang') if context is None: context = {} if not ids: return {} for code in [context.get('language', False) or 'en_US', 'en_US']: lang_ids = lang_obj.search(cursor, user, [ ('code', '=', code), ], context=context) if lang_ids: break lang = lang_obj.browse(cursor, user, lang_ids[0], context=context) res = {} for statement in self.browse(cursor, user, ids, context=context): res[statement.id] = statement.journal.name + ' ' + \ lang.currency(lang, statement.start_balance, statement.journal.currency, symbol=False, grouping=True) + \ lang.currency(lang, statement.end_balance, statement.journal.currency, symbol=False, grouping=True) return res def search_rec_name(self, cursor, user, name, args, context=None): args2 = [] i = 0 while i < len(args): ids = self.search(cursor, user, ['OR', ('start_balance', args[i][1], args[i][2]), ('end_balance', args[i][1], args[i][2]), ], context=context) if ids: args2.append(('id', 'in', ids)) else: args2.append(('journal', args[i][1], args[i][2])) i += 1 return args2 def get_move_lines(self, cursor, user, ids, name, args, context=None): ''' Return the move lines that have been generated by the statements. ''' res = {} for statement in self.browse(cursor, user, ids, context=context): res[statement.id] = [] for line in statement.lines: if not line.move: continue for move_line in line.move.lines: res[statement.id].append(move_line.id) return res def get_end_balance(self, cursor, user, ids, name, arg, context=None): statements = self.browse(cursor, user, ids, context=context) res = {} for statement in statements: res[statement.id] = statement.start_balance for line in statement.lines: res[statement.id] += line.amount return res def on_change_lines(self, cursor, user, ids, values, context=None): invoice_obj = self.pool.get('account.invoice') journal_obj = self.pool.get('account.statement.journal') currency_obj = self.pool.get('currency.currency') res = { 'lines': {}, } if values.get('journal') and values.get('lines'): journal = journal_obj.browse(cursor, user, values['journal'], context=context) invoice_ids = set() for line in values['lines']: if line['invoice']: invoice_ids.add(line['invoice']) invoice_id2amount_to_pay = {} for invoice in invoice_obj.browse(cursor, user, invoice_ids, context=context): invoice_id2amount_to_pay[invoice.id] = currency_obj.compute( cursor, user, invoice.currency, invoice.amount_to_pay, journal.currency, context=context) for line in values['lines']: if line['invoice'] and line['id']: amount_to_pay = invoice_id2amount_to_pay[line['invoice']] if abs(line['amount']) > amount_to_pay: res['lines'].setdefault('update', []) if currency_obj.is_zero(cursor, user, journal.currency, amount_to_pay): res['lines']['update'].append({ 'id': line['id'], 'invoice': False, }) else: res['lines']['update'].append({ 'id': line['id'], 'amount': amount_to_pay, }) res['lines'].setdefault('add', []) vals = line.copy() del vals['id'] vals['amount'] = abs(line['amount']) - amount_to_pay if line['amount'] < 0: vals['amount'] = - vals['amount'] vals['invoice'] = False res['lines']['add'].append(vals) invoice_id2amount_to_pay[line['invoice']] = \ amount_to_pay - abs(line['amount']) return res def set_state_validated(self, cursor, user, statement_id, context=None): statement_line_obj = self.pool.get('account.statement.line') lang_obj = self.pool.get('ir.lang') if context is None: context = {} statement = self.browse(cursor, user, statement_id, context=context) computed_end_balance = statement.start_balance for line in statement.lines: computed_end_balance += line.amount if computed_end_balance != statement.end_balance: for code in [context.get('language', False) or 'en_US', 'en_US']: lang_ids = lang_obj.search(cursor, user, [ ('code', '=', code), ], context=context) if lang_ids: break lang = lang_obj.browse(cursor, user, lang_ids[0], context=context) amount = lang_obj.format(lang, '%.' + str(statement.journal.currency.digits) + 'f', computed_end_balance, True) self.raise_user_error(cursor, 'wrong_end_balance', error_args=(amount,), context=context) for line in statement.lines: statement_line_obj.create_move(cursor, user, line, context=context) self.write(cursor, user, statement_id, { 'state':'validated', }, context=context) def set_state_posted(self, cursor, user, statement_id, context=None): statement_line_obj = self.pool.get('account.statement.line') statement = self.browse(cursor, user, statement_id, context=context) statement_line_obj.post_move(cursor, user, statement.lines, context=context) self.write(cursor, user, statement_id, { 'state':'posted', }, context=context) def set_state_cancel(self, cursor, user, statement_id, context=None): statement_line_obj = self.pool.get('account.statement.line') statement = self.browse(cursor, user, statement_id, context=context) statement_line_obj.delete_move(cursor, user, statement.lines, context=context) self.write(cursor, user, statement_id, { 'state':'cancel', }, context=context) def draft_workflow(self, cursor, user, ids, context=None): self.workflow_trigger_create(cursor, user, ids, context=context) self.write(cursor, user, ids, { 'state': 'draft', }, context=context) return True Statement() class Line(ModelSQL, ModelView): 'Account Statement Line' _name = 'account.statement.line' _description = __doc__ statement = fields.Many2One('account.statement', 'Statement', required=True, ondelete='CASCADE') date = fields.Date('Date', required=True) amount = fields.Numeric('Amount', required=True, digits="(16, _parent_statement.currency_digits)", on_change=['amount', 'party', 'account', 'invoice', '_parent_statement.journal']) party = fields.Many2One('party.party', 'Party', on_change=['amount', 'party', 'invoice']) account = fields.Many2One('account.account', 'Account', required=True, on_change=['account', 'invoice']) description = fields.Char('Description') move = fields.Many2One('account.move', 'Account Move', readonly=True) invoice = fields.Many2One('account.invoice', 'Invoice', domain=["('party', '=', party)", "('account', '=', account)", "('state', '=', _parent_statement.state == 'draft' and " \ "'open' or False)"], states={ 'readonly': "not bool(amount)", }) def __init__(self): super(Line, self).__init__() self._error_messages.update({ 'debit_credit_account_statement_journal': 'Please provide debit and ' \ 'credit account on statement journal.', 'same_debit_credit_account': 'Credit or debit account on ' \ 'journal is the same than the statement line account!', 'amount_greater_invoice_amount_to_pay': 'Amount (%s) greater than '\ 'the amount to pay of invoice!', }) def on_change_party(self, cursor, user, ids, value, context=None): party_obj = self.pool.get('party.party') account_obj = self.pool.get('account.account') invoice_obj = self.pool.get('account.invoice') res = {} if value.get('party'): party = party_obj.browse(cursor, user, value['party'], context=context) if value.get('amount'): if value['amount'] > Decimal("0.0"): account = party.account_receivable else: account = party.account_payable res['account'] = account.id res['account.rec_name'] = account.rec_name if value.get('invoice'): if value.get('party'): invoice = invoice_obj.browse(cursor, user, value['invoice'], context=context) if invoice.party != value['party']: res['invoice'] = False else: res['invoice'] = False return res def on_change_amount(self, cursor, user, ids, value, context=None): party_obj = self.pool.get('party.party') account_obj = self.pool.get('account.account') invoice_obj = self.pool.get('account.invoice') journal_obj = self.pool.get('account.statement.journal') currency_obj = self.pool.get('currency.currency') res = {} if value.get('party'): party = party_obj.browse(cursor, user, value['party'], context=context) if value.get('account') and value['account'] not in ( party.account_receivable.id, party.account_payable.id): # The user has entered a non-default value, we keep it. pass elif value.get('amount'): if value['amount'] > Decimal("0.0"): account = party.account_receivable else: account = party.account_payable res['account'] = account.id res['account.rec_name'] = account.rec_name if value.get('invoice'): if value.get('amount') and value.get('_parent_statement.journal'): invoice = invoice_obj.browse(cursor, user, value['invoice'], context=context) journal = journal_obj.browse(cursor, user, value['_parent_statement.journal'], context=context) amount_to_pay = currency_obj.compute(cursor, user, invoice.currency, invoice.amount_to_pay, journal.currency, context=context) if abs(value['amount']) > amount_to_pay: res['invoice'] = False else: res['invoice'] = False return res def on_change_account(self, cursor, user, ids, value, context=None): invoice_obj = self.pool.get('account.invoice') res = {} if value.get('invoice'): if value.get('account'): invoice = invoice_obj.browse(cursor, user, value['invoice'], context=context) if invoice.account.id != value['account']: res['invoice'] = False else: res['invoice'] = False return res def create_move(self, cursor, user, line, context=None): ''' Create move for the statement line :param cursor: the database cursor :param user: the user id :param line: a BrowseRecord of the line :param context: the contest :return: the move id ''' move_obj = self.pool.get('account.move') period_obj = self.pool.get('account.period') invoice_obj = self.pool.get('account.invoice') currency_obj = self.pool.get('currency.currency') move_line_obj = self.pool.get('account.move.line') lang_obj = self.pool.get('ir.lang') if context is None: context = {} period_id = period_obj.find(cursor, user, line.statement.journal.company.id, date=line.date, context=context) move_lines = self._get_move_lines(cursor, user, line, context=context) move_id = move_obj.create(cursor, user, { 'name': line.date, 'period': period_id, 'journal': line.statement.journal.journal.id, 'date': line.date, 'lines': [('create', x) for x in move_lines], }, context=context) self.write(cursor, user, line.id, { 'move': move_id, }, context=context) if line.invoice: amount_to_pay = currency_obj.compute(cursor, user, line.invoice.currency, line.invoice.amount_to_pay, line.statement.journal.currency, context=context) if amount_to_pay < abs(line.amount): for code in [context.get('language', False) or 'en_US', 'en_US']: lang_ids = lang_obj.search(cursor, user, [ ('code', '=', code), ], context=context) if lang_ids: break lang = lang_obj.browse(cursor, user, lang_ids[0], context=context) amount = lang_obj.format(lang, '%.' + str(line.statement.journal.currency.digits) + 'f', line.amount, True) self.raise_user_error(cursor, 'amount_greater_invoice_amount_to_pay', error_args=(amount,), context=context) amount = currency_obj.compute(cursor, user, line.statement.journal.currency, line.amount, line.statement.journal.company.currency, context=context) reconcile_lines = invoice_obj.get_reconcile_lines_for_amount(cursor, user, line.invoice, abs(amount)) move = move_obj.browse(cursor, user, move_id, context=context) line_id = None for move_line in move.lines: if move_line.account.id == line.invoice.account.id: line_id = move_line.id invoice_obj.write(cursor, user, line.invoice.id, { 'payment_lines': [('add', line_id)], }, context=context) break if reconcile_lines[1] == Decimal('0.0'): line_ids = reconcile_lines[0] + [line_id] move_line_obj.reconcile(cursor, user, line_ids, context=context) return move_id def post_move(self, cursor, user, lines, context=None): move_obj = self.pool.get('account.move') move_obj.post(cursor, user, [l.move.id for l in lines if l.move], context=context) def delete_move(self, cursor, user, lines, context=None): move_obj = self.pool.get('account.move') move_obj.delete(cursor, user, [l.move.id for l in lines if l.move], context=context) def _get_move_lines(self, cursor, user, statement_line, context=None): ''' Return the values of the move lines for the statement line :param cursor: the database cursor :param user: the user id :param statement_line: a BrowseRecord of the statement line :param context: the context :return: a list of dictionary of move line values ''' currency_obj = self.pool.get('currency.currency') zero = Decimal("0.0") amount = currency_obj.compute( cursor, user, statement_line.statement.journal.currency, statement_line.amount, statement_line.statement.journal.company.currency, context=context) if statement_line.statement.journal.currency.id != \ statement_line.statement.journal.company.currency.id: second_currency = statement_line.statement.journal.currency.id amount_second_currency = abs(statement_line.amount) else: amount_second_currency = False second_currency = None vals = [] vals.append({ 'name': statement_line.date, 'debit': amount < zero and -amount or zero, 'credit': amount >= zero and amount or zero, 'account': statement_line.account.id, 'party': statement_line.party and statement_line.party.id, 'second_currency': second_currency, 'amount_second_currency': amount_second_currency, }) journal = statement_line.statement.journal.journal if statement_line.amount >= zero: account = journal.credit_account else: account = journal.debit_account if not account: self.raise_user_error(cursor, 'debit_credit_account_statement_journal', context=context) if statement_line.account.id == account.id: self.raise_user_error(cursor, 'same_debit_credit_account', context=context) vals.append({ 'name': statement_line.date, 'debit': amount >= zero and amount or zero, 'credit': amount < zero and -amount or zero, 'account': account.id, 'party': statement_line.party and statement_line.party.id, 'second_currency': second_currency, 'amount_second_currency': amount_second_currency, }) return vals Line()