add support api gateway context

krddevdays · Dec 3, 2023 · 08f5095 · 08f5095
1 parent b39df70
commit 08f5095
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/krddevdays/wsgi.py b/krddevdays/wsgi.py
@@ -1,4 +1,6 @@
 import os
+import base64
+import json
 
 import dotenv
 from django.core.wsgi import get_wsgi_application
@@ -11,7 +13,8 @@
 
 
 def application(environ, start_response):
-    script_name = environ.get('HTTP_X_SCRIPT_NAME', '')
+    api_gateway_context = json.loads(base64.b64decode(environ.get('HTTP_X-YC_APIGATEWAY_OPERATION_CONTEXT', 'e30=')).decode('utf8').replace("'", '"'))
+    script_name = environ.get('HTTP_X_SCRIPT_NAME', api_gateway_context['X-Script-Name'] or '')
     if script_name:
         environ['SCRIPT_NAME'] = script_name
         path_info = environ['PATH_INFO']