build.app.decorators.validation.validate_with_pydantic()

Complexity

Conditions

6

Size

Total Lines	34
Code Lines	25

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

from enum import Enum
def validate_with_pydantic(
    *,
    payload_location: PayloadLocation,
    model: Type[BaseModel],
    json_force_load: bool = False,
):
    def decorator(fn):
        @wraps(fn)
        def wrapper(*args, **kwargs):
            if payload_location == PayloadLocation.JSON:
                if json_force_load:
                    payload = request.get_json(force=True)
                else:
                    payload = request.json
            else:
                payload = request.args

                if hasattr(payload, "to_dict"):
                    payload = payload.to_dict()

            try:
                instance = model(**payload)
            except ValidationError:
                raise
            except:
                instance = None

            context_property.request_payload = instance

            return fn(*args, **kwargs)

        return wrapper

    return decorator