149 lines
4.6 KiB
Python
149 lines
4.6 KiB
Python
"""Compass Command Line Interface"""
|
|
|
|
import sys
|
|
import os
|
|
from subprocess import Popen
|
|
|
|
from compass.actions.health_check import check
|
|
from compass.utils.util import pretty_print
|
|
|
|
ACTION_MAP = {"check": "apache celery dhcp dns hds misc os_installer "
|
|
"package_installer squid tftp".split(" "),
|
|
"refresh": "db sync".split(" "),
|
|
}
|
|
|
|
|
|
class BootCLI:
|
|
|
|
def __init__(self):
|
|
return
|
|
|
|
def run(self, args):
|
|
"""
|
|
cli takes the commands and calls respective modules
|
|
"""
|
|
action = self.get_action(args)
|
|
if action is None:
|
|
self.print_help()
|
|
else:
|
|
module = self.get_module(action, args)
|
|
if module == "invalid":
|
|
self.print_help(action)
|
|
else:
|
|
method = "self.run_" + action + "(module)"
|
|
eval(method)
|
|
|
|
def get_action(self, args):
|
|
"""
|
|
This method returns an action type.
|
|
For 'compass check dhcp' command, it will return 'check'.
|
|
"""
|
|
if len(args) == 1:
|
|
return None
|
|
elif args[1] in ACTION_MAP.keys():
|
|
return args[1]
|
|
return None
|
|
|
|
def get_module(self, action, args):
|
|
"""
|
|
This method returns a module.
|
|
For 'compass check dhcp' command, it will return 'dhcp'.
|
|
"""
|
|
if len(args) <= 2:
|
|
return None
|
|
elif args[2] in ACTION_MAP[action]:
|
|
return args[2]
|
|
return "invalid"
|
|
|
|
def run_check(self, module=None):
|
|
"""
|
|
param module default set to None.
|
|
This provides a flexible sanity check,
|
|
if parameter module is none. Compass checks all modules.
|
|
If module specified, Compass will only check such module.
|
|
|
|
"""
|
|
if module is None:
|
|
pretty_print("Starting: Compass Health Check",
|
|
"==============================")
|
|
c = check.BootCheck()
|
|
res = c.run()
|
|
self.output_check_result(res)
|
|
|
|
else:
|
|
pretty_print("Checking Module: %s" % module,
|
|
"============================")
|
|
c = check.BootCheck()
|
|
method = "c.check_" + module + "()"
|
|
res = eval(method)
|
|
print "\n".join(msg for msg in res[1])
|
|
|
|
def output_check_result(self, result):
|
|
if result == {}:
|
|
return
|
|
pretty_print("\n",
|
|
"===============================",
|
|
"* Compass Health Check Report *",
|
|
"===============================")
|
|
successful = True
|
|
num = 1
|
|
for key in result.keys():
|
|
if result[key][0] == 0:
|
|
successful = False
|
|
print "%s" % "\n".join(item for item in result[key][1])
|
|
|
|
print "===================="
|
|
if successful is True:
|
|
print "Compass Check completes. No problems found, all systems go"
|
|
sys.exit(0)
|
|
else:
|
|
print "Compass has ERRORS shown above. Please fix them before " \
|
|
"deploying!"
|
|
sys.exit(1)
|
|
|
|
def run_refresh(self, action=None):
|
|
## TODO: replace refresh.sh with refresh.py
|
|
if action is None:
|
|
pretty_print("Refreshing Compass...",
|
|
"=================")
|
|
Popen(['/opt/compass/bin/refresh.sh'], shell=True)
|
|
elif action == "db":
|
|
pretty_print("Refreshing Compass Database...",
|
|
"===================")
|
|
Popen(['/opt/compass/bin/manage_db.py createdb'], shell=True)
|
|
else:
|
|
pretty_print("Syncing with Installers...",
|
|
"================")
|
|
Popen(['/opt/compass/bin/manage_db.py sync_from_installers'],
|
|
shell=True)
|
|
|
|
def print_help(self, module_help=""):
|
|
if module_help == "":
|
|
pretty_print("usage\n=====",
|
|
"compass <refresh|check>",
|
|
"type 'compass {action} --help' for detailed "
|
|
"command list")
|
|
|
|
elif module_help == "refresh":
|
|
pretty_print("usage\n=====",
|
|
"compass refresh [%s]" %
|
|
"|".join(action for action in ACTION_MAP['refresh']))
|
|
|
|
else:
|
|
pretty_print("usage\n=====",
|
|
"compass check [%s]" %
|
|
"|".join(action for action in ACTION_MAP['check']))
|
|
sys.exit(2)
|
|
|
|
|
|
def main():
|
|
"""
|
|
Compass cli entry point
|
|
"""
|
|
cli = BootCLI()
|
|
output = cli.run(sys.argv)
|
|
return sys.exit(output)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|