From 903e4f6eaeaa522a0acd9fe661e67403c05e2a1b Mon Sep 17 00:00:00 2001
From: Jeroen Hoekx <jeroen.hoekx@hamok.be>
Date: Thu, 19 Apr 2012 08:26:54 +0200
Subject: [PATCH] Support dicts in inventory vars.

---
 lib/ansible/inventory.py | 15 ++++++++++-----
 test/TestInventory.py    |  4 ++--
 test/yaml_hosts          |  4 +++-
 3 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/lib/ansible/inventory.py b/lib/ansible/inventory.py
index 3295f3f2dd8..18d64d60294 100644
--- a/lib/ansible/inventory.py
+++ b/lib/ansible/inventory.py
@@ -216,11 +216,16 @@ class Inventory(object):
 
     def _parse_yaml_host(self, item, variables=[]):
         def set_variables(host, variables):
-            for variable in variables:
-                if len(variable) != 1:
-                    raise errors.AnsibleError("Only one item expected in %s"%(variable))
-                k, v = variable.items()[0]
-                self._set_variable(host, k, v)
+            if type(variables) == list:
+                for variable in variables:
+                    if len(variable) != 1:
+                        raise errors.AnsibleError("Only one item expected in %s"%(variable))
+                    k, v = variable.items()[0]
+                    self._set_variable(host, k, v)
+            elif type(variables) == dict:
+                for k, v in variables.iteritems():
+                    self._set_variable(host, k, v)
+
 
         if type(item) in [str, unicode]:
             set_variables(item, variables)
diff --git a/test/TestInventory.py b/test/TestInventory.py
index 750cbf189ac..96991d77d00 100644
--- a/test/TestInventory.py
+++ b/test/TestInventory.py
@@ -220,13 +220,13 @@ class TestInventory(unittest.TestCase):
         inventory = self.yaml_inventory()
         vars = inventory.get_variables('saturn')
 
-        assert vars == {"moon":"titan"}
+        assert vars == {"moon":"titan", "moon2":"enceladus"}
 
     def test_yaml_port(self):
         inventory = self.yaml_inventory()
         vars = inventory.get_variables('hera')
 
-        assert vars == {'ansible_ssh_port': 3000}
+        assert vars == {'ansible_ssh_port': 3000, 'ntp_server': 'olympus.example.com'}
 
     ### Test Runner class method
 
diff --git a/test/yaml_hosts b/test/yaml_hosts
index 7568ff4bda2..628c2843506 100644
--- a/test/yaml_hosts
+++ b/test/yaml_hosts
@@ -3,7 +3,8 @@
 - jupiter
 - host: saturn
   vars:
-  - moon: titan
+    moon: titan
+    moon2: enceladus
 
 - zeus
 
@@ -14,6 +15,7 @@
   - poseidon
   vars:
   - ansible_ssh_port: 3000
+  - ntp_server: olympus.example.com
 
 - group: norse
   hosts: