只限于Pydantic颜色类型

Question

我有一种有颜色特性的化脓性模型

from pydantic import BaseModel
from pydantic.color import Color

class MyModel(BaseModel):
    color: Color

我定义了FastAPI路由器，它返回MyModel

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

在此设置中，pydantic将颜色转换为指定颜色(例如，black)。

我的前额不认识名字的颜色。

我怎么才能指出平阳还六角形颜色？

Answer 1

使用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'

要在响应中获得十六进制值，可以添加自定义json编码器

class MyModel(BaseModel):
    color: Color

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