Files
account-reconcile/account_statement_cancel_line/statement.py

98 lines
4.2 KiB
Python

# -*- 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.
"""
_inherit = "account.bank.statement"
def button_confirm_bank(self, cr, uid, ids, context=None):
"""Change the state on the statement lines. Return super."""
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': 'confirmed'
}, context=context)
return super(Statement, self).button_confirm_bank(
cr, uid, ids, context)
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):
if st_line_obj.has_reconciliation(
cr,
uid,
[line.id for line in statement.line_ids],
context=context):
# ask confirmation, we have some reconciliation already
return {
'type': 'ir.actions.act_window',
'res_model': 'wizard.cancel.statement',
'view_type': 'form',
'view_mode': 'form',
'target': 'new',
'context': context,
}
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
return need_to_update_view