#!/usr/bin/env python # vim: tabstop=4 shiftwidth=4 softtabstop=4 # Copyright (C) 2012 Yahoo! Inc. All Rights Reserved. # # 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 logging import logging.config import sys import time import traceback from devstack import colorlog from devstack import date from devstack import log as lg from devstack import opts from devstack import settings from devstack import shell as sh from devstack import utils from devstack.progs import actions from devstack.progs import common LOG = lg.getLogger("devstack.stack") # This is used to map an action to a useful string for # the welcome display _WELCOME_MAP = { settings.INSTALL: "INSTALLER", settings.UNINSTALL: "UNINSTALLER", settings.START: "STARTER", settings.STOP: "STOPPER", } def dump_config(config_obj): def item_format(key, value): return "\t%s=%s" % (str(key), str(value)) def map_print(mp): for key in sorted(mp.keys()): value = mp.get(key) LOG.info(item_format(key, value)) passwords_gotten = config_obj.pws full_cfgs = config_obj.configs_fetched db_dsns = config_obj.db_dsns if passwords_gotten or full_cfgs or db_dsns: if passwords_gotten: LOG.info("Passwords:") map_print(passwords_gotten) if full_cfgs: filtered = dict((k, v) for (k, v) in full_cfgs.items() if k not in passwords_gotten) if filtered: LOG.info("Configs:") map_print(filtered) if db_dsns: LOG.info("Data source names:") map_print(db_dsns) def run(args): (distro, platform) = utils.determine_distro() if distro is None: print("Unsupported platform " + utils.color_text(platform, "red") + "!") return False action = args.pop("action").strip().lower() if not (action in settings.ACTIONS): print(utils.color_text("No valid action specified!", "red")) return False rootdir = args.pop("dir") if not rootdir: print(utils.color_text("No root directory specified!", "red")) return False #welcome! (repeat_string, line_max_len) = utils.welcome(_WELCOME_MAP.get(action)) print(utils.center_text("Action Runner", repeat_string, line_max_len)) #here on out we should be using the logger (and not print) start_time = time.time() config = common.get_config() pkg_manager = common.get_packager(distro, args.get('keep_old')) components = utils.parse_components(args.pop("components")) runner = actions.ActionRunner(distro, action, rootdir, config, pkg_manager, components=components, **args) LOG.info("Starting action [%s] on %s for distro [%s]" % (action, date.rcf8222date(), distro)) runner.run() LOG.info("It took (%s) to complete action [%s]" % (common.format_secs_taken((time.time() - start_time)), action)) LOG.info("After action [%s] your settings which were created or read are:" % (action)) dump_config(config) return True def configure_logging(args): # Debug by default root_logger = logging.getLogger('') root_logger.setLevel(logging.DEBUG) # Set our pretty logger console_logger = logging.StreamHandler(sys.stdout) console_format = '%(levelname)s: @%(name)s : %(message)s' console_logger.setFormatter(colorlog.TermFormatter(console_format)) root_logger.addHandler(console_logger) # Adjust logging verbose level based on the command line switch. log_level = logging.DEBUG if args['verbosity'] >= 2 else logging.INFO root_logger.setLevel(log_level) def main(): #do this first so people can see the help message... args = opts.parse() prog_name = sys.argv[0] # Configure logging configure_logging(args) LOG.debug("Command line options %s" % (args)) #will need root to setup openstack if not sh.got_root(): rest_args = sys.argv[1:] print("This program requires a user with sudo access.") msg = "Perhaps you should try %s %s" % \ (utils.color_text("sudo %s" % (prog_name), "red", True), " ".join(rest_args)) print(msg) return 1 try: #drop to usermode sh.user_mode(False) started_ok = run(args) if not started_ok: me = utils.color_text(prog_name, "red", True) me += " " + utils.color_text('--help', 'red') print("Perhaps you should try %s" % (me)) return 1 else: utils.goodbye(True) return 0 except Exception: utils.goodbye(False) traceback.print_exc(file=sys.stdout) return 1 if __name__ == "__main__": sys.exit(main())