From 5a2953ffc37fa63379cb556ed1477e75557577ee Mon Sep 17 00:00:00 2001 From: richard-dds Date: Tue, 14 Aug 2018 11:41:37 -0400 Subject: [PATCH] Split request update into new method for easier extension --- atst/routes/requests/jedi_request_flow.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/atst/routes/requests/jedi_request_flow.py b/atst/routes/requests/jedi_request_flow.py index d69f1928..79506916 100644 --- a/atst/routes/requests/jedi_request_flow.py +++ b/atst/routes/requests/jedi_request_flow.py @@ -122,9 +122,13 @@ class JEDIRequestFlow(object): ] def create_or_update_request(self): - request_data = {self.form_section: self.form.data} if self.request_id: - Requests.update(self.request_id, request_data) + self.update_request(self.form_section, self.form.data) else: + request_data = {self.form_section: self.form.data} request = Requests.create(self.current_user, request_data) self.request_id = request.id + + def update_request(self, section, data): + request_data = {section: data} + Requests.update(self.request_id, request_data)