Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách dùng Boto3 phân trang qua tất cả đối tượng của bucket S3 trong AWS Glue


Đề bài: Sử dụng thư viện boto3 trong Python để phân trang qua tất cả các đối tượng (objects) của một bucket S3 từ AWS Glue Data Catalog được tạo trong tài khoản của bạn.

Cách tiếp cận / Thuật toán

  • Bước 1: Import boto3 và botocore.exceptions để xử lý ngoại lệ.

  • Bước 2: max_items, page_size và starting_token là tham số tùy chọn, trong khi bucket_name là bắt buộc.

    • max_items chỉ định tổng số bản ghi trả về. Nếu số bản ghi có sẵn > max_items, một NextToken sẽ được cung cấp để tiếp tục phân trang.

    • page_size chỉ định kích thước mỗi trang.

    • starting_token giúp phân trang, sử dụng khóa cuối (last key) từ phản hồi trước.

  • Bước 3: Tạo phiên làm việc AWS bằng boto3. Đảm bảo region_name được cấu hình; nếu không, truyền rõ ràng.

  • Bước 4: Tạo client AWS cho dịch vụ S3.

  • Bước 5: Tạo đối tượng paginator chứa chi tiết các phiên bản đối tượng của bucket S3 bằng list_objects.

  • Bước 6: Gọi hàm paginate và truyền max_items, page_size, starting_token như PaginationConfig, bucket_name như Bucket.

  • Bước 7: Trả về số bản ghi dựa trên max_items và page_size.

  • Bước 8: Xử lý ngoại lệ chung nếu có lỗi khi phân trang.

Mã ví dụ

Sử dụng mã sau để phân trang qua tất cả đối tượng của bucket S3:

import boto3
from botocore.exceptions import ClientError

def paginate_through_all_objects_s3_bucket(bucket_name, max_items=None, page_size=None, starting_token=None):
    session = boto3.session.Session()
    s3_client = session.client('s3')
    try:
        paginator = s3_client.get_paginator('list_objects')
        response = paginator.paginate(
            Bucket=bucket_name,
            PaginationConfig={
                'MaxItems': max_items,
                'PageSize': page_size,
                'StartingToken': starting_token
            }
        )
        return response
    except ClientError as e:
        raise Exception("Lỗi client boto3 trong paginate_through_all_objects_s3_bucket: " + e.__str__())
    except Exception as e:
        raise Exception("Lỗi không mong muốn trong paginate_through_all_objects_s3_bucket: " + e.__str__())

# Lần chạy 1
a = paginate_through_all_objects_s3_bucket('s3-test-bucket', 2, 5)
print(*a)

# Lần chạy 2 (tiếp tục từ token)
for items in a:
    next_token = items['Contents'][max_items-1]['Key']
    b = paginate_through_all_objects_s3_bucket('s3-test-bucket', 2, 5, next_token)
    print(*b)

Kết quả

#1st Run
{'ResponseMetadata': {'RequestId': '5AE5VF7RK', 'HostId': **************, 'HTTPStatusCode': 200, 'HTTPHeaders': {'x-amz-id-2': **************, 'x-amz-request-id': '5AE5VF7RK', 'date': 'Sat, 03 Apr 2021 07:33:28 GMT', 'x-amz-bucket-region': 'us-east-1', 'content-type': 'application/xml', 'transfer-encoding': 'chunked', 'server': 'AmazonS3'}, 'RetryAttempts': 0}, 'IsTruncated': True, 'Marker': '', 'Contents': [{'Key': 'analytics-s3i/param.json', 'LastModified': datetime.datetime(2020, 10, 29, 19, 50, 55, tzinfo=tzutc()), 'ETag': '"e6f6b8e02"', 'Size': 1554, 'StorageClass': 'STANDARD', 'Owner': {'DisplayName': 'AWS.Development', 'ID': '92************************0'}}, {'Key': 'analytics-s3i/params.json', 'LastModified': datetime.datetime(2021, 3, 10, 20, 10, 47, tzinfo=tzutc()), 'ETag': '"22a4bf70c1ed2612"', 'Size': 1756, 'StorageClass': 'STANDARD', 'Owner': {'DisplayName': 'AWS.Development', 'ID': '92************************************************'}}], 'Name': 's3-test-bucket', 'Prefix': '', 'MaxKeys': 5, 'EncodingType': 'url', 'CommonPrefixes': None}

#2nd Run
{'ResponseMetadata': {'RequestId': '5AEAWX', 'HostId': 'Uc********************************', 'HTTPStatusCode': 200, 'HTTPHeaders': {'x-amz-id-2': *****************************, 'x-amz-request-id': '5AEAWX', 'date': 'Sat, 03 Apr 2021 07:33:28 GMT', 'x-amz-bucket-region': 'us-east-1', 'content-type': 'application/xml', 'transfer-encoding': 'chunked', 'server': 'AmazonS3'}, 'RetryAttempts': 0}, 'IsTruncated': True, 'Marker': 'analytics-s3i/params.json', 'Contents': [{'Key': 'analytics-s3i/template.json', 'LastModified': datetime.datetime(2020, 10, 29, 19, 50, 55, tzinfo=tzutc()), 'ETag': '"3af411f"', 'Size': 21334, 'StorageClass': 'STANDARD', 'Owner': {'DisplayName': 'AWS.Development', 'ID': '92************************'}}, {'Key': 'analytics-s3i2/param.json', 'LastModified': datetime.datetime(2020, 11, 20, 17, 32, 45, tzinfo=tzutc()), 'ETag': '"04c29cf86888f99"', 'Size': 1695, 'StorageClass': 'STANDARD', 'Owner': {'DisplayName': 'AWS.Development', 'ID': '92************************'}}], 'Name': 's3-test-bucket', 'Prefix': '', 'MaxKeys': 5, 'EncodingType': 'url', 'CommonPrefixes': None}