Refactored get instead of literal dict retrieval.

src/flask_session/dynamodb/dynamodb.py

@@ -79,9 +79,9 @@ def __init__(
 
     def _retrieve_session_data(self, store_id: str) -> Optional[dict]:
         # Get the saved session (document) from the database
-        document = self.store.get_item(Key={"id": store_id})["Item"]
+        document = self.store.get_item(Key={"id": store_id}).get("Item")
         if document:
-            serialized_session_data = want_bytes(document["val"].value)
+            serialized_session_data = want_bytes(document.get("val").value)
             return self.serializer.decode(serialized_session_data)
         return None
 

tests/test_dynamodb.py

@@ -22,20 +22,20 @@ def setup_dynamodb(self):
         try:
             scan = self.store.scan()
             with self.store.batch_writer() as batch:
-                for each in scan["Items"]:
+                for each in scan.get("Items"):
                     batch.delete_item(
                         Key={
-                            "id": each["id"],
+                            "id": each.get("id"),
                         }
                     )
             yield
         finally:
             scan = self.store.scan()
             with self.store.batch_writer() as batch:
-                for each in scan["Items"]:
+                for each in scan.get("Items"):
                     batch.delete_item(
                         Key={
-                            "id": each["id"],
+                            "id": each.get("id"),
                         }
                     )
             pass