41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import urllib.parse as urllib_parse
|
|
from email.utils import parseaddr
|
|
|
|
from aws_lambda_powertools import Logger
|
|
from aws_lambda_powertools.utilities.data_classes import SESEvent, event_source
|
|
from aws_lambda_powertools.utilities.typing import LambdaContext
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey
|
|
|
|
from boto3clients import dynamodb_client
|
|
from config import USER_TABLE
|
|
from ses_utils import get_header_value
|
|
|
|
logger = Logger(__name__)
|
|
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
|
|
|
|
|
@logger.inject_lambda_context
|
|
@event_source(data_class=SESEvent)
|
|
def lambda_handler(event: SESEvent, context: LambdaContext) -> dict:
|
|
ses = event.record.ses
|
|
to = urllib_parse.unquote(ses.receipt.recipients[0]).lower()
|
|
name, email_from = parseaddr(get_header_value(ses.mail.headers, 'from'))
|
|
|
|
org_id = user_layer.collection.get_item(
|
|
KeyPair('email', SortKey(to, path_spec='user_id')),
|
|
raise_on_error=False,
|
|
default={},
|
|
)
|
|
|
|
if not org_id:
|
|
return {'disposition': 'STOP_RULE_SET'}
|
|
|
|
print(
|
|
{
|
|
'id': f'mailbox#org#{org_id}',
|
|
'sk': ses.mail.message_id,
|
|
}
|
|
)
|
|
|
|
return {'disposition': 'CONTINUE'}
|