Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle ValueError raised on missing table name

Browse Source
This commit is contained in:
Dmitry Ryzhikov 2020-04-20 00:15:00 +03:00
parent 4f59a3584b
commit ed5e0b586c
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
moto/dynamodb2/responses.py
View File

@ -92,16 +92,24 @@ class DynamoHandler(BaseResponse):
def list_tables(self): def list_tables(self):
body = self.body body = self.body
limit = body.get("Limit", 100) limit = body.get("Limit", 100)
if body.get("ExclusiveStartTableName"): all_tables = list(self.dynamodb_backend.tables.keys())
last = body.get("ExclusiveStartTableName")
start = list(self.dynamodb_backend.tables.keys()).index(last) + 1 exclusive_start_table_name = body.get("ExclusiveStartTableName")
if exclusive_start_table_name:
try:
last_table_index = all_tables.index(exclusive_start_table_name)
except ValueError:
start = len(all_tables)
else:
start = last_table_index + 1
else: else:
start = 0 start = 0
all_tables = list(self.dynamodb_backend.tables.keys())
if limit: if limit:
tables = all_tables[start : start + limit] tables = all_tables[start : start + limit]
else: else:
tables = all_tables[start:] tables = all_tables[start:]
response = {"TableNames": tables} response = {"TableNames": tables}
if limit and len(all_tables) > start + limit: if limit and len(all_tables) > start + limit:
response["LastEvaluatedTableName"] = tables[-1] response["LastEvaluatedTableName"] = tables[-1]