From 997a7b55920b47eff4c6326ad86df1af40391616 Mon Sep 17 00:00:00 2001
From: Adam Dodman <adam.dodman@gmx.com>
Date: Thu, 5 Oct 2017 01:22:55 +0100
Subject: [PATCH] Dynamically load the User database

---
 main.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index e619528..20c3aca 100644
--- a/main.py
+++ b/main.py
@@ -16,6 +16,17 @@ def cleanup_string(text):
     text = text.encode("ascii", "replace").decode()
     return text.strip() 
 
+def get_userdb():
+    global userdb
+    global numusers
+
+    # Counting on primary key is quickest operation
+    cur_num = mysql_do("SELECT count(`uid`) FROM `Users`;")[0][0]
+
+    if numusers < cur_num:
+        # Update the database!
+        userdb = mysql_do("SELECT * FROM Users")
+    return userdb
 
 
 # Load User Table into variable
@@ -37,9 +48,10 @@ def app_init():
     # Generate random key for session cookies
     app.secret_key = rand(24)
 
-    # TODO: Replace this with something dynamic where needed
-    global userdb
-    userdb = mysql_do("SELECT * FROM Users")
+    # Init the counter, then run a query
+    global numusers
+    numusers = 0
+    get_userdb()
 
 
 def do_user_login(user, password):
@@ -137,7 +149,7 @@ def addquote():
     except KeyError:
         flash("INFO: Please login first.","info")
         return redirect(url_for("login"))
-    return render_template("add_quote.html", users=userdb)
+    return render_template("add_quote.html", users=get_userdb())
 
 
 @app.route("/logout")