79 lines
2.0 KiB
Python

from typing import Any, List, Optional
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from app import crud, schemas
from app.db.session import get_db
router = APIRouter()
@router.get("/", response_model=List[schemas.Item])
def read_items(
db: Session = Depends(get_db),
skip: int = 0,
limit: int = 100,
title: Optional[str] = None,
) -> Any:
"""
Retrieve items.
"""
if title:
items = crud.item.get_by_title(db, title=title)
else:
items = crud.item.get_multi(db, skip=skip, limit=limit)
return items
@router.post("/", response_model=schemas.Item, status_code=status.HTTP_201_CREATED)
def create_item(*, db: Session = Depends(get_db), item_in: schemas.ItemCreate) -> Any:
"""
Create new item.
"""
item = crud.item.create(db=db, obj_in=item_in)
return item
@router.get("/{id}", response_model=schemas.Item)
def read_item(*, db: Session = Depends(get_db), id: int) -> Any:
"""
Get item by ID.
"""
item = crud.item.get(db=db, id=id)
if not item:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, detail="Item not found"
)
return item
@router.put("/{id}", response_model=schemas.Item)
def update_item(
*, db: Session = Depends(get_db), id: int, item_in: schemas.ItemUpdate
) -> Any:
"""
Update an item.
"""
item = crud.item.get(db=db, id=id)
if not item:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, detail="Item not found"
)
item = crud.item.update(db=db, db_obj=item, obj_in=item_in)
return item
@router.delete("/{id}", status_code=status.HTTP_204_NO_CONTENT, response_model=None)
def delete_item(*, db: Session = Depends(get_db), id: int) -> Any:
"""
Delete an item.
"""
item = crud.item.get(db=db, id=id)
if not item:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, detail="Item not found"
)
crud.item.remove(db=db, id=id)
return None