{{cookiecutter.project_slug}}.app.decorators.validation

Complexity

Total Complexity

3

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	validate_with_pydantic()	0	18	3

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

from flask import request
from pydantic import BaseModel

from app.context import context_property


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


def validate_with_pydantic(payload_location: PayloadLocation, model: Type[BaseModel]):
    def decorator(fn):
        @wraps(fn)
        def wrapper(*args, **kwargs):
            payload = getattr(request, payload_location.value)

            if payload_location == PayloadLocation.ARGS and hasattr(payload, "to_dict"):
                payload = payload.to_dict()

            instance = model(**payload)

            context_property.request_payload = instance

            return fn(*args, **kwargs)

        return wrapper

    return decorator