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 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
|
Loading…
x
Reference in New Issue
Block a user