add 'cloud-init-query' tool, to query fields from cloud data

At this point, this is appears much like a cripped 'ec2metdata' tool.
However, it does provide a tool interface to some fields independent
of their DataSource.

3 files changed, 63 insertions, 1 deletions

diff --git a/cloud-init-query.py b/cloud-init-query.py
new file mode 100755
index 00000000..3e8c24ab
--- /dev/null
+++ b/cloud-init-query.py
@@ -0,0 +1,55 @@
+#!/usr/bin/python
+# vi: ts=4 expandtab
+#
+#    Copyright (C) 2009-2010 Canonical Ltd.
+#
+#    Author: Scott Moser <scott.moser@canonical.com>
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License version 3, as
+#    published by the Free Software Foundation.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import sys
+import cloudinit
+import cloudinit.CloudConfig
+import logging
+import os
+import traceback
+
+def Usage(out = sys.stdout):
+    out.write("Usage: %s name\n" % sys.argv[0])
+    
+def main():
+    # expect to be called with name of item to fetch
+    if len(sys.argv) != 2:
+        Usage(sys.stderr)
+        sys.exit(1)
+
+    cc = cloudinit.CloudConfig.CloudConfig(cloudinit.cloud_config)
+    cloud_config = cc.cfg
+    data = {
+        'user_data' : cc.cloud.get_userdata(),
+        'user_data_raw' : cc.cloud.get_userdata_raw(),
+        'instance_id' : cc.cloud.get_instance_id(),
+    }
+
+    name = sys.argv[1].replace('-','_')
+
+    if name not in data:
+        sys.stderr.write("unknown name '%s'.  Known values are:\n  %s\n" %
+            (sys.argv[1], ' '.join(data.keys())))
+        sys.exit(1)
+
+    print data[name]
+    sys.exit(0)
+
+if __name__ == '__main__':
+    main()
diff --git a/cloudinit/__init__.py b/cloudinit/__init__.py
index 7b300a63..fd20c7be 100644
--- a/cloudinit/__init__.py
+++ b/cloudinit/__init__.py
@@ -203,6 +203,12 @@ class CloudInit:
     def get_userdata(self):
         return(self.datasource.get_userdata())
 
+    def get_userdata_raw(self):
+        return(self.datasource.get_userdata_raw())
+
+    def get_instance_id(self):
+        return(self.datasource.get_instance_id())
+
     def update_cache(self):
         self.write_to_cache()
         self.store_userdata()
diff --git a/setup.py b/setup.py
index 6ec5a2a1..ab37367a 100755
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,8 @@ setup(name='cloud-init',
       packages=['cloudinit', 'cloudinit.CloudConfig' ],
       scripts=['cloud-init.py',
                'cloud-init-run-module.py',
-               'cloud-init-cfg.py'
+               'cloud-init-cfg.py',
+               'cloud-init-query.py'
                ],
       data_files=[('/etc/cloud', glob('config/*.cfg')),
                   ('/etc/cloud/templates', glob('templates/*')),