pydantic enum type

[wu-clan]

Use Enum type validation in pydantic, e.g.

class IntType(IntEnum):
    a = 1
    b = 2

class Pdv(BaseModel):
    num: IntType = IntType.a

This method does enum validation, but dict() returns the IntType instance instead of the value

Type errors are usually raised when storing the database

[wu-clan]

Hi, @downdawn

Do you have any good ideas?

[downdawn]

Simpler

class Pdv(BaseModel):
    num: Int = IntType.a

[wu-clan]

This will lose the enum checks and only requires the int type

[wu-clan]

For Enum type data, add a validator and return value

class Pdv(BaseModel):
    num: IntType = IntType.a

    @validator('num')
    def num_validator(cls, v):
        return v.value

[wu-clan]

For exception return 422

[wu-clan]

Another option, custom validation exceptions

class Pdv(BaseModel):
    num: int = IntType.a

    @validator('num')
    def num_validator(cls, v):
        if v not in Pdv.get_member_values():
            raise ValueError('Invalid num value, permitted: 1, 2')
        return v

[wu-clan]

assert syntax:

@validator('num')
def num_validator(cls, v):
    assert v in Pdv.get_member_values(), 'Invalid num value, permitted: 1, 2'
    return v

[wu-clan]

For exception return 500

[downdawn]

    status: int = Field(default=StatusType.enable)

    @validator('status')
    def status_validator(cls, v):
        status = StatusType.get_member_values()
        if v not in status:
            raise ValueError('status必须是{}中的值'.format(status))
        return v

[wu-clan]

It is better to use f-string

[wu-clan]

错过了官方文档，只需

Config:
  use_enum_values = True