Shared via mypy Playground

main.py

from typing import ParamSpec, TypeVar, Callable, Literal, NewType
from dataclasses import dataclass

S3EventType = Literal["S3"]
SQSEventType = Literal["SQS"]
SQSRecord = NewType("SQSRecord", str)

AWSEventTypes = S3EventType | SQSEventType


@dataclass(frozen=True)
class S3EventData:
    type: S3EventType
    object_key: str


@dataclass(frozen=True)
class SQSEventData:
    type: SQSEventType
    records: list[SQSRecord]



P = ParamSpec('P')
R = TypeVar('R')

EventHandler = Callable[P, R]

RegisteredEventHandlers = dict[AWSEventTypes, EventHandler[P, R]]

_registered_event_handlers: RegisteredEventHandlers[P, R] = {}