catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix flake8 failures

Browse source
This commit is contained in:
Karl Gutwin 2018-11-08 11:08:24 -05:00
commit ff6a57f443
5 changed files with 17 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

14
moto/dynamodbstreams/models.py
View file

@ -4,7 +4,6 @@ import os
import json
import boto3
import base64
import datetime
from moto.core import BaseBackend, BaseModel
from moto.dynamodb2.models import dynamodb_backends
@ -31,7 +30,7 @@ class ShardIterator(BaseModel):
self.stream_shard.table.table_arn,
self.stream_shard.table.latest_stream_label,
self.id)
def to_json(self):
return {
'ShardIterator': self.arn
@ -67,7 +66,7 @@ class DynamoDBStreamsBackend(BaseBackend):
region = self.region
self.__dict__ = {}
self.__init__(region)
@property
def dynamodb(self):
return dynamodb_backends[self.region]
@ -75,7 +74,7 @@ class DynamoDBStreamsBackend(BaseBackend):
def _get_table_from_arn(self, arn):
table_name = arn.split(':', 6)[5].split('/')[1]
return self.dynamodb.get_table(table_name)
def describe_stream(self, arn):
table = self._get_table_from_arn(arn)
resp = {'StreamDescription': {
@ -90,7 +89,7 @@ class DynamoDBStreamsBackend(BaseBackend):
'Shards': ([table.stream_shard.to_json()] if table.stream_shard
else [])
}}
return json.dumps(resp)
def list_streams(self, table_name=None):
@ -105,7 +104,7 @@ class DynamoDBStreamsBackend(BaseBackend):
'TableName': d['Table']['TableName'],
'StreamLabel': d['Table']['LatestStreamLabel']
})
return json.dumps({'Streams': streams})
def get_shard_iterator(self, arn, shard_id, shard_iterator_type, sequence_number=None):
@ -116,14 +115,13 @@ class DynamoDBStreamsBackend(BaseBackend):
shard_iterator_type,
sequence_number)
self.shard_iterators[shard_iterator.arn] = shard_iterator
return json.dumps(shard_iterator.to_json())
def get_records(self, iterator_arn, limit):
shard_iterator = self.shard_iterators[iterator_arn]
return json.dumps(shard_iterator.get(limit))
available_regions = boto3.session.Session().get_available_regions(
'dynamodbstreams')

4
moto/dynamodbstreams/responses.py
View file

@ -1,7 +1,5 @@
from __future__ import unicode_literals
import json
from moto.core.responses import BaseResponse
from .models import dynamodbstreams_backends
@ -12,7 +10,7 @@ class DynamoDBStreamsHandler(BaseResponse):
@property
def backend(self):
return dynamodbstreams_backends[self.region]
def describe_stream(self):
arn = self._get_param('StreamArn')
return self.backend.describe_stream(arn)