Reduce complexity of Templar._lookup slightly

changelogs/fragments/73277-reduce-lookup-complexity.yml

@@ -0,0 +1,3 @@
+minor_changes:
+- Templar - reduce the complexity of ``Templar._lookup``
+  (https://github.com/ansible/ansible/pull/73277)

lib/ansible/template/__init__.py

@@ -56,6 +56,7 @@
 from ansible.utils.collection_loader import AnsibleCollectionRef
 from ansible.utils.display import Display
 from ansible.utils.collection_loader._collection_finder import _get_collection_metadata
+from ansible.utils.listify import listify_lookup_plugin_terms
 from ansible.utils.unsafe_proxy import wrap_var
 
 display = Display()
@@ -1004,58 +1005,58 @@ def _query_lookup(self, name, *args, **kwargs):
     def _lookup(self, name, *args, **kwargs):
         instance = self._lookup_loader.get(name, loader=self._loader, templar=self)
 
-        if instance is not None:
-            wantlist = kwargs.pop('wantlist', False)
-            allow_unsafe = kwargs.pop('allow_unsafe', C.DEFAULT_ALLOW_UNSAFE_LOOKUPS)
-            errors = kwargs.pop('errors', 'strict')
+        if instance is None:
+            raise AnsibleError("lookup plugin (%s) not found" % name)
 
-            from ansible.utils.listify import listify_lookup_plugin_terms
-            loop_terms = listify_lookup_plugin_terms(terms=args, templar=self, loader=self._loader, fail_on_undefined=True, convert_bare=False)
-            # safely catch run failures per #5059
-            try:
-                ran = instance.run(loop_terms, variables=self._available_variables, **kwargs)
-            except (AnsibleUndefinedVariable, UndefinedError) as e:
-                raise AnsibleUndefinedVariable(e)
-            except Exception as e:
-                if self._fail_on_lookup_errors:
-                    msg = u"An unhandled exception occurred while running the lookup plugin '%s'. Error was a %s, original message: %s" % \
-                          (name, type(e), to_text(e))
-                    if errors == 'warn':
-                        display.warning(msg)
-                    elif errors == 'ignore':
-                        display.display(msg, log_only=True)
-                    else:
-                        raise AnsibleError(to_native(msg))
-                ran = [] if wantlist else None
+        wantlist = kwargs.pop('wantlist', False)
+        allow_unsafe = kwargs.pop('allow_unsafe', C.DEFAULT_ALLOW_UNSAFE_LOOKUPS)
+        errors = kwargs.pop('errors', 'strict')
 
-            if ran and not allow_unsafe:
-                if wantlist:
-                    ran = wrap_var(ran)
+        loop_terms = listify_lookup_plugin_terms(terms=args, templar=self, loader=self._loader, fail_on_undefined=True, convert_bare=False)
+        # safely catch run failures per #5059
+        try:
+            ran = instance.run(loop_terms, variables=self._available_variables, **kwargs)
+        except (AnsibleUndefinedVariable, UndefinedError) as e:
+            raise AnsibleUndefinedVariable(e)
+        except Exception as e:
+            if self._fail_on_lookup_errors:
+                msg = u"An unhandled exception occurred while running the lookup plugin '%s'. Error was a %s, original message: %s" % \
+                      (name, type(e), to_text(e))
+                if errors == 'warn':
+                    display.warning(msg)
+                elif errors == 'ignore':
+                    display.display(msg, log_only=True)
                 else:
-                    try:
-                        if self.jinja2_native and isinstance(ran[0], NativeJinjaText):
-                            ran = wrap_var(NativeJinjaText(",".join(ran)))
-                        else:
-                            ran = wrap_var(",".join(ran))
-                    except TypeError:
-                        # Lookup Plugins should always return lists.  Throw an error if that's not
-                        # the case:
-                        if not isinstance(ran, Sequence):
-                            raise AnsibleError("The lookup plugin '%s' did not return a list."
-                                               % name)
-
-                        # The TypeError we can recover from is when the value *inside* of the list
-                        # is not a string
-                        if len(ran) == 1:
-                            ran = wrap_var(ran[0])
-                        else:
-                            ran = wrap_var(ran)
-
-                if self.cur_context:
-                    self.cur_context.unsafe = True
-            return ran
-        else:
-            raise AnsibleError("lookup plugin (%s) not found" % name)
+                    raise AnsibleError(to_native(msg))
+            return [] if wantlist else None
+
+        if ran and allow_unsafe is False:
+            if self.cur_context:
+                self.cur_context.unsafe = True
+
+            if wantlist:
+                return wrap_var(ran)
+
+            try:
+                if self.jinja2_native and isinstance(ran[0], NativeJinjaText):
+                    ran = wrap_var(NativeJinjaText(",".join(ran)))
+                else:
+                    ran = wrap_var(",".join(ran))
+            except TypeError:
+                # Lookup Plugins should always return lists.  Throw an error if that's not
+                # the case:
+                if not isinstance(ran, Sequence):
+                    raise AnsibleError("The lookup plugin '%s' did not return a list."
+                                       % name)
+
+                # The TypeError we can recover from is when the value *inside* of the list
+                # is not a string
+                if len(ran) == 1:
+                    ran = wrap_var(ran[0])
+                else:
+                    ran = wrap_var(ran)
+
+        return ran
 
     def do_template(self, data, preserve_trailing_newlines=True, escape_backslashes=True, fail_on_undefined=None, overrides=None, disable_lookups=False):
         if self.jinja2_native and not isinstance(data, string_types):

lib/ansible/utils/listify.py

@@ -21,7 +21,6 @@
 
 from ansible.module_utils.six import string_types
 from ansible.module_utils.common._collections_compat import Iterable
-from ansible.template.safe_eval import safe_eval
 
 
 __all__ = ['listify_lookup_plugin_terms']