Fix how MissingTargetException is raised (GoogleCloudPlatform#87)

di · web-flow · commit ba6a869fe328 · 2020-10-20T17:50:37.000-05:00
diff --git a/src/functions_framework/__init__.py b/src/functions_framework/__init__.py
@@ -247,14 +247,13 @@ def handle_none(rv):
         app.make_response = handle_none
 
     # Extract the target function from the source file
-    try:
-        function = getattr(source_module, target)
-    except AttributeError:
+    if not hasattr(source_module, target):
         raise MissingTargetException(
             "File {source} is expected to contain a function named {target}".format(
                 source=source, target=target
             )
         )
+    function = getattr(source_module, target)
 
     # Check that it is a function
     if not isinstance(function, types.FunctionType):
