diff options
| -rwxr-xr-x | cloud-init-query.py | 55 | ||||
| -rw-r--r-- | cloudinit/__init__.py | 6 | ||||
| -rwxr-xr-x | setup.py | 3 |
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() @@ -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/*')), |