Update code in endpoints/colour.get.py

This commit is contained in:
Backend IM Bot 2025-03-28 18:30:39 +00:00
parent 537333bf58
commit c7949ea1bc

View File

@ -1,11 +1,16 @@
from fastapi import APIRouter, status from fastapi import APIRouter, Depends, HTTPException, status
from typing import Dict from sqlalchemy.orm import Session
import random 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 = APIRouter()
@router.get("/colour", status_code=status.HTTP_200_OK, response_model=Dict[str, str]) @router.get("/colour", status_code=200, response_model=List[ColourSchema])
async def get_random_color(): async def get_colours(
"""Get a random color in hexadecimal format""" db: Session = Depends(get_db)
color = "#{:06x}".format(random.randint(0, 0xFFFFFF)) ):
return {"color": color} colours = get_all_colours(db)
return colours