Pydantic color type hex only

user_7704231 0
Viewed 75

I have the pydantic model that have color property

from pydantic import BaseModel
from pydantic.color import Color

class MyModel(BaseModel):
    color: Color

And I define FastAPI router that returns MyModel

@router.get("/", response_model=MyModel)
async def get_my_model() -> MyModel:
    item = await DbItem.get()
    return MyModel(**item.dict())

In this setup pydantic converts colors to named colors (e.g. black).

My frontend don't recognize named colors.

How can I point pydantic to return hex colors?

pydantic fastapi colors python
1 Answers

Use Color.as_hex:

from pydantic import BaseModel
from pydantic.color import Color


class MyModel(BaseModel):
    color: Color


my_model = MyModel(color=Color("black"))
print(my_model.color.as_hex())  # '#000'

To get the hex value in your response, you can add a custom json encoder:

class MyModel(BaseModel):
    color: Color

    class Config:
        json_encoders = {Color: lambda c: c.as_hex()}
user_80910930
Related