statement.py

# -*- coding: utf-8 -*-
###############################################################################
#                                                                             #
#   Author: Leonardo Pistone
#   Copyright 2014 Camptocamp SA
#                                                                             #
#   This program is free software: you can redistribute it and/or modify      #
#   it under the terms of the GNU Affero General Public License as            #
#   published by the Free Software Foundation, either version 3 of the        #
#   License, or (at your option) any later version.                           #
#                                                                             #
#   This program is distributed in the hope that it will be useful,           #
#   but WITHOUT ANY WARRANTY; without even the implied warranty of            #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the             #
#   GNU Affero General Public License for more details.                       #
#                                                                             #
#   You should have received a copy of the GNU Affero General Public License  #
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.     #
#                                                                             #
###############################################################################
"""Account Statement Cancel Line."""

from openerp.osv import orm


class Statement(orm.Model):

    """Bank Statement.

    Minimal changes to allow cancelling single lines and checking if there are
    any lines that are already reconciled.

    """

    _inherit = "account.bank.statement"

    def button_confirm_bank(self, cr, uid, ids, context=None):
        """If all lines are draft, change their state.
        Otherwise, confirm line by line to avoid duplicate moves.
        Return super.

        """
        st_line_obj = self.pool['account.bank.statement.line']

        statement_ids_fully_confirm = []
        for st in self.browse(cr, uid, ids, context=context):
            if all(l.state == 'draft' for l in st.line_ids):
                statement_ids_fully_confirm.append(st.id)
                st_line_obj.write(cr, uid, [l.id for l in st.line_ids], {
                    'state': 'confirmed'
                }, context=context)
            else:
                st_line_obj.confirm(cr, uid, [l.id for l in st.line_ids],
                                    context=context)

        if statement_ids_fully_confirm:
            return super(Statement, self).button_confirm_bank(
                cr, uid, statement_ids_fully_confirm, context)
        else:
            return True

    def button_cancel(self, cr, uid, ids, context=None):
        """Check if there is any reconciliation. Return action."""
        st_line_obj = self.pool['account.bank.statement.line']
        for statement in self.browse(cr, uid, ids, context=context):
            ctx = context.copy()
            ctx['default_reconcile_warning'] = st_line_obj.has_reconciliation(
                cr,
                uid,
                [line.id for line in statement.line_ids],
                context=context)
            return {
                'type': 'ir.actions.act_window',
                'res_model': 'wizard.cancel.statement',
                'view_type': 'form',
                'view_mode': 'form',
                'target': 'new',
                'context': ctx,
            }

        self.do_cancel(cr, uid, ids, context=context)

    def do_cancel(self, cr, uid, ids, context=None):
        """Change the state on the statement lines. Return super.

        This method is called directly when there are no reconciliations, or
        from the warning wizard, if there are reconciliations.

        """
        st_line_obj = self.pool['account.bank.statement.line']
        for st_data in self.read(cr, uid, ids, ['line_ids'], context=context):
            st_line_obj.write(cr, uid, st_data['line_ids'], {
                'state': 'draft'
            }, context=context)

        return super(Statement, self).button_cancel(
            cr, uid, ids, context)

    def confirm_statement_from_lines(self, cr, uid, ids, context=None):
        """If all lines are confirmed, so is the whole statement.

        Return True if we changed anything.

        """
        need_to_update_view = False
        for statement in self.browse(cr, uid, ids, context=context):
            if all(line.state == 'confirmed' for line in statement.line_ids):
                self.write(cr, uid, [statement.id], {
                    'state': 'confirm'
                }, context=context)
                need_to_update_view = True
                self.balance_check(
                    cr,
                    uid,
                    statement.id,
                    journal_type=statement.journal_id.type,
                    context=context)
        return need_to_update_view