Fix a typo error on task argument

Modify unit test to be more robust
This commit is contained in:
venusbb
2018-01-17 12:21:56 +00:00
parent 63bff9d9c3
commit 9179802717
3 changed files with 18 additions and 8 deletions

View File

@@ -119,8 +119,12 @@ def get_list_of_files_by_suffix(bucket_name, subfolder='', suffix='', last_modif
)
for page in page_iterator:
for obj in page['Contents']:
key = obj['Key'].lower()
if key.endswith(suffix.lower()):
if not last_modified or obj['LastModified'] >= last_modified:
yield key
try:
for obj in page['Contents']:
key = obj['Key'].lower()
if key.endswith(suffix.lower()):
if not last_modified or obj['LastModified'] >= last_modified:
yield key
except KeyError:
# Not content for today
pass