From 48b0d4194e58965164e4bd8e95f6a0928442dfdd Mon Sep 17 00:00:00 2001 From: Chris Hill-Scott Date: Mon, 5 Nov 2018 16:17:44 +0000 Subject: [PATCH] Use model to toggle research mode Just a nice bit of encapsulation, rather than passing `current_service` through to a method on `current_service`. --- app/main/views/service_settings.py | 4 +--- app/models/service.py | 5 +++++ 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/app/main/views/service_settings.py b/app/main/views/service_settings.py index c65df9920..09909769b 100644 --- a/app/main/views/service_settings.py +++ b/app/main/views/service_settings.py @@ -221,9 +221,7 @@ def service_switch_live(service_id): @login_required @user_is_platform_admin def service_switch_research_mode(service_id): - current_service.update_with_properties( - {'research_mode': not current_service.research_mode} - ) + current_service.toggle_research_mode() return redirect(url_for('.service_settings', service_id=service_id)) diff --git a/app/models/service.py b/app/models/service.py index 4893d9275..b34dc95b0 100644 --- a/app/models/service.py +++ b/app/models/service.py @@ -78,6 +78,11 @@ class Service(): self.update_with_properties(data) + def toggle_research_mode(self): + self.update_with_properties({ + 'research_mode': not self.research_mode, + }) + @property def trial_mode(self): return self._dict['restricted']