diff --git a/shop.py b/shop.py index 1e6d97d..b901858 100644 --- a/shop.py +++ b/shop.py @@ -110,13 +110,27 @@ def get_items(user_id, list_id=None): def get_list_ids(user_id): ret = {} - query = "select USER_META.list_id, LISTS.name from USER_META inner join LISTS on LISTS.id = USER_META.list_id where USER_META.user_id = %s" - data = (user_id,) - res = runQuery(query, data) + if user_id == "admin": + query = "SELECT id, name from LISTS;" + res = runQuery(query) + else: + query = "select USER_META.list_id, LISTS.name from USER_META inner join LISTS on LISTS.id = USER_META.list_id where USER_META.user_id = %s" + data = (user_id,) + res = runQuery(query, data) for lid, name in res: ret[lid] = name return ret +def addList(list_name): + query = "insert into LISTS (name) values (%s)" + data = (list_name,) + runQuery(query, data) + +def addUserToList(userid, listid): + query = "insert into USER_META (user_id, list_id) values (%s, %s)" + data = (userid, listid) + runQuery(query, data) + application = Flask(__name__) application.config["DEBUG"] = True application.secret_key = b'*$#@U9423jr92jioJKL_)_;dasfj()12' @@ -140,6 +154,7 @@ def index(): @application.route('/post', methods=['POST']) def handle_data(): + print(request.form) if "addValue" in request.form: for x in request.form: if request.form[x] == '': @@ -186,6 +201,15 @@ def handle_data(): if "list" in request.form: session["active_id"] = request.form['list'] + + if "addList" in request.form: + print(request.form["addList"]) + addList(request.form["addList"]) + + if "add2list" in request.form: + userid = request.form["add2list"] + listid = request.form["listID"] + addUserToList(userid, listid) return redirect(url_for('index')) @@ -199,7 +223,8 @@ def admin(): #get a list of users userList = get_users() - data = {"users": userList} + lists = get_list_ids("admin") + data = {"users": userList, "lists": lists} return render_template("admin.html", data=data) diff --git a/templates/admin.html b/templates/admin.html index 404893e..f77e222 100644 --- a/templates/admin.html +++ b/templates/admin.html @@ -42,6 +42,20 @@ +
| Username | +User | +
|---|---|
| {{data["lists"][x]}} | ++ |