Flask 如何验证前端传过来的数据

from enum import Enum from pydantic import ( BaseModel, Field, ) class HolidayTypeEnum(str, Enum): ANNUAL_LEAVE = "annual_leave" SICK_LEAVE = "sick_leave" class Holiday(BaseModel): holiday_name: str = Field(..., description="请填写假期名称") holiday_type: HolidayTypeEnum = Field(..., description="请填写假期类型") data = { "holiday_name": "年假", "holiday_type": "annual_leave1", } holiday = Holiday(**data)

抛出的异常是：

pydantic.error_wrappers.ValidationError: 1 validation error for Holiday holiday_type value is not a valid enumeration member; permitted: 'annual_leave', 'sick_leave' (type=type_error.enum; enum_values=[<HolidayTypeEnum.ANNUAL_LEAVE: 'annual_leave'>, <HolidayTypeEnum.SICK_LEAVE: 'sick_leave'>])

如何提示给前端到底错在哪里（之前手动校验的时候，是可以直接抛出指定异常，比如说 raise ValueError("假期类型错误，只能为 XXX 、XXX")）

也许 Pydantic 是很好，但是我好像不太会用，有没有大佬有正确的打开方式，或者说其他可以拿来就用的第三方库

16 replies 2023-09-07 09:24:53 +08:00

BingoXuan

Sep 6, 2023

try except 捕获 ValidationError ，然后返回 http code 400 和错误信息

高级一点就是自己写个 middleware 处理

maocat

Sep 6, 2023

flask_marshmallow 很久之前用的了，不知道现在啥情况了

raptor

Sep 6, 2023

flask-restx 试试？

tf2

Sep 6, 2023

> Django 的 Forms 和 DRF 的 Serializer 都很香

说明你们数据校验需求真的很简单。

maocat

Sep 6, 2023

#2 看样子还在更新 https://github.com/marshmallow-code/marshmallow

XueXianqi

Sep 6, 2023

@BingoXuan
我看了一下捕获到的 ValidationError ，内容是这样的：

ValidationError(
model='Holiday',
errors=[
{
'loc': (
'holiday_type',
),
'msg': "value is not a valid enumeration member; permitted: 'annual_leave', 'sick_leave'",
'type': 'type_error.enum',
'ctx': {
'enum_values': [
<HolidayTypeEnum.ANNUAL_LEAVE: 'annual_leave'>,
<HolidayTypeEnum.SICK_LEAVE: 'sick_leave'>,
],
},
},
],
)

具体的错误信息应该如何解析呢（比如说这里只有英文内容的 msg ，有类似于 Django 设置国际化，转换成中文的配置么）

XueXianqi

Sep 6, 2023

@lambdaq Django 的 Forms 只是初学 Django 的时候觉得好用，实际业务中是用的 DRF 的 Serializer ，还是有很多自定义的 validator 的，不可能单纯地用原有的校验数据类型、长度等

mikurasa

Sep 6, 2023

https://flask-restx.readthedocs.io/en/latest/parsing.html 解君愁

Baloneo

Sep 6, 2023

```
def validate_schema(schema_class):
assert issubclass(schema_class, Schema) # Marshmallow

def decorator(view_func):
@functools.wraps(view_func)
def inner(*args, **kwargs):

if request.method == "GET":
form_data = request.args
else:
form_data = request.json
try:
data = schema_class().load(form_data)
_request_ctx_stack.top.schema_data = data
except ValidationError as e:
return error_json(
ResponseCode.ERROR, parse_err_messages(e), data=e.messages
)
return view_func(*args, **kwargs)

return inner

return decorator

@validate_schema(val.PostTestSchema)
def api_test():
device_id = current_schema_data.get('device_id')
```