python/fastapi-admin
1
0
Fork 0
mirror of https://github.com/fastapi-admin/fastapi-admin.git synced 2026-03-13 10:32:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make get_current_admin error to 401 and add 401 error page.

Browse Source
This commit is contained in:
long2ice
2021-08-31 11:27:23 +08:00
parent 2fcbc814de
commit e99d673191
5 changed files with 44 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastapi_admin/depends.py
View File

@@ -3,7 +3,7 @@ from typing import List, Optional, Type
from fastapi import Depends, HTTPException
from fastapi.params import Path
from starlette.requests import Request
from starlette.status import HTTP_404_NOT_FOUND
from starlette.status import HTTP_401_UNAUTHORIZED, HTTP_404_NOT_FOUND
from tortoise import Tortoise
from fastapi_admin.exceptions import InvalidResource
@@ -68,5 +68,5 @@ def get_redis(request: Request):
def get_current_admin(request: Request):
admin = request.state.admin
if not admin:
raise HTTPException(status_code=HTTP_404_NOT_FOUND)
raise HTTPException(status_code=HTTP_401_UNAUTHORIZED)
return admin