From 8ff8bb625c8d656389f17e4db003210b431fb2ed Mon Sep 17 00:00:00 2001
From: XD-DENG <xd_deng@hotmail.com>
Date: Mon, 3 Jul 2017 19:58:46 +0800
Subject: [PATCH] Refine layout template s.t. private/admin page hidden when
 not logged in

---
 templates/layout.html | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/templates/layout.html b/templates/layout.html
index af0f79c..6a14379 100644
--- a/templates/layout.html
+++ b/templates/layout.html
@@ -21,8 +21,12 @@
         <ul class="nav navbar-nav">
             <li><a href="/">Home</a></li>
             <li><a href="/public/">Public</a></li>
-            <li><a href="/private/">Private</a></li>
-            <li><a href="/admin">Admin Dashboard</a></li>
+            {% if session.get("current_user", None) != None %}
+	       <li><a href="/private/">Private</a></li>
+	    {% endif%}
+            {% if session.get("current_user", None) == "ADMIN" %}
+	      <li><a href="/admin">Admin Dashboard</a></li>
+	    {% endif %}
         </ul>
         
         <ul class="nav navbar-nav navbar-right">