revert some changes; optimize usage for BASE_EVENT_CLS_TO_APP_TYPE_MAPPING

Author: kossman

src/corva/__init__.py

@@ -1,4 +1,22 @@
-from .models.scheduled.scheduled import ScheduledDataTimeEvent
+from .api import Api
+from .handlers import scheduled, stream, task, partial_rerun_merge
+from .logger import CORVA_LOGGER as Logger
+from .models.rerun import RerunDepth, RerunDepthRange, RerunTime, RerunTimeRange
+from .models.scheduled.scheduled import (
+    ScheduledDataTimeEvent,
+    ScheduledDepthEvent,
+    ScheduledNaturalTimeEvent,
+)
+from .models.stream.stream import (
+    StreamDepthEvent,
+    StreamDepthRecord,
+    StreamTimeEvent,
+    StreamTimeRecord,
+)
+from .models.merge.merge import PartialRerunMergeEvent
+from .models.task import TaskEvent
+from .service.cache_sdk import UserRedisSdk as Cache
+from .shared import SECRETS as secrets
 
 
 def __getattr__(name):

src/corva/validate_app_init.py

@@ -12,6 +12,12 @@
 from corva.models.stream.raw import RawStreamEvent
 from corva.models.task import RawTaskEvent
 
+BASE_EVENT_CLS_TO_APP_TYPE_MAPPING: Dict[str, Type[RawBaseEvent]] = {
+    "task": RawTaskEvent,
+    "stream": RawStreamEvent,
+    "scheduled": RawScheduledEvent,
+}
+
 
 def find_leaf_subclasses(base_class):
     leaf_classes = []
@@ -42,19 +48,14 @@ def validate_manifested_type(manifest: Dict[str, Any], app_decorator_type: str)
 
 
 def validate_event_payload(aws_event, app_decorator_type) -> None:
-    base_event_cls_to_app_type_mapping: Dict[str, Type[RawBaseEvent]] = {
-        "task": RawTaskEvent,
-        "stream": RawStreamEvent,
-        "scheduled": RawScheduledEvent,
-    }
 
     if event_cls := guess_event_type(aws_event):
         if issubclass(event_cls, RawPartialRerunMergeEvent):
             # RawPartialRerunMergeEvent(-s) should be ignored here since
             # it is not new app type itself it's just a run mode for existing app types
             return
 
-        expected_base_event_cls = base_event_cls_to_app_type_mapping[app_decorator_type]
+        expected_base_event_cls = BASE_EVENT_CLS_TO_APP_TYPE_MAPPING[app_decorator_type]
         if not issubclass(event_cls, expected_base_event_cls):
             raise RuntimeError(
                 f'Application with type "{app_decorator_type}" '