build.app.decorators.validation

Complexity

Total Complexity

6

Size/Duplication

Total Lines	50
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
B	validate_with_pydantic()	0	34	6

from enum import Enum
from functools import wraps
from typing import Type

from flask import request
from pydantic import BaseModel, ValidationError

from app.context import context_property


class PayloadLocation(Enum):
    ARGS = "args"
    JSON = "json"


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