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 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}
@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