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.
This commit is contained in:
Scott Moser 2011-01-19 04:26:27 +00:00
parent bb37cc6a8e
commit 0f5358d4d9
3 changed files with 63 additions and 1 deletions

55
cloud-init-query.py Executable file
View File

@ -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()

View File

@ -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()

View File

@ -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/*')),