AdamAnt/QuotesDB
1
0
Fork 0
mirror of https://github.com/Adam-Ant/QuotesDB synced 2024-12-20 11:34:35 +00:00
Code Issues Releases Wiki Activity

Dynamically load the User database

Browse Source
This commit is contained in:
Adam Dodman 2017-10-05 01:22:55 +01:00
parent 517e8a5cfa
commit 997a7b5592
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
main.py
View File

@ -16,6 +16,17 @@ def cleanup_string(text):
text = text.encode("ascii", "replace").decode() text = text.encode("ascii", "replace").decode()
return text.strip() 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 # Load User Table into variable
@ -37,9 +48,10 @@ def app_init():
# Generate random key for session cookies # Generate random key for session cookies
app.secret_key = rand(24) app.secret_key = rand(24)
# TODO: Replace this with something dynamic where needed # Init the counter, then run a query
global userdb global numusers
userdb = mysql_do("SELECT * FROM Users") numusers = 0
get_userdb()
def do_user_login(user, password): def do_user_login(user, password):
@ -137,7 +149,7 @@ def addquote():
except KeyError: except KeyError:
flash("INFO: Please login first.","info") flash("INFO: Please login first.","info")
return redirect(url_for("login")) 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") @app.route("/logout")