1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
# -*- coding: utf-8 -*-
# @Author : llc
# @Time : 2021/5/11 16:17
from typing import Any
from pydantic.json_schema import JsonSchemaValue
from pydantic_core import core_schema
from werkzeug.datastructures import FileStorage as _FileStorage
class FileStorage(_FileStorage):
"""
An uploaded file included as part of the request data.
"""
@classmethod
def __get_pydantic_json_schema__(cls, *_args: Any, **_kwargs: Any) -> JsonSchemaValue:
field_schema = {"format": "binary", "type": "string"}
return field_schema
@classmethod
def __get_pydantic_core_schema__(cls, *_args: Any, **_kwargs: Any) -> core_schema.CoreSchema:
return core_schema.with_info_plain_validator_function(cls.validate)
@classmethod
def validate(cls, value: _FileStorage, *_args: Any, **_kwargs: Any) -> _FileStorage:
return value
|
