From 256377166a70e110c27eb721e05d875ebcf74740 Mon Sep 17 00:00:00 2001
From: Michael DeHaan <michael.dehaan@gmail.com>
Date: Sun, 6 May 2012 14:47:05 -0400
Subject: [PATCH] Reinstate external inventory script support this time using
 the new more OO-ey inventory system. Next up: YAML format.

---
 lib/ansible/inventory.py | 26 ++++++++++++++++++++++++++
 test/TestInventory.py    |  8 ++------
 2 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/lib/ansible/inventory.py b/lib/ansible/inventory.py
index 65bc8c7f4f..38a094e984 100644
--- a/lib/ansible/inventory.py
+++ b/lib/ansible/inventory.py
@@ -18,9 +18,12 @@
 #############################################
 
 import fnmatch
+import os
 
 import constants as C
+import subprocess
 from ansible.inventory_parser import InventoryParser
+from ansible.inventory_script import InventoryScript
 from ansible.group import Group
 from ansible.host import Host
 from ansible import errors
@@ -36,11 +39,18 @@ class Inventory(object):
         # FIXME: re-support YAML inventory format
         # FIXME: re-support external inventory script (?)
 
+        self.host_list = host_list
         self.groups = []
         self._restriction = None
+        self._is_script = False
+
         if host_list:
             if type(host_list) == list:
                 self.groups = self._groups_from_override_hosts(host_list)
+            elif os.access(host_list, os.X_OK):
+                self._is_script = True
+                self.parser = InventoryScript(filename=host_list)
+                self.groups = self.parser.groups.values()
             else:
                 self.parser = InventoryParser(filename=host_list)
                 self.groups = self.parser.groups.values()
@@ -93,6 +103,22 @@ class Inventory(object):
         return group.get_variables()
 
     def get_variables(self, hostname):
+
+        if self._is_script:
+            # TODO: move this to inventory_script.py 
+            host = self.get_host(hostname)
+            cmd = subprocess.Popen(
+                [self.host_list,"--host",hostname], 
+                stdout=subprocess.PIPE, 
+                stderr=subprocess.PIPE
+            )
+            (out, err) = cmd.communicate()
+            results = utils.parse_json(out)
+            results['inventory_hostname'] = hostname
+            groups = [ g.name for g in host.get_groups() if g.name != 'all' ]
+            results['group_names'] = groups
+            return results
+
         host = self.get_host(hostname)
         if host is None:
             raise Exception("host not found: %s" % hostname)
diff --git a/test/TestInventory.py b/test/TestInventory.py
index 40312a500c..caaaf41d15 100644
--- a/test/TestInventory.py
+++ b/test/TestInventory.py
@@ -108,7 +108,6 @@ class TestInventory(unittest.TestCase):
     ### Inventory API tests
 
     def test_script(self):
-        raise SkipTest
         inventory = self.script_inventory()
         hosts = inventory.list_hosts()
 
@@ -119,7 +118,6 @@ class TestInventory(unittest.TestCase):
         assert sorted(hosts) == sorted(expected_hosts)
 
     def test_script_all(self):
-        raise SkipTest
         inventory = self.script_inventory()
         hosts = inventory.list_hosts('all')
 
@@ -127,7 +125,6 @@ class TestInventory(unittest.TestCase):
         assert sorted(hosts) == sorted(expected_hosts)
 
     def test_script_norse(self):
-        raise SkipTest
         inventory = self.script_inventory()
         hosts = inventory.list_hosts("norse")
 
@@ -135,7 +132,6 @@ class TestInventory(unittest.TestCase):
         assert sorted(hosts) == sorted(expected_hosts)
 
     def test_script_combined(self):
-        raise SkipTest
         inventory = self.script_inventory()
         hosts = inventory.list_hosts("norse:greek")
 
@@ -143,7 +139,6 @@ class TestInventory(unittest.TestCase):
         assert sorted(hosts) == sorted(expected_hosts)
 
     def test_script_restrict(self):
-        raise SkipTest
         inventory = self.script_inventory()
 
         restricted_hosts = ['hera', 'poseidon', 'thor']
@@ -160,10 +155,11 @@ class TestInventory(unittest.TestCase):
         assert sorted(hosts) == sorted(expected_hosts)
 
     def test_script_vars(self):
-        raise SkipTest
         inventory = self.script_inventory()
         vars = inventory.get_variables('thor')
 
+        print "VARS=%s" % vars
+
         assert vars == {'hammer':True,
                         'group_names': ['norse'],
                         'inventory_hostname': 'thor'}