Make all lookup plugins work with lists

Lookup plugins should accept a string or a list, and always return
a list, even if it is just one item.

lib/ansible/runner/lookup_plugins/dnstxt.py

@@ -41,20 +41,24 @@ class LookupModule(object):
             raise errors.AnsibleError("Can't LOOKUP(dnstxt): module dns.resolver is not installed")
 
     def run(self, terms, **kwargs):
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            domain = term.split()[0]
+            string = []
+            try:
+                answers = dns.resolver.query(domain, 'TXT')
+                for rdata in answers:
+                    s = rdata.to_text()
+                    string.append(s[1:-1])  # Strip outside quotes on TXT rdata
 
-        domain = terms.split()[0]
-        string = []
-        try:
-            answers = dns.resolver.query(domain, 'TXT')
-            for rdata in answers:
-                s = rdata.to_text()
-                string.append(s[1:-1])  # Strip outside quotes on TXT rdata
-
-        except dns.resolver.NXDOMAIN:
-            string = 'NXDOMAIN'
-        except dns.resolver.Timeout:
-            string = ''
-        except dns.exception.DNSException as e:
-            raise errors.AnsibleError("dns.resolver unhandled exception", e)
-
-        return ''.join(string)
+            except dns.resolver.NXDOMAIN:
+                string = 'NXDOMAIN'
+            except dns.resolver.Timeout:
+                string = ''
+            except dns.exception.DNSException as e:
+                raise errors.AnsibleError("dns.resolver unhandled exception", e)
+
+            ret.append(''.join(string))
+        return ret

lib/ansible/runner/lookup_plugins/env.py

@@ -24,6 +24,10 @@ class LookupModule(object):
         self.basedir = basedir
 
     def run(self, terms, **kwargs):
-
-        var = terms.split()[0]
-        return os.getenv(var, '')
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            var = term.split()[0]
+            ret.append(os.getenv(var, ''))
+        return ret

lib/ansible/runner/lookup_plugins/file.py

@@ -24,7 +24,12 @@ class LookupModule(object):
         self.basedir = basedir
 
     def run(self, terms, **kwargs):
-        path = utils.path_dwim(self.basedir, terms)
-        if not os.path.exists(path):
-            raise errors.AnsibleError("%s does not exist" % path)
-        return [open(path).read().rstrip()]
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            path = utils.path_dwim(self.basedir, term)
+            if not os.path.exists(path):
+                raise errors.AnsibleError("%s does not exist" % path)
+            ret.append(open(path).read().rstrip())
+        return ret

lib/ansible/runner/lookup_plugins/fileglob.py

@@ -25,10 +25,13 @@ class LookupModule(object):
         self.basedir = basedir
 
     def run(self, terms, **kwargs):
-        dwimterms = utils.path_dwim(self.basedir, terms)
-        # This skips whatever prefix the dwim added, leaving just the filename for the item
-        dwim_prefix_len = len(dwimterms) - len(terms)
-        return [ f[dwim_prefix_len:] for f in glob.glob(dwimterms) if os.path.isfile(f) ]
-
-
-
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            dwimterms = utils.path_dwim(self.basedir, term)
+            # This skips whatever prefix the dwim added, leaving just the filename for the item
+            dwim_prefix_len = len(dwimterms) - len(term)
+            ret.extend([ f[dwim_prefix_len:]
+                         for f in glob.glob(dwimterms) if os.path.isfile(f) ])
+        return ret

lib/ansible/runner/lookup_plugins/items.py

@@ -21,7 +21,6 @@ class LookupModule(object):
         pass
 
     def run(self, terms, **kwargs):
-        return terms
-
-
-
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        return [term for term in terms]

lib/ansible/runner/lookup_plugins/lines.py

@@ -24,9 +24,14 @@ class LookupModule(object):
         self.basedir = basedir
 
     def run(self, terms, **kwargs):
-        p = subprocess.Popen(terms, cwd=self.basedir, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
-        (stdout, stderr) = p.communicate()
-        if p.returncode == 0:
-            return stdout.splitlines()
-        else:
-            raise errors.AnsibleError("lookup_plugin.lines(%s) returned %d" % (terms, p.returncode))
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            p = subprocess.Popen(term, cwd=self.basedir, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+            (stdout, stderr) = p.communicate()
+            if p.returncode == 0:
+                ret.extend(stdout.splitlines())
+            else:
+                raise errors.AnsibleError("lookup_plugin.lines(%s) returned %d" % (term, p.returncode))
+        return ret

lib/ansible/runner/lookup_plugins/pipe.py

@@ -24,9 +24,14 @@ class LookupModule(object):
         self.basedir = basedir
 
     def run(self, terms, **kwargs):
-        p = subprocess.Popen(terms, cwd=self.basedir, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
-        (stdout, stderr) = p.communicate()
-        if p.returncode == 0:
-            return [stdout.rstrip()]
-        else:
-            raise errors.AnsibleError("lookup_plugin.pipe(%s) returned %d" % (terms, p.returncode))
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            p = subprocess.Popen(term, cwd=self.basedir, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+            (stdout, stderr) = p.communicate()
+            if p.returncode == 0:
+                ret.append(stdout.rstrip())
+            else:
+                raise errors.AnsibleError("lookup_plugin.pipe(%s) returned %d" % (term, p.returncode))
+        return ret

lib/ansible/runner/lookup_plugins/redis_kv.py

@@ -40,28 +40,32 @@ class LookupModule(object):
             raise errors.AnsibleError("Can't LOOKUP(redis_kv): module redis is not installed")
 
     def run(self, terms, **kwargs):
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            (url,key) = term.split(',')
+            if url == "":
+                url = 'redis://localhost:6379'
 
-        (url,key) = terms.split(',')
-        if url == "":
-            url = 'redis://localhost:6379'
+            # urlsplit on Python 2.6.1 is broken. Hmm. Probably also the reason
+            # Redis' from_url() doesn't work here.
 
-        # urlsplit on Python 2.6.1 is broken. Hmm. Probably also the reason
-        # Redis' from_url() doesn't work here.
+            p = '(?P<scheme>[^:]+)://?(?P<host>[^:/ ]+).?(?P<port>[0-9]*).*'
 
-        p = '(?P<scheme>[^:]+)://?(?P<host>[^:/ ]+).?(?P<port>[0-9]*).*'
+            try:
+                m = re.search(p, url)
+                host = m.group('host')
+                port = int(m.group('port'))
+            except AttributeError:
+                raise errors.AnsibleError("Bad URI in redis lookup")
 
-        try:
-            m = re.search(p, url)
-            host = m.group('host')
-            port = int(m.group('port'))
-        except AttributeError:
-            raise errors.AnsibleError("Bad URI in redis lookup")
-
-        try:
-            conn = redis.Redis(host=host, port=port)
-            res = conn.get(key)
-            if res is None:
-                res = ""
-            return res
-        except:
-            return ""  # connection failed or key not found
+            try:
+                conn = redis.Redis(host=host, port=port)
+                res = conn.get(key)
+                if res is None:
+                    res = ""
+                ret.append(res)
+            except:
+                ret.append("")  # connection failed or key not found
+        return ret

lib/ansible/runner/lookup_plugins/template.py

@@ -23,5 +23,9 @@ class LookupModule(object):
         self.basedir = basedir
 
     def run(self, terms, inject=None, **kwargs):
-        return utils.template_from_file(self.basedir, terms, inject)
-
+        if isinstance(terms, basestring):
+            terms = [ terms ]
+        ret = []
+        for term in terms:
+            ret.append(utils.template_from_file(self.basedir, term, inject))
+        return ret