moving flag set out of get_binding

Author: hallvictoria

azure_functions_worker/bindings/meta.py

@@ -224,10 +224,6 @@ def get_deferred_binding(bind_name: str,
 
         # Checks if pytype is a supported sdk type
         if DEFERRED_BINDINGS_REGISTRY.check_supported_type(pytype):
-            # Set flag once
-            global DEFERRED_BINDINGS_ENABLED
-            if not DEFERRED_BINDINGS_ENABLED:
-                DEFERRED_BINDINGS_ENABLED = True
             # Returns deferred binding converter
             binding = DEFERRED_BINDINGS_REGISTRY.get(bind_name)
 
@@ -237,6 +233,7 @@ def get_deferred_binding(bind_name: str,
         # This will catch if DEFERRED_BINDINGS_REGISTRY is None
         # It will be None if the library isn't imported
         # Ensure that the flag is set to False in this case
+        global DEFERRED_BINDINGS_ENABLED
         DEFERRED_BINDINGS_ENABLED = False
 
 

azure_functions_worker/functions.py

@@ -144,6 +144,9 @@ def validate_function_params(params: dict, bound_params: dict,
             param_has_anno = param.name in annotations
             param_anno = annotations.get(param.name)
 
+            # Check the declared type
+            set_deferred_bindings_flag(param_anno)
+
             if param_has_anno:
                 if typing_inspect.is_generic_type(param_anno):
                     param_anno_origin = typing_inspect.get_origin(param_anno)
@@ -423,3 +426,14 @@ def add_indexed_function(self, function):
                                                           input_types,
                                                           output_types,
                                                           return_type)
+
+
+def set_deferred_bindings_flag(param_anno: type):
+    # If flag hasn't already been set
+    # If DEFERRED_BINDINGS_REGISTRY is not None
+    # If the binding type is a deferred binding type
+    if (not bindings_utils.meta.DEFERRED_BINDINGS_ENABLED
+            and bindings_utils.meta.DEFERRED_BINDINGS_REGISTRY is not None
+            and bindings_utils.meta.DEFERRED_BINDINGS_REGISTRY.check_supported_type(
+                param_anno)):
+        bindings_utils.meta.DEFERRED_BINDINGS_ENABLED = True