diff --git a/server.py b/server.py index 46e143a..b8c3bd6 100644 --- a/server.py +++ b/server.py @@ -20,17 +20,29 @@ def show_all(): @app.post("/") def create(): if 'parent_id' in request.form: - cursor.execute('INSERT INTO items (title, description,parent_id) VALUES(%s,%s,%s)', [request.form['title'],request.form['description'],int(request.form['parent_id'])]) + cursor.execute( + 'INSERT INTO items (title, description,parent_id) VALUES(%s,%s,%s)', + [request.form['title'],request.form['description'],int(request.form['parent_id'])] + ) return redirect('/'+request.form['parent_id']) else: - cursor.execute('INSERT INTO items (title, description) VALUES(%s,%s)', [request.form['title'],request.form['description']]) + cursor.execute( + 'INSERT INTO items (title, description) VALUES(%s,%s)', + [request.form['title'],request.form['description']] + ) return redirect('/') @app.post("/") -def delete(id): - # print(request.args.get('_method')) +def deleteUpdate(id): + print(request.args.get('_method')) # TODO: deal with deleting item that has children - cursor.execute('DELETE FROM items WHERE id=%s LIMIT 1', [id]) + if request.args.get('_method') == 'DELETE': + cursor.execute('DELETE FROM items WHERE id=%s LIMIT 1', [id]) + else: + cursor.execute( + 'UPDATE items SET title=%s, description=%s, parent_id=%s WHERE id=%s', + [request.form['title'], request.form['description'], request.form['parent_id'] if request.form['parent_id'] != '' else None, id] + ) if 'parent_id' in request.form: return redirect('/'+request.form['parent_id']) else: