Refine logic adding new ID
This commit is contained in:
5
app.py
5
app.py
@@ -78,8 +78,7 @@ def FUN_delete_user(id):
|
||||
|
||||
@app.route("/add_user", methods = ["POST"])
|
||||
def FUN_add_user():
|
||||
if session.get("current_user", None) == "ADMIN":
|
||||
|
||||
if session.get("current_user", None) == "ADMIN": # only Admin should be able to add user.
|
||||
# before we add the user, we need to ensure this is doesn't exsit in database. We also need to ensure the id is valid.
|
||||
if request.form.get('id').upper() in list_users():
|
||||
user_list = list_users()
|
||||
@@ -87,7 +86,7 @@ def FUN_add_user():
|
||||
user_list,\
|
||||
[x + y for x,y in zip(["/delete_user/"] * len(user_list), user_list)])
|
||||
return(render_template("admin.html", id_is_duplicated = True, users = user_table))
|
||||
if " " in request.form.get('id'):
|
||||
if " " in request.form.get('id') or "'" in request.form.get('id'):
|
||||
user_list = list_users()
|
||||
user_table = zip(range(1, len(user_list)+1),\
|
||||
user_list,\
|
||||
|
||||
Reference in New Issue
Block a user