Create proxy_server.py

proxy_server.py

@@ -0,0 +1,70 @@
+from fastapi import FastAPI, HTTPException
+import httpx
+import os
+from pydantic import BaseModel
+import uvicorn
+
+app = FastAPI()
+
+# Pydantic model for earthquake request
+class EarthquakeRequest(BaseModel):
+    start_date: str
+    end_date: str
+
+@app.post("/api/proxy")
+async def proxy_to_lambda(earthquake_request: EarthquakeRequest):
+    """Proxy requests to your API Gateway Lambda"""
+    try:
+        # Get API key from environment variables
+        api_key = os.environ.get('API_GATEWAY_KEY')
+        
+        if not api_key:
+            raise HTTPException(status_code=500, detail="API key not configured")
+        
+        # Your actual API Gateway URL
+        api_url = os.environ.get('API_GATEWAY_URL')
+        
+        if not api_url:
+            raise HTTPException(status_code=500, detail="API Gateway URL not configured")
+        
+        # Headers to send to your API Gateway
+        headers = {
+            'Content-Type': 'application/json',
+            'X-API-Key': api_key
+        }
+        
+        # Request body
+        request_body = {
+            'start_date': earthquake_request.start_date,
+            'end_date': earthquake_request.end_date
+        }
+        
+        # Make request to your API Gateway
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.post(
+                api_url,
+                json=request_body,
+                headers=headers
+            )
+            
+            # Return the response from your API Gateway
+            if response.status_code == 200:
+                return response.json()
+            else:
+                raise HTTPException(
+                    status_code=response.status_code, 
+                    detail=f"API Gateway error: {response.text}"
+                )
+                
+    except httpx.TimeoutException:
+        raise HTTPException(status_code=504, detail="Request timeout")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Proxy error: {str(e)}")
+
+# Health check endpoint
+@app.get("/api/health")
+async def health_check():
+    return {"status": "healthy", "service": "earthquake-proxy"}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)