ai_router.py

from fastapi import APIRouter, HTTPException, status
import ai_model
import ai_service
import ai_exception
from pydantic import BaseModel


class Message(BaseModel):
    message: str


ai_router = APIRouter()


@ai_router.post(
    "/newspaper/",
    status_code=status.HTTP_201_CREATED,
    responses={
        201: {"description": "Successful Response and Create Newspaper"},
        403: {"description": "Forbidden, The page cannot be crawled"},
        404: {"description": "Not Found, The URL Type is not matched"},
        419: {"description": "Not Support, This platform does not support the service"},
    },
)
async def post_newspaper(
    body: ai_model.APIMODEL.NewsPaperBody,
) -> ai_model.APIMODEL.NewsPaper:
    try:
        return ai_service.crawl_and_write_newspaper(body.url)
    except ai_exception.URLNotCrawlableError as e:
        raise HTTPException(status_code=403, detail={"message": e.args[0]})
    except ai_exception.InvalidURLError as e:
        raise HTTPException(status_code=404, detail={"message": e.args[0]})
    except ai_exception.NotSupportedException as e:
        raise HTTPException(status_code=419, detail={"message": e.args[0]})
    except Exception as e:
        raise HTTPException(status_code=500, detail={"message": e})


# @ai_router.get(
#     "/newspapaers/{USER_ID}",
#     responses={404: {"description": "Not Found, Check User Id"}},
# )
# def get_newspapers(user_id: int) -> ai_model.APIMODEL.Newspapers:
#     try:
#         return ai_service.get_newspapers_for_user(user_id)
#     except ai_exception.UserNotFoundError as e:
#         raise HTTPException(status_code=404, detail={"message": e.args[0]})
#     except Exception as e:
#         raise HTTPException(status_code=500, detail={"message": e.args[0]})