Execute the source module w/in the app context

GoogleCloudPlatform · Jul 31, 2020 · cdd9c17 · cdd9c17
1 parent 0ef9bcd
commit cdd9c17
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/functions_framework/__init__.py b/src/functions_framework/__init__.py
@@ -168,12 +168,14 @@ def create_app(target=None, source=None, signature_type=None):
     # 4. Add the module to sys.modules
     sys.modules[name] = source_module
 
-    # 5. Execute the module
-    spec.loader.exec_module(source_module)
-
+    # 5. Create the application
     app = flask.Flask(target, template_folder=template_folder)
     app.config["MAX_CONTENT_LENGTH"] = MAX_CONTENT_LENGTH
 
+    # 6. Execute the module, within the application context
+    with app.app_context():
+        spec.loader.exec_module(source_module)
+
     # Extract the target function from the source file
     try:
         function = getattr(source_module, target)