From 8a945f8bafc45ea5a122c117b8cac7557829dd21 Mon Sep 17 00:00:00 2001 From: Lance Edgar Date: Sat, 23 Sep 2017 16:26:56 -0500 Subject: [PATCH] Pass form along to `before_create_flush()` in master3 --- tailbone/views/master3.py | 4 ++-- tailbone/views/upgrades.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tailbone/views/master3.py b/tailbone/views/master3.py index c25e60a6..d9e9ac3b 100644 --- a/tailbone/views/master3.py +++ b/tailbone/views/master3.py @@ -155,12 +155,12 @@ class MasterView3(MasterView2): def save_create_form(self, form): self.before_create(form) obj = form.schema.objectify(self.form_deserialized) - self.before_create_flush(obj) + self.before_create_flush(obj, form) self.Session.add(obj) self.Session.flush() return obj - def before_create_flush(self, obj): + def before_create_flush(self, obj, form): pass def save_edit_form(self, form): diff --git a/tailbone/views/upgrades.py b/tailbone/views/upgrades.py index 7475d0e1..f2751701 100644 --- a/tailbone/views/upgrades.py +++ b/tailbone/views/upgrades.py @@ -303,7 +303,7 @@ class UpgradeView(MasterView): def download_content_type(self, path, filename): return 'text/plain' - def before_create_flush(self, upgrade): + def before_create_flush(self, upgrade, form): upgrade.created_by = self.request.user upgrade.status_code = self.enum.UPGRADE_STATUS_PENDING