diff --git a/ereuse_devicehub/inventory/views.py b/ereuse_devicehub/inventory/views.py index b2b413e5..1e072eb8 100644 --- a/ereuse_devicehub/inventory/views.py +++ b/ereuse_devicehub/inventory/views.py @@ -311,11 +311,9 @@ class NewActionView(View): def dispatch_request(self): self.form = self.form_class() - # import pdb; pdb.set_trace() if self.form.validate_on_submit(): - instance = self.form.save() - messages.success('Action "{}" created successfully!'.format(instance.type)) + messages.success('Action "{}" created successfully!'.format(self.form.type.data)) next_url = self.get_next_url() return flask.redirect(next_url) @@ -337,8 +335,7 @@ class NewAllocateView(NewActionView, DeviceListMix): self.form = self.form_class() if self.form.validate_on_submit(): - instance = self.form.save() - messages.success('Action "{}" created successfully!'.format(instance.type)) + messages.success('Action "{}" created successfully!'.format(self.form.type.data)) next_url = self.get_next_url() return flask.redirect(next_url) @@ -357,8 +354,7 @@ class NewDataWipeView(NewActionView, DeviceListMix): self.form = self.form_class() if self.form.validate_on_submit(): - instance = self.form.save() - messages.success('Action "{}" created successfully!'.format(instance.type)) + messages.success('Action "{}" created successfully!'.format(self.form.type.data)) next_url = self.get_next_url() return flask.redirect(next_url) @@ -377,8 +373,7 @@ class NewTradeView(NewActionView, DeviceListMix): self.form = self.form_class() if self.form.validate_on_submit(): - instance = self.form.save() - messages.success('Action "{}" created successfully!'.format(instance.type)) + messages.success('Action "{}" created successfully!'.format(self.form.type.data)) next_url = self.get_next_url() return flask.redirect(next_url)