From 667cae15dfc8199de8add9d7837bf52113f09ca2 Mon Sep 17 00:00:00 2001
From: dmccue <dmccue@gmail.com>
Date: Thu, 23 Jul 2015 15:18:58 +0100
Subject: [PATCH] support for _meta retrieval

---
 contrib/inventory/cobbler.py | 42 ++++++++++++++++++++----------------
 1 file changed, 24 insertions(+), 18 deletions(-)

diff --git a/contrib/inventory/cobbler.py b/contrib/inventory/cobbler.py
index f352c8cf9d2..14b7540e188 100755
--- a/contrib/inventory/cobbler.py
+++ b/contrib/inventory/cobbler.py
@@ -30,9 +30,14 @@ See http://ansible.github.com/api.html for more info
 Tested with Cobbler 2.0.11.
 
 Changelog:
+    - 2015-06-21 dmccue: Heavily modified to support run-once _meta retrieval, results in
+         higher performance at ansible startup.  Groups are determined by owner rather than
+         default mgmt_classes.  DNS name determined from hostname.
+
     - 2013-09-01 pgehres: Refactored implementation to make use of caching and to
         limit the number of connections to external cobbler server for performance.
         Added use of cobbler.ini file to configure settings. Tested with Cobbler 2.4.0
+
 """
 
 # (c) 2012, Michael DeHaan <michael.dehaan@gmail.com>
@@ -54,7 +59,6 @@ Changelog:
 
 ######################################################################
 
-
 import argparse
 import ConfigParser
 import os
@@ -75,6 +79,7 @@ except ImportError:
 class CobblerInventory(object):
 
     def __init__(self):
+
         """ Main execution path """
         self.conn = None
 
@@ -97,15 +102,15 @@ class CobblerInventory(object):
         data_to_print = ""
 
         # Data to print
-        if self.args.host:
-            data_to_print = self.get_host_info()
+        #if self.args.host:
+        #    data_to_print = self.get_host_info()
 
-        elif self.args.list:
-            # Display list of instances for inventory
-            data_to_print = self.json_format_dict(self.inventory, True)
+        self.inventory['_meta'] = { 'hostvars': {} }
+        for hostname in self.cache:
+            self.inventory['_meta']['hostvars'][hostname] = {'cobbler': self.cache[hostname] }
 
-        else:  # default action with no options
-            data_to_print = self.json_format_dict(self.inventory, True)
+
+        data_to_print = self.json_format_dict(self.inventory, True)
 
         print data_to_print
 
@@ -144,7 +149,7 @@ class CobblerInventory(object):
 
         parser = argparse.ArgumentParser(description='Produce an Ansible Inventory file based on Cobbler')
         parser.add_argument('--list', action='store_true', default=True, help='List instances (default: True)')
-        parser.add_argument('--host', action='store', help='Get all the variables about a specific instance')
+        #parser.add_argument('--host', action='store', help='Get all the variables about a specific instance')
         parser.add_argument('--refresh-cache', action='store_true', default=False,
                             help='Force refresh of cache by making API requests to cobbler (default: False - use cache files)')
         self.args = parser.parse_args()
@@ -160,21 +165,22 @@ class CobblerInventory(object):
 
         for host in data:
             # Get the FQDN for the host and add it to the right groups
-            dns_name = None
+            dns_name = host['hostname'] #None
             ksmeta = None
             interfaces = host['interfaces']
-            for (iname, ivalue) in interfaces.iteritems():
-                if ivalue['management']:
-                    this_dns_name = ivalue.get('dns_name', None)
-                    if this_dns_name is not None and this_dns_name is not "":
-                        dns_name = this_dns_name
+            #for (iname, ivalue) in interfaces.iteritems():
+            #    if ivalue['management']:
+            #        this_dns_name = ivalue.get('dns_name', None)
+            #        #this_dns_name = ivalue.get('ip_address', None)
+            #        if this_dns_name is not None and this_dns_name is not "":
+            #            dns_name = this_dns_name
 
             if dns_name is None:
                 continue
 
             status = host['status']
             profile = host['profile']
-            classes = host['mgmt_classes']
+            classes = host['owners'] #host['mgmt_classes']
 
             if status not in self.inventory:
                 self.inventory[status] = []
@@ -193,7 +199,7 @@ class CobblerInventory(object):
 
             # The old way was ksmeta only -- provide backwards compatibility
 
-            self.cache[dns_name] = dict()
+            self.cache[dns_name] = host #dict()   # sub dict with host to output json
             if "ks_meta" in host:
                 for key, value in host["ks_meta"].iteritems():
                     self.cache[dns_name][key] = value
@@ -242,7 +248,7 @@ class CobblerInventory(object):
 
     def write_to_cache(self, data, filename):
         """ Writes data in JSON format to a file """
-
+        if data: print "DEBUG: data = " + str(data)
         json_data = self.json_format_dict(data, True)
         cache = open(filename, 'w')
         cache.write(json_data)