Skip to content

Commit

Permalink
remove manual auth checking, 'prebuilt' middleware
Browse files Browse the repository at this point in the history
  • Loading branch information
Chats committed Apr 30, 2024
1 parent 5118b7c commit e07125b
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 45 deletions.
77 changes: 33 additions & 44 deletions src/routes/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,59 +3,48 @@

import src.utils.log as log
from src.cfg.settings import ahgs_api as api_cfg
from src.utils.middleware.authentication import verify_password

router = APIRouter(
prefix="/admin",
tags=["admin"],
include_in_schema=True,
responses={401: {"description": "Unauthorized"}},
include_in_schema=False,
)


@router.get("/settings", include_in_schema=False)
@router.get("/settings")
async def settings(request: Request, source: str = None):
if verify_password(request.headers.get("Authorization")):
log.info(request, status.HTTP_200_OK, source)
return {
"log_level": log.logger.level,
"session_token": api_cfg["auth_headers"].get("Authorization", ""),
}
log.info(request, status.HTTP_401_UNAUTHORIZED, source)
raise HTTPException(status_code=401, detail="Unauthorized")
log.info(request, status.HTTP_200_OK, source)
return {
"log_level": log.logger.level,
"session_token": api_cfg["auth_headers"].get("Authorization", ""),
}


@router.post("/session", include_in_schema=False)
@router.post("/session")
async def settings(request: Request, source: str = None):
if verify_password(request.headers.get("Authorization")):
request_json = await request.json()
token = request_json["token"]
if token:
api_cfg["auth_headers"]["Authorization"] = f"{token}"
log.info(request, status.HTTP_202_ACCEPTED, source)
raise HTTPException(status_code=status.HTTP_202_ACCEPTED)
else:
log.info(request, status.HTTP_400_BAD_REQUEST, source)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, detail="Did not provide token"
)
log.info(request, status.HTTP_401_UNAUTHORIZED, source)
raise HTTPException(status_code=401, detail="Unauthorized")


@router.post("/loglevel", include_in_schema=False)
request_json = await request.json()
token = request_json["token"]
if token:
api_cfg["auth_headers"]["Authorization"] = f"{token}"
log.info(request, status.HTTP_202_ACCEPTED, source)
raise HTTPException(status_code=status.HTTP_202_ACCEPTED)
else:
log.info(request, status.HTTP_400_BAD_REQUEST, source)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, detail="Did not provide token"
)


@router.post("/loglevel")
async def settings(request: Request, source: str = None):
if verify_password(request.headers.get("Authorization")):
request_json = await request.json()
loglevel_in = request_json["loglevel"]
if loglevel_in:
log.update_log_level(log_level="loglevel_in")
log.info(request, status.HTTP_202_ACCEPTED, source)
raise HTTPException(status_code=status.HTTP_202_ACCEPTED)
else:
log.info(request, status.HTTP_400_BAD_REQUEST, source)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, detail="Did not provide token"
)
log.info(request, status.HTTP_401_UNAUTHORIZED, source)
raise HTTPException(status_code=401, detail="Unauthorized")
request_json = await request.json()
loglevel_in = request_json["loglevel"]
if loglevel_in:
log.update_log_level(log_level="loglevel_in")
log.info(request, status.HTTP_202_ACCEPTED, source)
raise HTTPException(status_code=status.HTTP_202_ACCEPTED)
else:
log.info(request, status.HTTP_400_BAD_REQUEST, source)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, detail="Did not provide token"
)
1 change: 0 additions & 1 deletion src/utils/middleware/prebuilt.py

This file was deleted.

0 comments on commit e07125b

Please sign in to comment.