TouchDesigner MCP

""" Error handling utilities for TouchDesigner MCP Web server Provides standardized error categorization, formatting, and handling """ from enum import Enum import functools import traceback from typing import Callable, Optional, TypeVar from utils.logging import log_message from utils.result import Result, error_result from utils.types import LogLevel T = TypeVar("T") class ErrorCategory(Enum): """ Error categories for better classification and handling """ INTERNAL = "INTERNAL" VALIDATION = "VALIDATION" NOT_FOUND = "NOT_FOUND" PERMISSION = "PERMISSION" NETWORK = "NETWORK" EXTERNAL = "EXTERNAL" def __str__(self) -> str: return self.value def categorize_error(exception: Exception) -> ErrorCategory: """ Categorize an exception based on its type and message Args: exception: The exception to categorize Returns: Appropriate ErrorCategory """ error_message = str(exception).lower() if isinstance(exception, ValueError): return ErrorCategory.VALIDATION elif ( isinstance(exception, FileNotFoundError) or "not found" in error_message or "doesn't exist" in error_message ): return ErrorCategory.NOT_FOUND elif "permission" in error_message or "access denied" in error_message: return ErrorCategory.PERMISSION elif "network" in error_message or "connection" in error_message: return ErrorCategory.NETWORK elif "external" in error_message or "service unavailable" in error_message: return ErrorCategory.EXTERNAL return ErrorCategory.INTERNAL def format_error(message: str, category: Optional[ErrorCategory] = None) -> str: """ Format an error message with its category Args: message: The error message category: Error category (defaults to INTERNAL) Returns: Formatted error message """ if category is None: category = ErrorCategory.INTERNAL return f"{category}: {message}" def handle_service_errors(func: Callable[..., Result]) -> Callable[..., Result]: """ Decorator to handle errors in service methods and convert them to Result type Args: func: The service method to decorate Returns: Wrapped function that catches exceptions and returns Result """ @functools.wraps(func) def wrapper(*args, **kwargs) -> Result: try: return func(*args, **kwargs) except Exception as e: category = categorize_error(e) func_name = func.__name__ log_message(f"Error in {func_name}: {str(e)}", LogLevel.ERROR) log_message(traceback.format_exc(), LogLevel.DEBUG) error_message = format_error(str(e), category) return error_result(error_message, {"errorCategory": category}) return wrapper