import json
import sys
+from utils.inventory_mgr import InventoryMgr
from utils.mongo_access import MongoAccess
from utils.util import ClassResolver
-DEFAULTS = {
- 'env': 'WebEX-Mirantis@Cisco',
- 'inventory': 'inventory',
- 'loglevel': 'WARNING'
-}
-
-
-
-def get_args():
- parser = argparse.ArgumentParser()
- parser.add_argument("-m", "--mongo_config", nargs="?", type=str,
- default="",
- help="name of config file with MongoDB server " +
- "access details")
- parser.add_argument("-e", "--env", nargs="?", type=str,
- default=DEFAULTS['env'],
- help="name of environment to scan \n" +
- "(default: {})".format(DEFAULTS['env']))
- parser.add_argument("-y", "--inventory", nargs="?", type=str,
- default=DEFAULTS['inventory'],
- help="name of inventory collection \n" +
- "(default: {}".format(DEFAULTS['inventory']))
- parser.add_argument('-i', '--inputfile', nargs='?', type=str,
- default='',
- help="read input from the specifed file \n" +
- "(default: from stdin)")
- parser.add_argument("-l", "--loglevel", nargs="?", type=str,
- default=DEFAULTS["loglevel"],
- help="logging level \n(default: '{}')"
- .format(DEFAULTS["loglevel"]))
- args = parser.parse_args()
- return args
-
-input = None
-args = get_args()
-MongoAccess.set_config_file(args.mongo_config)
-if args.inputfile:
- try:
- with open(args.inputfile, 'r') as input_file:
- input = input_file.read()
- except Exception as e:
- raise FileNotFoundError("failed to open input file: " + args.inputfile)
- exit(1)
-else:
- input = sys.stdin.read()
- if not input:
- raise ValueError("No input provided on stdin")
- exit(1)
-
-check_result_full = json.loads(input)
-check_client = check_result_full['client']
-check_result = check_result_full['check']
-check_result['id'] = check_result_full['id']
-name = check_result['name']
-status = check_result['status']
-object_type = name[:name.index('_')]
-object_id = name[name.index('_')+1:]
-if 'environment' in check_client:
- args.env = check_client['environment']
-
-handler = None
-basic_handling_types = ['vedge', 'vservice']
-if object_type in basic_handling_types:
- from monitoring.handlers.basic_check_handler import BasicCheckHandler
- handler = BasicCheckHandler(args)
-else:
- module_name = 'handle_' + object_type
- handler = ClassResolver.get_instance_single_arg(args,
- module_name=module_name,
- package_name='monitoring.handlers')
-if handler:
- handler.handle(object_id, check_result)
+
+class Monitor:
+ DEFAULTS = {
+ 'env': 'WebEX-Mirantis@Cisco',
+ 'inventory': 'inventory',
+ 'loglevel': 'WARNING'
+ }
+ object_types = []
+
+ def __init__(self):
+ self.args = self.get_args()
+ MongoAccess.set_config_file(self.args.mongo_config)
+ self.inv = InventoryMgr()
+ self.inv.set_collections(self.args.inventory)
+ self.input_text = None
+
+ def get_args(self):
+ parser = argparse.ArgumentParser()
+ parser.add_argument("-m", "--mongo_config", nargs="?", type=str,
+ default="",
+ help="name of config file with MongoDB server " +
+ "access details")
+ parser.add_argument("-e", "--env", nargs="?", type=str,
+ default=self.DEFAULTS['env'],
+ help="name of environment to scan \n" +
+ "(default: {})".format(self.DEFAULTS['env']))
+ parser.add_argument("-y", "--inventory", nargs="?", type=str,
+ default=self.DEFAULTS['inventory'],
+ help="name of inventory collection \n" +
+ "(default: {}".format(self.DEFAULTS['inventory']))
+ parser.add_argument('-i', '--inputfile', nargs='?', type=str,
+ default='',
+ help="read input from the specifed file \n" +
+ "(default: from stdin)")
+ parser.add_argument("-l", "--loglevel", nargs="?", type=str,
+ default=self.DEFAULTS["loglevel"],
+ help="logging level \n(default: '{}')"
+ .format(self.DEFAULTS["loglevel"]))
+ args = parser.parse_args()
+ return args
+
+ def get_object_types(self) -> list:
+ if not self.object_types:
+ docs = self.inv.find_items({'name': 'object_types'},
+ collection='constants')
+ for types_list in docs:
+ self.object_types = [t['value'] for t in types_list['data']]
+ if not self.object_types:
+ raise ValueError('Unable to fetch object types')
+ return self.object_types
+
+ def match_object_types(self, check_name: str) -> list:
+ object_types = self.get_object_types()
+ matches = [t for t in object_types if check_name.startswith(t + '_')]
+ return matches
+
+ def find_object_type_and_id(self, check_name: str):
+ # if we have multiple matching host types, then take the longest
+ # of these. For example, if matches are ['host', 'host_pnic'],
+ # then take 'host_pnic'.
+ # To facilitate this, we sort the matches by reverse order.
+ matching_object_types = sorted(self.match_object_types(check_name),
+ reverse=True)
+ if not matching_object_types:
+ raise ValueError('Unable to match check name "{}" with object type'
+ .format(check_name))
+ obj_type = matching_object_types[0]
+ obj_id = check_name[len(obj_type)+1:]
+ return obj_type, obj_id
+
+ def read_input(self):
+ if self.args.inputfile:
+ try:
+ with open(self.args.inputfile, 'r') as input_file:
+ self.input_text = input_file.read()
+ except Exception as e:
+ raise FileNotFoundError("failed to open input file {}: {}"
+ .format(self.args.inputfile, str(e)))
+ else:
+ self.input_text = sys.stdin.read()
+ if not self.input_text:
+ raise ValueError("No input provided on stdin")
+
+ def get_handler_by_type(self, obj_type):
+ module_name = 'handle_' + obj_type
+ package = 'monitoring.handlers'
+ handler = ClassResolver.get_instance_single_arg(self.args,
+ module_name=module_name,
+ package_name=package)
+ return handler
+
+ def get_handler(self, obj_type):
+ basic_handling_types = ['vedge', 'vservice']
+ if obj_type not in basic_handling_types:
+ return self.get_handler_by_type(obj_type)
+ from monitoring.handlers.basic_check_handler \
+ import BasicCheckHandler
+ return BasicCheckHandler(self.args)
+
+ def process_input(self):
+ check_result_full = json.loads(self.input_text)
+ check_client = check_result_full['client']
+ check_result = check_result_full['check']
+ check_result['id'] = check_result_full['id']
+ name = check_result['name']
+ object_type, object_id = monitor.find_object_type_and_id(name)
+ if 'environment' in check_client:
+ self.args.env = check_client['environment']
+
+ check_handler = self.get_handler(object_type)
+ if check_handler:
+ check_handler.handle(object_id, check_result)
+
+ def process_check_result(self):
+ self.read_input()
+ self.process_input()
+
+monitor = Monitor()
+monitor.process_check_result()
Code Review / calipso.git / commitdiff