Fix workout deleting option

1 files changed, 17 insertions, 0 deletions

diff --git a/app/routes/api.py b/app/routes/api.py
new file mode 100644
index 0000000..58b4732
--- /dev/null
+++ b/app/routes/api.py
@@ -0,0 +1,17 @@
+from app import app, db
+from flask import redirect, request
+from flask_login import current_user, login_required
+from tables import Workout
+
+@app.route("/api/workout/delete", methods=['GET'])
+@login_required
+def api_workout_delete():
+    if 'id' not in request.args:
+        return redirect('/home');
+
+    workout = Workout.query.filter_by(id=int(request.args['id']), user_id=current_user.id).first()
+    if workout:
+        db.session.delete(workout)
+        db.session.commit()
+
+    return redirect('/home')