diff --git a/statement.py b/statement.py index 78a9788e17..8a4d8de00d 100644 --- a/statement.py +++ b/statement.py @@ -260,12 +260,10 @@ class Statement(ModelWorkflow, ModelSQL, ModelView): amount_to_pay - abs(line['amount']) return res - def set_state_validated(self, statement_id): + def wkf_validated(self, statement): statement_line_obj = Pool().get('account.statement.line') lang_obj = Pool().get('ir.lang') - statement = self.browse(statement_id) - computed_end_balance = statement.start_balance for line in statement.lines: computed_end_balance += line.amount @@ -284,25 +282,23 @@ class Statement(ModelWorkflow, ModelSQL, ModelView): self.raise_user_error('wrong_end_balance', error_args=(amount,)) for line in statement.lines: statement_line_obj.create_move(line) - self.write(statement_id, { + self.write(statement.id, { 'state':'validated', }) - def set_state_posted(self, statement_id): + def wkf_posted(self, statement): statement_line_obj = Pool().get('account.statement.line') - statement = self.browse(statement_id) statement_line_obj.post_move(statement.lines) - self.write(statement_id, { + self.write(statement.id, { 'state':'posted', }) - def set_state_cancel(self, statement_id): + def wkf_cancel(self, statement): statement_line_obj = Pool().get('account.statement.line') - statement = self.browse(statement_id) statement_line_obj.delete_move(statement.lines) - self.write(statement_id, { + self.write(statement.id, { 'state':'cancel', }) diff --git a/statement.xml b/statement.xml index 3e00ef6730..7a0ad34280 100644 --- a/statement.xml +++ b/statement.xml @@ -216,21 +216,18 @@ this repository contains the full copyright notices and license terms. --> - function - set_state_validated() + wkf_validated Validated - function - set_state_posted() + wkf_posted Posted - function - set_state_cancel() + wkf_cancel Canceled