posts.py

from typing import List
from uuid import UUID

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session

from database.connection import get_db
from schemas.models import DeletePostResponse, Post, UpdatePost
from utils.post_crud import (
    post_create,
    post_delete,
    post_get_one,
    post_update,
    posts_get_all,
)

router = APIRouter(tags=["posts"])


@router.post("/create", status_code=status.HTTP_201_CREATED, response_model=Post)
def create_post(post: Post, db: Session = Depends(get_db)):
    return post_create(db=db, post=post)


@router.get("/list/all", status_code=status.HTTP_200_OK, response_model=List[Post])
def get_all_posts(db: Session = Depends(get_db)):
    return posts_get_all(db=db)


@router.get("/get/{id}", status_code=status.HTTP_200_OK, response_model=Post)
def get_one_post(id: UUID, db: Session = Depends(get_db)):
    return post_get_one(db=db, id=id)


@router.delete(
    "/delete/{id}", status_code=status.HTTP_200_OK, response_model=DeletePostResponse
)
def delete_post(id: UUID, db: Session = Depends(get_db)):
    delete_status = post_delete(db=db, id=id)
    if delete_status.detail == "Doesnt Exist":
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND, detail="Post Not Found"
        )
    else:
        return delete_status


@router.patch("/update", status_code=status.HTTP_200_OK, response_model=Post)
def update_post(post: UpdatePost, db: Session = Depends(get_db)):
    return post_update(db=db, post=post)