eoapi-cdk 11.4.1 → 11.6.0

package/lib/ingestor-api/runtime/src/handler.py

@@ -1,9 +1,32 @@
-"""
-Entrypoint for Lambda execution.
-"""
+"""Entrypoint for Lambda execution."""
+
+import asyncio
+from typing import Any
 
 from mangum import Mangum
 
 from .main import app
 
-handler = Mangum(app, lifespan="off", api_gateway_base_path=app.root_path)
+
+def _ensure_event_loop() -> asyncio.AbstractEventLoop:
+    """Return the current event loop, creating and installing one if needed."""
+    try:
+        return asyncio.get_running_loop()
+    except RuntimeError:
+        pass
+
+    try:
+        return asyncio.get_event_loop()
+    except RuntimeError:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        return loop
+
+
+_asgi_handler = Mangum(app, lifespan="off", api_gateway_base_path=app.root_path)
+
+
+def handler(event: Any, context: Any) -> dict[str, Any]:
+    """Handle AWS Lambda events with a guaranteed current event loop."""
+    _ensure_event_loop()
+    return _asgi_handler(event, context)

package/lib/ingestor-api/runtime/src/ingestor.py

@@ -42,7 +42,7 @@ def update_dynamodb(
     with table.batch_writer(overwrite_by_pkeys=["created_by", "id"]) as batch:
         for ingestion in ingestions:
             batch.put_item(
-                Item=ingestion.copy(
+                Item=ingestion.model_copy(
                     update={
                         "status": status,
                         "message": message,

package/lib/ingestor-api/runtime/src/schemas.py

@@ -3,17 +3,15 @@ import binascii
 import enum
 import json
 from datetime import datetime
-from typing import TYPE_CHECKING, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 from urllib.parse import urlparse
 
 from fastapi.encoders import jsonable_encoder
-from fastapi.exceptions import RequestValidationError
 from pydantic import (
     BaseModel,
     Json,
     PositiveInt,
     dataclasses,
-    error_wrappers,
     field_validator,
 )
 from stac_pydantic import Collection, Item, shared
@@ -107,25 +105,19 @@ class Ingestion(BaseModel):
 class ListIngestionRequest:
     status: Status = Status.queued
     limit: Optional[PositiveInt] = None
-    next: Optional[str] = None
-
-    def __post_init_post_parse__(self) -> None:
-        # https://github.com/tiangolo/fastapi/issues/1474#issuecomment-1049987786
-        if self.next is None:
-            return
+    next: Optional[Any] = None
 
+    @field_validator("next", mode="before")
+    @classmethod
+    def decode_next_token(cls, v: Optional[str]) -> Optional[Any]:
+        """Decode the base64-encoded JSON pagination token supplied as a query param."""
+        if v is None:
+            return None
         try:
-            self.next = json.loads(base64.b64decode(self.next))
+            return json.loads(base64.b64decode(v))
         except (UnicodeDecodeError, binascii.Error) as e:
-            raise RequestValidationError(
-                [
-                    error_wrappers.ErrorWrapper(
-                        ValueError(
-                            "Unable to decode next token. Should be base64 encoded JSON"
-                        ),
-                        "query.next",
-                    )
-                ]
+            raise ValueError(
+                "Unable to decode next token. Should be base64 encoded JSON"
             ) from e
 
 

package/lib/ingestor-api/runtime/src/services.py

@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, List
 
 from boto3.dynamodb import conditions
-from pydantic import parse_obj_as
+from pydantic import TypeAdapter
 
 from . import schemas
 
@@ -35,7 +35,9 @@ class Database:
             **{"ExclusiveStartKey": next} if next else {},
         )
         return {
-            "items": parse_obj_as(List[schemas.Ingestion], response["Items"]),
+            "items": TypeAdapter(List[schemas.Ingestion]).validate_python(
+                response["Items"]
+            ),
             "next": response.get("LastEvaluatedKey"),
         }