add helper for print logging

app/main.py

@@ -14,7 +14,7 @@ from openai import OpenAI
 import psycopg2
 from psycopg2 import sql
 import os
-from app.utils import get_api_key, get_user_info, update_user, upload_file_to_s3, get_user, upload_mementos_to_db
+from app.utils import get_api_key, get_user_info, print_log, update_user, upload_file_to_s3, get_user, upload_mementos_to_db
 from dotenv import load_dotenv
 import logging.config
 import time
@@ -239,6 +239,7 @@ class ErrorResponse(BaseModel):
 
 @app.get("/ok")
 def ok_endpoint():
+    print_log("INFO", "health check endpoint")
     logger.info("Health check endpoint called", extra={"endpoint": "/ok"})
     return {"message": "ok"}
 

app/utils.py

@@ -17,6 +17,7 @@ from psycopg2 import sql
 import os
 from dotenv import load_dotenv
 from app.user import User
+import datetime
 
 load_dotenv()
 
@@ -254,4 +255,24 @@ def upload_mementos_to_db(user_id):
         raise ConnectionError(f"Database error: {str(e)}")
     except Exception as e:
         logger.error(f"Unexpected error uploading mementos: {str(e)}", extra={'user_id': user_id, 'endpoint': function_name})
-        return False
+        return False
+
+def print_log(level, message, **kwargs):
+    """
+    Print log in JSON format for better readability in CloudWatch.
+    
+    Parameters:
+        level (str): The log level (e.g., "INFO", "ERROR", "DEBUG").
+        message (str): The log message.
+        **kwargs: Additional key-value pairs to include in the log.
+        
+    example:
+    print_log("INFO", "User logged in", user_id=123, action="login")
+    """
+    log_entry = {
+        "timestamp": datetime.datetime.utcnow().isoformat() + "Z",
+        "level": level,
+        "message": message,
+    }
+    log_entry.update(kwargs)
+    print(json.dumps(log_entry, ensure_ascii=False))