26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
from fastapi import Request
|
|
from starlette import status
|
|
from starlette.exceptions import HTTPException
|
|
|
|
import schemas
|
|
from chalicelib.core import projects
|
|
from or_dependencies import OR_context
|
|
|
|
|
|
class ProjectAuthorizer:
|
|
def __init__(self, project_identifier):
|
|
self.project_identifier: str = project_identifier
|
|
|
|
async def __call__(self, request: Request) -> None:
|
|
if len(request.path_params.keys()) == 0 or request.path_params.get(self.project_identifier) is None:
|
|
return
|
|
current_user: schemas.CurrentContext = await OR_context(request)
|
|
value = request.path_params[self.project_identifier]
|
|
if (self.project_identifier == "projectId" \
|
|
and (not (isinstance(value, int) or isinstance(value, str) and value.isnumeric())
|
|
or projects.get_project(project_id=value, tenant_id=current_user.tenant_id) is None)) \
|
|
or (self.project_identifier == "projectKey" \
|
|
and projects.get_internal_project_id(project_key=value) is None):
|
|
print("project not found")
|
|
print(value)
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="project not found.")
|