use return

This commit is contained in:
2025-03-09 21:44:35 +08:00
parent fa3dd701ea
commit 54b337173a
3 changed files with 10 additions and 9 deletions

View File

@@ -11,12 +11,12 @@ def _require(roles):
@wraps(func)
def wrapper(request, *args, **kwargs):
if not request.user.is_authenticated:
raise HttpError(401, "用户未登录")
return HttpError(401, "用户未登录")
try:
if request.user.role not in roles:
raise HttpError(403, "你没有权限")
return HttpError(403, "你没有权限")
except User.DoesNotExist:
raise HttpError(404, "用户不存在")
return HttpError(404, "用户不存在")
return func(request, *args, **kwargs)
return wrapper