From 00038396341a80b3c1a627f36de165c70ea686d6 Mon Sep 17 00:00:00 2001 From: "mw-middleware-labs-sandbox[bot]" <224139880+mw-middleware-labs-sandbox[bot]@users.noreply.github.com> Date: Wed, 12 Aug 2026 05:24:56 +0000 Subject: [PATCH] fix: return 404 for unknown username in /user/ instead of unhandled KeyError --- flask/app.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/flask/app.py b/flask/app.py index c30b439..bb640b4 100644 --- a/flask/app.py +++ b/flask/app.py @@ -7,7 +7,7 @@ ) ) -from flask import Flask, render_template, request, jsonify +from flask import Flask, render_template, request, jsonify, abort import logging import os from datetime import datetime @@ -60,9 +60,9 @@ def generate_exception(): @app.route('/user/') def user_profile(username): print(f"User profile requested for {username}") - test = user_data[username] + test = user_data.get(username) if not test: - raise + abort(404, description=f"User '{username}' not found") return jsonify({"message": f"Profile for {username}", "data": test}) @app.route('/process', methods=['POST'])