From 091553be566d97816ed898207b56d2694eb2efdd Mon Sep 17 00:00:00 2001
From: 0livd <0livd@users.noreply.github.com>
Date: Tue, 16 May 2017 23:21:41 +0100
Subject: Add ACTIVATE_DEMO_PROJECT setting (#209)

When set to False (True by default), it deactivates
the demo project
---
 budget/web.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'budget/web.py')

diff --git a/budget/web.py b/budget/web.py
index 18867c5..efb427c 100644
--- a/budget/web.py
+++ b/budget/web.py
@@ -121,8 +121,11 @@ def authenticate(project_id=None):
 def home():
     project_form = ProjectForm()
     auth_form = AuthenticationForm()
+    is_demo_project_activated = current_app.config['ACTIVATE_DEMO_PROJECT']
+
     return render_template("home.html", project_form=project_form,
-            auth_form=auth_form, session=session)
+                           is_demo_project_activated=is_demo_project_activated,
+                           auth_form=auth_form, session=session)
 
 
 @main.route("/create", methods=["GET", "POST"])
@@ -258,9 +261,15 @@ def demo():
     the bills list for this project.
 
     Create a demo project if it doesnt exists yet (or has been deleted)
+    If the demo project is deactivated, one is redirected to the create project form
     """
+    is_demo_project_activated = current_app.config['ACTIVATE_DEMO_PROJECT']
     project = Project.query.get("demo")
-    if not project:
+
+    if not project and not is_demo_project_activated:
+        raise Redirect303(url_for(".create_project",
+                                  project_id='demo'))
+    if not project and is_demo_project_activated:
         project = Project(id="demo", name=u"demonstration", password="demo",
                 contact_email="demo@notmyidea.org")
         db.session.add(project)
-- 
cgit v1.1