pydantic and decorator

parameter.py

from pydantic import BaseModel


def func_helper(parameter_type, return_type=None):
    def decorator(function):
        def wrapper(*args, **kwargs):
            result = function(parameter_type(**kwargs))
            return return_type(**result) if return_type else result

        return wrapper

    return decorator


class Parameters(BaseModel):
    a: str
    b: str = 'b'
    c: int = 0


class ReturnValue(BaseModel):
    message: str


@func_helper(Parameters, ReturnValue)
def run(parms: Parameters):
    return {'message': parms.json()}


print(run(a='a', c='1', test='1'))