Updated functionalities

app/routes/auth.py

@@ -171,7 +171,9 @@ def create_user():
 @login_required
 @supervisor_required
 def edit_user(user_id):
-    user = User.query.get_or_404(user_id)
+    user = db.session.get(User, user_id)
+    if user is None:
+        abort(404)
     form = UserForm(user=user, obj=user)
 
     if form.validate_on_submit():
@@ -198,7 +200,9 @@ def edit_user(user_id):
 @login_required
 @supervisor_required
 def delete_user(user_id):
-    user = User.query.get_or_404(user_id)
+    user = db.session.get(User, user_id)
+    if user is None:
+        abort(404)
 
     if user.id == current_user.id:
         flash('Cannot delete your own account.', 'danger')
@@ -229,7 +233,9 @@ def delete_user(user_id):
 @login_required
 @supervisor_required
 def toggle_active(user_id):
-    user = User.query.get_or_404(user_id)
+    user = db.session.get(User, user_id)
+    if user is None:
+        abort(404)
 
     if user.id == current_user.id:
         flash('You cannot disable your own account.', 'danger')