use return

This commit is contained in:
xuyue committed 2025-03-09 21:44:35 +08:00
1 parent fa3dd701ea
commit 54b337173a
3 files changed
+10 -9

No files matched your search

+3 -3
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