fix: add missing ValueError handling

simple-login · Nov 18, 2024 · 92eba9e · 92eba9e
1 parent 7172325
commit 92eba9e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/dashboard/views/setting.py b/app/dashboard/views/setting.py
@@ -174,7 +174,12 @@ def setting():
             flash("Your preference has been updated", "success")
             return redirect(url_for("dashboard.setting"))
         elif request.form.get("form-name") == "random-alias-suffix":
-            scheme = int(request.form.get("random-alias-suffix-generator"))
+            try:
+                scheme = int(request.form.get("random-alias-suffix-generator"))
+            except ValueError:
+                flash("Invalid value", "error")
+                return redirect(url_for("dashboard.setting"))
+
             if AliasSuffixEnum.has_value(scheme):
                 current_user.random_alias_suffix = scheme
                 Session.commit()