os-collect-config/os_collect_config/collect.py

77 lines
2.3 KiB
Python

# Copyright (c) 2013 Hewlett-Packard Development Company, L.P.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import json
import os
import subprocess
from openstack.common import log
from os_collect_config import cache
from os_collect_config import cfn
from os_collect_config import common
from os_collect_config import ec2
from oslo.config import cfg
opts = [
cfg.StrOpt('command',
short='c',
help='Command to run on metadata changes.'),
cfg.StrOpt('cachedir',
default='/var/run/os-collect-config',
help='Directory in which to store local cache of metadata'),
]
CONF = cfg.CONF
logger = log.getLogger('os-collect-config')
def setup_conf():
ec2_group = cfg.OptGroup(name='ec2',
title='EC2 Metadata options')
cfn_group = cfg.OptGroup(name='cfn',
title='CloudFormation API Metadata options')
CONF.register_group(ec2_group)
CONF.register_group(cfn_group)
CONF.register_cli_opts(ec2.opts, group='ec2')
CONF.register_cli_opts(cfn.opts, group='cfn')
CONF.register_cli_opts(opts)
def __main__(ec2_requests=common.requests):
setup_conf()
CONF(prog="os-collect-config")
log.setup("os-collect-config")
ec2_content = ec2.CollectEc2(requests_impl=ec2_requests).collect()
if CONF.command:
(changed, ec2_path) = cache.store('ec2', ec2_content)
if changed:
paths = [ec2_path]
env = dict(os.environ)
env["OS_CONFIG_FILES"] = ':'.join(paths)
logger.info("Executing %s" % CONF.command)
subprocess.call(CONF.command, env=env, shell=True)
cache.commit('ec2')
else:
content = {'ec2': ec2_content}
print json.dumps(content, indent=1)
if __name__ == '__main__':
__main__()