diff --git a/endpoints/colour.get.py b/endpoints/colour.get.py index faa0808..9abde32 100644 --- a/endpoints/colour.get.py +++ b/endpoints/colour.get.py @@ -1,11 +1,16 @@ -from fastapi import APIRouter, status -from typing import Dict -import random +from fastapi import APIRouter, Depends, HTTPException, status +from sqlalchemy.orm import Session +from typing import List +from core.database import get_db +from models.colour import Colour +from schemas.colour import ColourSchema +from helpers.colour_helpers import get_all_colours router = APIRouter() -@router.get("/colour", status_code=status.HTTP_200_OK, response_model=Dict[str, str]) -async def get_random_color(): - """Get a random color in hexadecimal format""" - color = "#{:06x}".format(random.randint(0, 0xFFFFFF)) - return {"color": color} \ No newline at end of file +@router.get("/colour", status_code=200, response_model=List[ColourSchema]) +async def get_colours( + db: Session = Depends(get_db) +): + colours = get_all_colours(db) + return colours \ No newline at end of file