chore: update policy

Signed-off-by: Adetokunbo Ige <[email protected]>

Author: Adetokunbo Ige

todo-app/lambda_function/lambda.py

@@ -1,115 +1,116 @@
-import os
 import logging
+import os
+
 import boto3
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from mangum import Mangum
-
 from pydantic import BaseModel
 from typing import List, Dict
 
+# Configure logging
 logging.basicConfig(level=logging.DEBUG)
 
+# Initialize FastAPI application
 app = FastAPI()
 
-# Enable CORS
+# Configure CORS middleware
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Adjust this as needed for your use case
+    allow_origins=["*"],  # Adjust the allowed origins for your use case
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
 # Connect to DynamoDB
-dynamodb = boto3.resource('dynamodb', region_name='us-east-1')
+dynamodb = boto3.resource("dynamodb", region_name="us-east-1")
 table = dynamodb.Table("todo-dev")
 
-@app.get("/todos", response_model=List[dict])
+
+# Define a Pydantic model for a todo item
+class TodoItem(BaseModel):
+    id: str
+    text: str
+    timestamp: int
+
+
+@app.get("/todos", response_model=List[TodoItem])
 async def get_todos():
     try:
         # Perform a scan operation on the DynamoDB table
         response = table.scan()
-        # Return the 'Items' from the scan result, which contains the table data
-        return response.get('Items', [])
+        # Return the 'Items' from the scan result
+        items = response.get("Items", [])
+        logging.debug(f"Fetched {len(items)} items from DynamoDB")
+        return items
     except Exception as e:
         logging.error(f"Error getting todos: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
-
-# POST request to create a new todo
-@app.post("/todos", status_code=201)
-async def create_todo(todo: Dict):
-    if not todo:
-        raise HTTPException(status_code=400, detail="Invalid input")
+        raise HTTPException(status_code=500, detail="Error fetching todos")
 
-    # Ensure 'id' is present in the todo item, or else raise an error
-    if 'id' not in todo:
-        raise HTTPException(status_code=400, detail="Missing 'id' in request body")
 
+@app.post("/todos", status_code=201, response_model=TodoItem)
+async def create_todo(todo: TodoItem):
     try:
         # Add the todo item to DynamoDB
-        table.put_item(Item=todo)
-        return todo  # Return the created todo item
-
+        response = table.put_item(Item=todo.dict())
+        logging.debug(f"DynamoDB put_item response: {response}")
+        return todo
+    except boto3.exceptions.Boto3Error as boto_error:
+        logging.error(f"Boto3Error: {boto_error}")
+        raise HTTPException(status_code=500, detail=f"DynamoDB Error: {str(boto_error)}")
     except Exception as e:
-        logging.error(f"Error creating todo: {e}")
-        raise HTTPException(status_code=500, detail="Internal server error")
-
-# PUT request to update an existing todo
-@app.put("/todos/{id}")
-async def update_todo(id: str, todo: Dict):
-    # Validate that 'text' is in the request body
-    if 'text' not in todo:
+        logging.error(f"Unexpected error creating todo: {e}")
+        raise HTTPException(status_code=500, detail="Error creating todo")
+
+
+@app.put("/todos/{id}", response_model=TodoItem)
+async def update_todo(id: str, todo: Dict[str, str]):
+    if "text" not in todo:
         raise HTTPException(status_code=400, detail="Missing 'text' in request body")
 
     try:
-        # Update the todo item in DynamoDB based on the 'id'
+        # Update the todo item in DynamoDB
         response = table.update_item(
-            Key={'id': id},
-            UpdateExpression='SET #t = :t',
-            ExpressionAttributeNames={'#t': 'text'},
-            ExpressionAttributeValues={':t': todo['text']},
-            ReturnValues="ALL_NEW"  # Return the updated item
+            Key={"id": id},
+            UpdateExpression="SET #t = :t",
+            ExpressionAttributeNames={"#t": "text"},
+            ExpressionAttributeValues={":t": todo["text"]},
+            ReturnValues="ALL_NEW"
         )
-
-        # Check if the item was updated and return the updated todo
-        updated_todo = response.get('Attributes')
+        updated_todo = response.get("Attributes")
         if not updated_todo:
             raise HTTPException(status_code=404, detail="Todo not found")
-
         return updated_todo
-
     except Exception as e:
         logging.error(f"Error updating todo: {e}")
-        raise HTTPException(status_code=500, detail="Internal server error")
+        raise HTTPException(status_code=500, detail="Error updating todo")
+
 
-# DELETE request to delete a todo
 @app.delete("/todos/{id}", status_code=204)
 async def delete_todo(id: str):
     try:
-        # Attempt to delete the todo item from DynamoDB
-        response = table.delete_item(Key={'id': id})
-
-        # Check if the item exists based on the HTTP status code
-        if response.get('ResponseMetadata', {}).get('HTTPStatusCode') != 200:
+        # Delete the todo item from DynamoDB
+        response = table.delete_item(Key={"id": id})
+        if response.get("ResponseMetadata", {}).get("HTTPStatusCode") != 200:
             raise HTTPException(status_code=404, detail="Todo not found")
-
-        return {"detail": "Todo deleted successfully"}  # Confirm deletion
-
+        return {"detail": "Todo deleted successfully"}
     except Exception as e:
         logging.error(f"Error deleting todo: {e}")
-        raise HTTPException(status_code=500, detail="Internal server error")
+        raise HTTPException(status_code=500, detail="Error deleting todo")
+
+
 @app.get("/health")
 async def health():
     try:
         return {"message": "Everything looks good!"}
     except Exception as e:
         logging.error(f"Health check error: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail="Error performing health check")
+
 
 # Define the Lambda handler function
 def handler(event, context):
-    """ Entry point for AWS Lambda """
     logging.info(f"Received event: {event}")
     mangum_handler = Mangum(app)
     return mangum_handler(event, context)