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])