def put_subscription_filter(*args, **kwargs):
log_group_name = args[0]
filter_name = args[1]
filter_pattern = args[2]
destination_arn = args[3]
role_arn = args[4]
log_group = logs_models.logs_backends[
aws_stack.get_region()].groups.get(log_group_name)
if not log_group:
raise ResourceNotFoundException(
'The specified log group does not exist.')
if ':lambda:' in destination_arn:
client = aws_stack.connect_to_service('lambda')
lambda_name = aws_stack.lambda_function_name(destination_arn)
try:
client.get_function(FunctionName=lambda_name)
except Exception:
raise InvalidParameterException(
'destinationArn for vendor lambda cannot be used with roleArn'
)
elif ':kinesis:' in destination_arn:
client = aws_stack.connect_to_service('kinesis')
stream_name = aws_stack.kinesis_stream_name(destination_arn)
try:
client.describe_stream(StreamName=stream_name)
except Exception:
raise InvalidParameterException(
'Could not deliver test message to specified Kinesis stream. '
'Check if the given kinesis stream is in ACTIVE state. '
)
elif ':firehose:' in destination_arn:
client = aws_stack.connect_to_service('firehose')
firehose_name = aws_stack.firehose_name(destination_arn)
try:
client.describe_delivery_stream(
DeliveryStreamName=firehose_name)
except Exception:
raise InvalidParameterException(
'Could not deliver test message to specified Firehose stream. '
'Check if the given Firehose stream is in ACTIVE state.'
)
else:
service = aws_stack.extract_service_from_arn(destination_arn)
raise InvalidParameterException(
'PutSubscriptionFilter operation cannot work with destinationArn for vendor %s'
% service)
log_group.put_subscription_filter(filter_name, filter_pattern,
destination_arn, role_arn)
def moto_put_subscription_filter(fn, self, *args, **kwargs):
log_group_name = args[0]
filter_name = args[1]
filter_pattern = args[2]
destination_arn = args[3]
role_arn = args[4]
log_group = self.groups.get(log_group_name)
if not log_group:
raise ResourceNotFoundException("The specified log group does not exist.")
if ":lambda:" in destination_arn:
client = aws_stack.connect_to_service("lambda")
lambda_name = aws_stack.lambda_function_name(destination_arn)
try:
client.get_function(FunctionName=lambda_name)
except Exception:
raise InvalidParameterException(
"destinationArn for vendor lambda cannot be used with roleArn"
)
elif ":kinesis:" in destination_arn:
client = aws_stack.connect_to_service("kinesis")
stream_name = aws_stack.kinesis_stream_name(destination_arn)
try:
client.describe_stream(StreamName=stream_name)
except Exception:
raise InvalidParameterException(
"Could not deliver test message to specified Kinesis stream. "
"Check if the given kinesis stream is in ACTIVE state. "
)
elif ":firehose:" in destination_arn:
client = aws_stack.connect_to_service("firehose")
firehose_name = aws_stack.firehose_name(destination_arn)
try:
client.describe_delivery_stream(DeliveryStreamName=firehose_name)
except Exception:
raise InvalidParameterException(
"Could not deliver test message to specified Firehose stream. "
"Check if the given Firehose stream is in ACTIVE state."
)
else:
service = aws_stack.extract_service_from_arn(destination_arn)
raise InvalidParameterException(
f"PutSubscriptionFilter operation cannot work with destinationArn for vendor {service}"
)
if filter_pattern:
for stream in log_group.streams.values():
stream.filter_pattern = filter_pattern
log_group.put_subscription_filter(filter_name, filter_pattern, destination_arn, role_arn)