diff --git a/addons/account_cancel/models/account_bank_statement.py b/addons/account_cancel/models/account_bank_statement.py index 821014abf37..3e8fb178f2e 100644 --- a/addons/account_cancel/models/account_bank_statement.py +++ b/addons/account_cancel/models/account_bank_statement.py @@ -12,12 +12,18 @@ class BankStatement(models.Model): def button_draft(self): self.state = 'draft' + @api.multi + def button_cancel(self): + return super(BankStatement, self.with_context(bank_statement_cancel=True)).button_cancel() + + class BankStatementLine(models.Model): _inherit = 'account.bank.statement.line' @api.multi def cancel(self): - for line in self: - if line.statement_id.state == 'confirm': - raise Warning(_("Please set the bank statement to New before canceling.")) + if not self.env.context.get('bank_statement_cancel'): + for line in self: + if line.statement_id.state == 'confirm': + raise Warning(_("Please set the bank statement to New before canceling.")) return super(BankStatementLine, self).cancel()