refactoring preflight and cors

Author: maceto

functions/adoption/libs/network.py

@@ -0,0 +1,28 @@
+
+"""
+Network
+
+Handles formatting responses to match the tuple pattern required by
+the flask/GCP wrapper for Cloud Functions.
+"""
+
+PREFLIGHT_HEADERS = {
+    "Access-Control-Allow-Origin": "*",
+    "Access-Control-Allow-Methods": "GET",
+    "Access-Control-Allow-Headers": "Content-Type",
+    "Access-Control-Max-Age": "3600",
+  }
+
+HEADERS = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
+
+def respond_cors():
+  """
+  To be used to return OPTIONS responses to satisfy CORS preflight requests.
+  """
+  return ("", 204, PREFLIGHT_HEADERS)
+
+def respond(data, status=200):
+  """
+  To be used to return responses to satisfy CORS requests.
+  """
+  return (data, status, HEADERS)

functions/adoption/main.py

@@ -2,24 +2,13 @@
 from .libs.validator import Validator
 from .libs.utils import output
 from .libs.queries import list_data
+from .libs.network import respond_cors, respond
 
 @functions_framework.http
 def dispatcher(request):
-  # For more information about CORS and CORS preflight requests, see:
-  # https://developer.mozilla.org/en-US/docs/Glossary/Preflight_request
 
-  # Set CORS headers for the preflight request
   if request.method == "OPTIONS":
-      # Allows GET requests from any origin with the Content-Type
-      # header and caches preflight response for an 3600s
-      headers = {
-          "Access-Control-Allow-Origin": "*",
-          "Access-Control-Allow-Methods": "GET",
-          "Access-Control-Allow-Headers": "Content-Type",
-          "Access-Control-Max-Age": "3600",
-      }
-
-      return ("", 204, headers)
+    return respond_cors()
 
   # Set CORS headers for the main request
   headers = {"Access-Control-Allow-Origin": "*"}