From 8656fe16a781ab0cf60e012a85ced1f091862c6f 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, 25 Mar 2026 20:14:41 +0000 Subject: [PATCH] refactor: Refactor user_profile endpoint for safe user retrieval and proper error handling --- flask/app.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/flask/app.py b/flask/app.py index c30b439..1bbd398 100644 --- a/flask/app.py +++ b/flask/app.py @@ -60,10 +60,10 @@ def generate_exception(): @app.route('/user/') def user_profile(username): print(f"User profile requested for {username}") - test = user_data[username] - if not test: - raise - return jsonify({"message": f"Profile for {username}", "data": test}) + user = user_data.get(username) + if not user: + return jsonify({"error": f"User '{username}' not found"}), 404 + return jsonify({"message": f"Profile for {username}", "data": user}) @app.route('/process', methods=['POST']) def process_user_data():