Skip to content

Commit cc8edc5

Browse files
NihalHarishrahul003
authored andcommitted
Ensure that the prefix name in s3_trial always includes a trailing '/' (aws#246)
* ensure prefix name in s3 trial is a dir * refactored list_s3_objects * run tests
1 parent cee43fb commit cc8edc5

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tornasole/core/s3_utils.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,15 @@ def _list_s3_prefixes(list_info):
1313
return files
1414

1515

16-
def list_s3_objects(bucket, prefix, start_after_key=None):
16+
def list_s3_objects(bucket, prefix, start_after_key=None, delimiter=""):
1717
last_token = None
1818
if start_after_key is None:
1919
start_after_key = prefix
2020
logger.debug(f'Trying to load index files after {start_after_key}')
21-
list_params = {'Bucket': bucket, 'Prefix': prefix, 'StartAfter': start_after_key}
22-
req = ListRequest(**list_params)
21+
req = ListRequest(Bucket=bucket,
22+
Prefix=prefix,
23+
StartAfter=start_after_key,
24+
Delimiter=delimiter)
2325
objects = _list_s3_prefixes([req])
2426
if len(objects) > 0:
2527
last_token = objects[-1]

tornasole/trials/s3_trial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def __init__(self, name, bucket_name, prefix_name,
3232
parallel=False, check=check, index_mode=index_mode, cache=cache)
3333
self.logger.info(f'Loading trial {name} at path s3://{bucket_name}/{prefix_name}')
3434
self.bucket_name = bucket_name
35-
self.prefix_name = prefix_name
35+
self.prefix_name = os.path.join(prefix_name, '')
3636
self.path = "s3://"+os.path.join(self.bucket_name, self.prefix_name)
3737
self.s3_handler = S3Handler()
3838
self._load_collections()

0 commit comments

Comments
 (0)