PhotosAPI/extensions/exceptions.py

from fastapi import Request
from fastapi.responses import UJSONResponse
from starlette.status import (
    HTTP_400_BAD_REQUEST,
    HTTP_401_UNAUTHORIZED,
    HTTP_403_FORBIDDEN,
    HTTP_404_NOT_FOUND,
    HTTP_406_NOT_ACCEPTABLE,
    HTTP_409_CONFLICT,
    HTTP_422_UNPROCESSABLE_ENTITY,
)

from classes.exceptions import (
    AccessTokenInvalidError,
    AlbumAlreadyExistsError,
    AlbumIncorrectError,
    AlbumNotFoundError,
    PhotoNotFoundError,
    PhotoSearchQueryEmptyError,
    SearchPageInvalidError,
    SearchTokenInvalidError,
    UserAlreadyExists,
    UserCredentialsInvalid,
    UserEmailCodeInvalid,
    UserMediaQuotaReached,
    VideoNotFoundError,
    VideoSearchQueryEmptyError,
)
from modules.app import app


@app.exception_handler(AlbumNotFoundError)
async def album_not_found_exception_handler(request: Request, exc: AlbumNotFoundError):
    return UJSONResponse(
        status_code=HTTP_404_NOT_FOUND,
        content={"detail": f"Could not find album with id '{exc.id}'."},
    )


@app.exception_handler(AlbumAlreadyExistsError)
async def album_already_exists_exception_handler(
    request: Request, exc: AlbumAlreadyExistsError
):
    return UJSONResponse(
        status_code=HTTP_409_CONFLICT,
        content={"detail": f"Album with name '{exc.name}' already exists."},
    )


@app.exception_handler(AlbumIncorrectError)
async def album_incorrect_exception_handler(request: Request, exc: AlbumIncorrectError):
    return UJSONResponse(
        status_code=HTTP_406_NOT_ACCEPTABLE,
        content={"detail": f"Album {exc.place} invalid: {exc.error}"},
    )


@app.exception_handler(PhotoNotFoundError)
async def photo_not_found_exception_handler(request: Request, exc: PhotoNotFoundError):
    return UJSONResponse(
        status_code=HTTP_404_NOT_FOUND,
        content={"detail": f"Could not find photo with id '{exc.id}'."},
    )


@app.exception_handler(PhotoSearchQueryEmptyError)
async def photo_search_query_empty_exception_handler(
    request: Request, exc: PhotoSearchQueryEmptyError
):
    return UJSONResponse(
        status_code=HTTP_422_UNPROCESSABLE_ENTITY,
        content={
            "detail": "You must provide query, caption or coordinates to look for photos."
        },
    )


@app.exception_handler(VideoNotFoundError)
async def video_not_found_exception_handler(request: Request, exc: VideoNotFoundError):
    return UJSONResponse(
        status_code=HTTP_404_NOT_FOUND,
        content={"detail": f"Could not find video with id '{exc.id}'."},
    )


@app.exception_handler(VideoSearchQueryEmptyError)
async def video_search_query_empty_exception_handler(
    request: Request, exc: VideoSearchQueryEmptyError
):
    return UJSONResponse(
        status_code=HTTP_422_UNPROCESSABLE_ENTITY,
        content={
            "detail": "You must provide query, caption or coordinates to look for photos."
        },
    )


@app.exception_handler(SearchPageInvalidError)
async def search_page_invalid_exception_handler(
    request: Request, exc: SearchPageInvalidError
):
    return UJSONResponse(
        status_code=HTTP_400_BAD_REQUEST,
        content={
            "detail": "Parameters 'page' and 'page_size' must be greater or equal to 1."
        },
    )


@app.exception_handler(SearchPageInvalidError)
async def search_token_invalid_exception_handler(
    request: Request, exc: SearchPageInvalidError
):
    return UJSONResponse(
        status_code=HTTP_401_UNAUTHORIZED,
        content={
            "detail": "Parameters 'page' and 'page_size' must be greater or equal to 1."
        },
    )


@app.exception_handler(SearchTokenInvalidError)
async def search_token_invalid_exception_handler(
    request: Request, exc: SearchTokenInvalidError
):
    return UJSONResponse(
        status_code=HTTP_401_UNAUTHORIZED,
        content={
            "detail": "Invalid search token."
        },
    )


@app.exception_handler(UserEmailCodeInvalid)
async def user_email_code_invalid_exception_handler(
    request: Request, exc: UserEmailCodeInvalid
):
    return UJSONResponse(
        status_code=HTTP_400_BAD_REQUEST,
        content={"detail": "Confirmation code is invalid."},
    )


@app.exception_handler(UserAlreadyExists)
async def user_already_exists_exception_handler(
    request: Request, exc: UserAlreadyExists
):
    return UJSONResponse(
        status_code=HTTP_409_CONFLICT,
        content={"detail": "User with this username already exists."},
    )


@app.exception_handler(AccessTokenInvalidError)
async def access_token_invalid_exception_handler(
    request: Request, exc: AccessTokenInvalidError
):
    return UJSONResponse(
        status_code=HTTP_401_UNAUTHORIZED,
        content={"detail": "Invalid access token."},
    )


@app.exception_handler(UserCredentialsInvalid)
async def user_credentials_invalid_exception_handler(
    request: Request, exc: UserCredentialsInvalid
):
    return UJSONResponse(
        status_code=HTTP_401_UNAUTHORIZED,
        content={"detail": "Invalid credentials."},
    )


@app.exception_handler(UserMediaQuotaReached)
async def user_media_quota_reached_exception_handler(
    request: Request, exc: UserMediaQuotaReached
):
    return UJSONResponse(
        status_code=HTTP_403_FORBIDDEN,
        content={"detail": "Media quota has been reached, media upload impossible."},
    )