|
from typing import Any, Callable, Dict, Iterable, Type, TypeVar |
|
|
|
from starlette.datastructures import URL as URL |
|
from starlette.datastructures import Address as Address |
|
from starlette.datastructures import FormData as FormData |
|
from starlette.datastructures import Headers as Headers |
|
from starlette.datastructures import QueryParams as QueryParams |
|
from starlette.datastructures import State as State |
|
from starlette.datastructures import UploadFile as StarletteUploadFile |
|
|
|
|
|
class UploadFile(StarletteUploadFile): |
|
@classmethod |
|
def __get_validators__(cls: Type["UploadFile"]) -> Iterable[Callable[..., Any]]: |
|
yield cls.validate |
|
|
|
@classmethod |
|
def validate(cls: Type["UploadFile"], v: Any) -> Any: |
|
if not isinstance(v, StarletteUploadFile): |
|
raise ValueError(f"Expected UploadFile, received: {type(v)}") |
|
return v |
|
|
|
@classmethod |
|
def __modify_schema__(cls, field_schema: Dict[str, Any]) -> None: |
|
field_schema.update({"type": "string", "format": "binary"}) |
|
|
|
|
|
class DefaultPlaceholder: |
|
""" |
|
You shouldn't use this class directly. |
|
|
|
It's used internally to recognize when a default value has been overwritten, even |
|
if the overridden default value was truthy. |
|
""" |
|
|
|
def __init__(self, value: Any): |
|
self.value = value |
|
|
|
def __bool__(self) -> bool: |
|
return bool(self.value) |
|
|
|
def __eq__(self, o: object) -> bool: |
|
return isinstance(o, DefaultPlaceholder) and o.value == self.value |
|
|
|
|
|
DefaultType = TypeVar("DefaultType") |
|
|
|
|
|
def Default(value: DefaultType) -> DefaultType: |
|
""" |
|
You shouldn't use this function directly. |
|
|
|
It's used internally to recognize when a default value has been overwritten, even |
|
if the overridden default value was truthy. |
|
""" |
|
return DefaultPlaceholder(value) |
|
|