adding in cloud logs

Author: greyhoundforty

apps/cos-event-trigger/app.py

@@ -1,10 +1,67 @@
 #!/usr/bin/env python3
 from flask import Flask, request, jsonify
 import json
+import os
+import httpx
 from datetime import datetime
 
 app = Flask(__name__)
 
+# IBM Cloud Logging configuration
+IBM_INSTANCE_ID = os.environ.get("IBM_INSTANCE_ID")
+CE_REGION = os.environ.get("CE_REGION", "us-south")
+IBM_APP_NAME = os.environ.get("CE_APP", "cos-event-trigger")
+IBM_SUBSYSTEM_NAME = os.environ.get("CE_PROJECT_ID", "event-processor")
+IBM_LOG_SEVERITY = os.environ.get("IBM_LOG_SEVERITY", "info")
+
+def get_iam_token():
+    ibmcloud_api_key = os.environ.get('IBMCLOUD_API_KEY')
+    if not ibmcloud_api_key:
+        raise ValueError("IBMCLOUD_API_KEY environment variable not found")
+    hdrs = { 'Accept': 'application/json', 'Content-Type' : 'application/x-www-form-urlencoded' }
+    params = { 'grant_type' : 'urn:ibm:params:oauth:grant-type:apikey',
+            'apikey': ibmcloud_api_key }
+    resp = httpx.post('https://iam.cloud.ibm.com/identity/token', data = params, headers = hdrs)
+    # raise exception if invalid status
+    resp.raise_for_status()
+    json_payload = resp.json()
+    token = json_payload['access_token']
+    return token
+
+def send_to_ibm_logging(log_text, severity=None):
+    """Send log to IBM Cloud Logging"""
+    if not IBM_INSTANCE_ID:
+        print("IBM Cloud Logging not configured. Set IBM_INSTANCE_ID and IBM_IAM_TOKEN environment variables.")
+        return False
+    
+    url = f"https://{IBM_INSTANCE_ID}.ingress.{CE_REGION}.logs.cloud.ibm.com/logs/v1/singles"
+    
+    iam_token = get_iam_token()
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": iam_token
+    }
+    
+    payload = [{
+        "applicationName": IBM_APP_NAME,
+        "subsystemName": IBM_SUBSYSTEM_NAME,
+        "severity": severity or IBM_LOG_SEVERITY,
+        "text": log_text
+    }]
+    
+    try:
+        response = httpx.post(url, headers=headers, json=payload, timeout=10.0)
+        if response.status_code >= 200 and response.status_code < 300:
+            print(f"Successfully sent log to IBM Cloud Logging: {response.status_code}")
+            return True
+        else:
+            print(f"Failed to send log to IBM Cloud Logging: {response.status_code}, {response.text}")
+            return False
+    except Exception as e:
+        print(f"Error sending log to IBM Cloud Logging: {str(e)}")
+        return False
+
 # Equivalent to EventStats in Go
 class EventStats:
     def __init__(self):
@@ -46,6 +103,10 @@ def handle_event():
     print(f"{current_time} - Received:")
     print(f"\nBody: {body}")
 
+    # Send to IBM Cloud Logging
+    log_message = f"COS Event: {operation} on {bucket}/{key} - {body}"
+    send_to_ibm_logging(log_message)
+    
     return "OK"
 
 if __name__ == '__main__':

apps/cos-event-trigger/requirements.txt

@@ -1,8 +1,16 @@
+anyio==4.9.0
 blinker==1.9.0
+certifi==2025.1.31
 click==8.1.8
 flask==3.1.0
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+idna==3.10
 itsdangerous==2.2.0
 jinja2==3.1.6
 markupsafe==3.0.2
 pip==24.3.1
+sniffio==1.3.1
+typing-extensions==4.12.2
 werkzeug==3.1.3