56 lines
1.2 KiB
Python
56 lines
1.2 KiB
Python
import csv
|
|
from io import StringIO
|
|
from typing import TYPE_CHECKING
|
|
|
|
from aws_lambda_powertools.utilities.data_classes import (
|
|
EventBridgeEvent,
|
|
event_source,
|
|
)
|
|
from aws_lambda_powertools.utilities.typing import LambdaContext
|
|
|
|
from boto3clients import s3_client
|
|
|
|
if TYPE_CHECKING:
|
|
from mypy_boto3_s3.client import S3Client
|
|
else:
|
|
S3Client = object
|
|
|
|
transport_params = {'client': s3_client}
|
|
|
|
|
|
@event_source(data_class=EventBridgeEvent)
|
|
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
|
|
new_image = event.detail['new_image']
|
|
csvfile = new_image['s3_uri']
|
|
|
|
data = _get_s3_object_range(
|
|
csvfile,
|
|
start_byte=new_image['start_byte'],
|
|
end_byte=new_image['end_byte'],
|
|
s3_client=s3_client,
|
|
)
|
|
reader = csv.reader(data)
|
|
|
|
for x in reader:
|
|
print(x)
|
|
|
|
return True
|
|
|
|
|
|
def _get_s3_object_range(
|
|
s3_uri: str,
|
|
*,
|
|
start_byte: int,
|
|
end_byte: int,
|
|
s3_client: S3Client,
|
|
) -> StringIO:
|
|
bucket, key = s3_uri.replace('s3://', '').split('/', 1)
|
|
|
|
response = s3_client.get_object(
|
|
Bucket=bucket,
|
|
Key=key,
|
|
Range=f'bytes={start_byte}-{end_byte}',
|
|
)
|
|
|
|
return StringIO(response['Body'].read().decode('utf-8'))
|