Function aws_operation

scripts/agents/devops/src/aws/base.py:434–461

Decorator for AWS operations that handles common errors. Args: operation_name: Name of the operation, defaults to the function name. Returns: Decorated function.

(operation_name=None)

# Common decorator for AWS operations with error handling
def aws_operation(operation_name=None):
    """
    Decorator for AWS operations that handles common errors.
    
    Args:
        operation_name: Name of the operation, defaults to the function name.
    
    Returns:
        Decorated function.
    """
    def decorator(func):
        def wrapper(self, *args, **kwargs):
            op_name = operation_name or func.__name__
            try:
                return func(self, *args, **kwargs)
            except (AWSServiceError, ResourceNotFoundError, 
                   PermissionDeniedError, ValidationError,
                   RateLimitError, ResourceLimitError) as e:
                # Re-raise known exceptions
                raise
            except Exception as e:
                # Handle and convert unknown exceptions
                if isinstance(self, AWSBaseService):
                    self.handle_error(e, op_name)
                else:
                    raise AWSServiceError(f"{op_name} failed: {e}")
        return wrapper
    return decorator