| from functools import wraps |
| from flask import request, jsonify |
| from typing import Callable, Type |
| from marshmallow import Schema, ValidationError |
|
|
| def validate_input(serializer: Type[Schema], partial: bool = False) -> Callable: |
| """ |
| Validates the input data using a Marshmallow serializer. |
| |
| Args: |
| serializer (Type[Schema]): The Marshmallow serializer to use. |
| partial (bool, optional): Whether to allow partial updates. Defaults to False. |
| |
| Returns: |
| Callable: The decorated function. |
| """ |
| def decorator(func): |
| @wraps(func) |
| def wrapper(*args, **kwargs): |
| try: |
| data = request.get_json() |
| if not data: |
| return jsonify({'error': 'No input data provided'}), 400 |
| deserialized_data = serializer().load(data, partial=partial) |
| request.deserialized_data = deserialized_data |
| return func(*args, **kwargs) |
| except ValidationError as e: |
| return jsonify({'error': str(e)}), 400 |
| except Exception as e: |
| return jsonify({'error': 'Invalid input data'}), 400 |
| return wrapper |
| return decorator |
|
|
