diff --git a/endpoints/trees.get.py b/endpoints/trees.get.py index e69de29..c79570a 100644 --- a/endpoints/trees.get.py +++ b/endpoints/trees.get.py @@ -0,0 +1,22 @@ +from fastapi import APIRouter, Depends, HTTPException, status +from sqlalchemy.orm import Session +from typing import List +from core.database import get_db +from models.tree import Tree +from schemas.tree import TreeSchema +from helpers.tree_helpers import get_all_trees + +router = APIRouter() + +@router.get("/trees", response_model=List[TreeSchema], status_code=status.HTTP_200_OK) +async def get_trees( + db: Session = Depends(get_db) +): + """Get all trees""" + trees = get_all_trees(db) + if not trees: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail="No trees found" + ) + return trees \ No newline at end of file