From 5af725f58d954154779806a2667d31ee74723b43 Mon Sep 17 00:00:00 2001
From: dsc <dsc@xmr.pm>
Date: Sun, 13 Mar 2022 09:56:21 +0200
Subject: [PATCH] return 404 when user cannot be found

---
 yellow/routes.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/yellow/routes.py b/yellow/routes.py
index ae5723d..233f90e 100644
--- a/yellow/routes.py
+++ b/yellow/routes.py
@@ -75,10 +75,13 @@ async def user_page(name: str):
     if not name or len(name) <= 1:
         raise Exception("invalid name")
 
-    _user = User.select().where(
-        User.username == name,
-        User.address.is_null(False)
-    ).get()
+    try:
+        _user = User.select().where(
+            User.username == name,
+            User.address.is_null(False)
+        ).get()
+    except:
+        return abort(404)
 
     return await render_template('user.html', users=[_user])