Lints.
This commit is contained in:
parent
1433f28846
commit
f37bad0e00
260 changed files with 6363 additions and 3766 deletions
|
|
@ -4,7 +4,6 @@ import json
|
|||
|
||||
from moto.core.responses import BaseResponse
|
||||
from .models import kinesis_backends
|
||||
from werkzeug.exceptions import BadRequest
|
||||
|
||||
|
||||
class KinesisResponse(BaseResponse):
|
||||
|
|
@ -25,7 +24,8 @@ class KinesisResponse(BaseResponse):
|
|||
def create_stream(self):
|
||||
stream_name = self.parameters.get('StreamName')
|
||||
shard_count = self.parameters.get('ShardCount')
|
||||
self.kinesis_backend.create_stream(stream_name, shard_count, self.region)
|
||||
self.kinesis_backend.create_stream(
|
||||
stream_name, shard_count, self.region)
|
||||
return ""
|
||||
|
||||
def describe_stream(self):
|
||||
|
|
@ -50,7 +50,8 @@ class KinesisResponse(BaseResponse):
|
|||
stream_name = self.parameters.get("StreamName")
|
||||
shard_id = self.parameters.get("ShardId")
|
||||
shard_iterator_type = self.parameters.get("ShardIteratorType")
|
||||
starting_sequence_number = self.parameters.get("StartingSequenceNumber")
|
||||
starting_sequence_number = self.parameters.get(
|
||||
"StartingSequenceNumber")
|
||||
|
||||
shard_iterator = self.kinesis_backend.get_shard_iterator(
|
||||
stream_name, shard_id, shard_iterator_type, starting_sequence_number,
|
||||
|
|
@ -64,7 +65,8 @@ class KinesisResponse(BaseResponse):
|
|||
shard_iterator = self.parameters.get("ShardIterator")
|
||||
limit = self.parameters.get("Limit")
|
||||
|
||||
next_shard_iterator, records = self.kinesis_backend.get_records(shard_iterator, limit)
|
||||
next_shard_iterator, records = self.kinesis_backend.get_records(
|
||||
shard_iterator, limit)
|
||||
|
||||
return json.dumps({
|
||||
"NextShardIterator": next_shard_iterator,
|
||||
|
|
@ -77,7 +79,8 @@ class KinesisResponse(BaseResponse):
|
|||
stream_name = self.parameters.get("StreamName")
|
||||
partition_key = self.parameters.get("PartitionKey")
|
||||
explicit_hash_key = self.parameters.get("ExplicitHashKey")
|
||||
sequence_number_for_ordering = self.parameters.get("SequenceNumberForOrdering")
|
||||
sequence_number_for_ordering = self.parameters.get(
|
||||
"SequenceNumberForOrdering")
|
||||
data = self.parameters.get("Data")
|
||||
|
||||
sequence_number, shard_id = self.kinesis_backend.put_record(
|
||||
|
|
@ -105,7 +108,7 @@ class KinesisResponse(BaseResponse):
|
|||
stream_name = self.parameters.get("StreamName")
|
||||
shard_to_split = self.parameters.get("ShardToSplit")
|
||||
new_starting_hash_key = self.parameters.get("NewStartingHashKey")
|
||||
response = self.kinesis_backend.split_shard(
|
||||
self.kinesis_backend.split_shard(
|
||||
stream_name, shard_to_split, new_starting_hash_key
|
||||
)
|
||||
return ""
|
||||
|
|
@ -114,15 +117,17 @@ class KinesisResponse(BaseResponse):
|
|||
stream_name = self.parameters.get("StreamName")
|
||||
shard_to_merge = self.parameters.get("ShardToMerge")
|
||||
adjacent_shard_to_merge = self.parameters.get("AdjacentShardToMerge")
|
||||
response = self.kinesis_backend.merge_shards(
|
||||
self.kinesis_backend.merge_shards(
|
||||
stream_name, shard_to_merge, adjacent_shard_to_merge
|
||||
)
|
||||
return ""
|
||||
|
||||
''' Firehose '''
|
||||
|
||||
def create_delivery_stream(self):
|
||||
stream_name = self.parameters['DeliveryStreamName']
|
||||
redshift_config = self.parameters.get('RedshiftDestinationConfiguration')
|
||||
redshift_config = self.parameters.get(
|
||||
'RedshiftDestinationConfiguration')
|
||||
|
||||
if redshift_config:
|
||||
redshift_s3_config = redshift_config['S3Configuration']
|
||||
|
|
@ -149,7 +154,8 @@ class KinesisResponse(BaseResponse):
|
|||
's3_compression_format': s3_config.get('CompressionFormat'),
|
||||
's3_buffering_hings': s3_config['BufferingHints'],
|
||||
}
|
||||
stream = self.kinesis_backend.create_delivery_stream(stream_name, **stream_kwargs)
|
||||
stream = self.kinesis_backend.create_delivery_stream(
|
||||
stream_name, **stream_kwargs)
|
||||
return json.dumps({
|
||||
'DeliveryStreamARN': stream.arn
|
||||
})
|
||||
|
|
@ -177,7 +183,8 @@ class KinesisResponse(BaseResponse):
|
|||
stream_name = self.parameters['DeliveryStreamName']
|
||||
record_data = self.parameters['Record']['Data']
|
||||
|
||||
record = self.kinesis_backend.put_firehose_record(stream_name, record_data)
|
||||
record = self.kinesis_backend.put_firehose_record(
|
||||
stream_name, record_data)
|
||||
return json.dumps({
|
||||
"RecordId": record.record_id,
|
||||
})
|
||||
|
|
@ -188,7 +195,8 @@ class KinesisResponse(BaseResponse):
|
|||
|
||||
request_responses = []
|
||||
for record in records:
|
||||
record_response = self.kinesis_backend.put_firehose_record(stream_name, record['Data'])
|
||||
record_response = self.kinesis_backend.put_firehose_record(
|
||||
stream_name, record['Data'])
|
||||
request_responses.append({
|
||||
"RecordId": record_response.record_id
|
||||
})
|
||||
|
|
@ -207,7 +215,8 @@ class KinesisResponse(BaseResponse):
|
|||
stream_name = self.parameters.get('StreamName')
|
||||
exclusive_start_tag_key = self.parameters.get('ExclusiveStartTagKey')
|
||||
limit = self.parameters.get('Limit')
|
||||
response = self.kinesis_backend.list_tags_for_stream(stream_name, exclusive_start_tag_key, limit)
|
||||
response = self.kinesis_backend.list_tags_for_stream(
|
||||
stream_name, exclusive_start_tag_key, limit)
|
||||
return json.dumps(response)
|
||||
|
||||
def remove_tags_from_stream(self):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue