Update code in endpoints/colour.get.py
This commit is contained in:
parent
537333bf58
commit
c7949ea1bc
@ -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
|
Loading…
x
Reference in New Issue
Block a user