From 051cb7195619bd8a50fa2e3d2da86e0e2c069117 Mon Sep 17 00:00:00 2001 From: Lance Edgar Date: Wed, 9 Jan 2019 14:18:30 -0600 Subject: [PATCH] Let batch view customize logic for marking batch as (in)complete --- tailbone/views/batch/core.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tailbone/views/batch/core.py b/tailbone/views/batch/core.py index c3f52b99..b1b20f36 100644 --- a/tailbone/views/batch/core.py +++ b/tailbone/views/batch/core.py @@ -491,17 +491,26 @@ class BatchMasterView(MasterView): else: form = forms.Form(schema=ToggleComplete(), request=self.request) if form.validate(newstyle=True): - batch.complete = form.validated['complete'] + if form.validated['complete']: + self.mark_batch_complete(batch) + else: + self.mark_batch_incomplete(batch) return self.redirect(self.get_action_url('view', batch)) + def mark_batch_complete(self, batch): + batch.complete = True + + def mark_batch_incomplete(self, batch): + batch.complete = False + def mobile_mark_complete(self): batch = self.get_instance() - batch.complete = True + self.mark_batch_complete(batch) return self.redirect(self.get_index_url(mobile=True)) def mobile_mark_pending(self): batch = self.get_instance() - batch.complete = False + self.mark_batch_incomplete(batch) return self.redirect(self.get_action_url('view', batch, mobile=True)) def rows_creatable_for(self, batch):