33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from arnparse import arnparse
|
|
from aws_lambda_powertools.utilities.data_classes import (
|
|
DynamoDBStreamEvent,
|
|
event_source,
|
|
)
|
|
from aws_lambda_powertools.utilities.typing import LambdaContext
|
|
from meilisearch import Client as Meilisearch
|
|
|
|
from config import MEILISEARCH_API_KEY, MEILISEARCH_HOST
|
|
from meili import Op
|
|
|
|
meili_client = Meilisearch(MEILISEARCH_HOST, MEILISEARCH_API_KEY)
|
|
|
|
|
|
@event_source(data_class=DynamoDBStreamEvent)
|
|
def lambda_handler(event: DynamoDBStreamEvent, context: LambdaContext):
|
|
with Op(meili_client) as op:
|
|
for record in event.records:
|
|
pk = record.dynamodb.keys['id'] # type: ignore
|
|
new_image = record.dynamodb.new_image # type: ignore
|
|
index = table_from_arn(record.event_source_arn) # type: ignore
|
|
|
|
op.append(
|
|
index,
|
|
op=record.event_name, # type: ignore
|
|
data=new_image or pk,
|
|
)
|
|
|
|
|
|
def table_from_arn(arn: str) -> str:
|
|
arn_ = arnparse(arn)
|
|
return arn_.resource.split('/')[0]
|