Chức năng tiện ích s3keys
của tôi về cơ bản là một phiên bản tối ưu hóa câu trả lời của @ Hephaestus:
import boto3
s3_paginator = boto3.client('s3').get_paginator('list_objects_v2')
def keys(bucket_name, prefix='/', delimiter='/', start_after=''):
prefix = prefix[1:] if prefix.startswith(delimiter) else prefix
start_after = (start_after or prefix) if prefix.endswith(delimiter) else start_after
for page in s3_paginator.paginate(Bucket=bucket_name, Prefix=prefix, StartAfter=start_after):
for content in page.get('Contents', ()):
yield content['Key']
Trong các thử nghiệm của tôi (boto3 1.9.84), nó nhanh hơn đáng kể so với mã tương đương (nhưng đơn giản hơn):
import boto3
def keys(bucket_name, prefix='/', delimiter='/'):
prefix = prefix[1:] if prefix.startswith(delimiter) else prefix
bucket = boto3.resource('s3').Bucket(bucket_name)
return (_.key for _ in bucket.objects.filter(Prefix=prefix))
Vì S3 đảm bảo kết quả được sắp xếp nhị phân UTF-8 , start_after
tối ưu hóa đã được thêm vào chức năng đầu tiên.