cold

AoM_Service/AoM_Configs/library/args.py

@@ -0,0 +1,84 @@
+# Contians the arg parser options.
+
+
+import argparse
+import sys
+
+
+def get_builder_args():
+    """
+    Gets the arguments passed in to the aom_builder main call
+
+    :return: parser object
+    """
+    parser = argparse.ArgumentParser(description="Generates a valid yaml file for alerting on metrics. "
+                                                 "If you are familiar with the yaml structure for an alert"
+                                                 "you don't have to use this builder, it's just convenient")
+    parser.add_argument('-q', '--query', help="The Kariosdb query string to use")
+    parser.add_argument('-i', '--interval', type=int, default=60, help="The interval that the check will run. "
+                                                                       "This value is in seconds")
+    parser.add_argument('-t', '--threshold', '--upperthreshold', help="The upper threshold is the value that when reached will cause an alert "
+                                                                      "depending on the threshold logic. "
+                                                                      "Use in conjunction with lower threshold to define a normal band.")
+    parser.add_argument('-b', '--lowerthreshold', help="The lower threshold is the value that when reached will cause an alert "
+                                                                      "depending on the threshold logic"
+                                                                      "Use in conjunction with upper threshold to define a normal band.")
+    parser.add_argument('-m', '--measure', choices=['gt', 'lt', 'eq'], help="The measure to use to compare the "
+                                                                            "threshold to the values of the alerts")
+    parser.add_argument('-a', '--alert_config', help='A valid Yaml representation of your alerting block')
+    parser.add_argument('-l', '--log_level', type=int, default=0, help="The log level for the aom_builder run. "
+                                                                       "[0=Error, 1=Info, 2=Debug]")
+    parser.add_argument('-p', '--port', type=int, default=8080, help="The port to run the webapp on")
+
+    return args_to_dict(parser)
+
+def get_tester_service_args():
+    """
+    Gets arguments passed into aom_tester.py
+    Returns: parser object
+    """
+    parser = argparse.ArgumentParser(description="Parameters to start the alerting on metrics dummy tester service")
+    parser.add_argument('-l', '--log_level', type=int, default=0, help="The log level for the aom_service app"
+                                                                       "[0=Error, 1=Info, 2=Debug]")
+    parser.add_argument('-a', '--alert_configs', default=None,
+                        help="If provided will override the folder location read from the config with the value passed "
+                             "in. Is helpful for testing and troubleshooting alerts")
+    parser.add_argument('--hostname', help="If provided, will override the actual hostname check with this value")
+    parser.add_argument('-p', '--port', type=int, default=8080, help="The port to run the webapp on")
+    return args_to_dict(parser)
+
+def get_service_args():
+    """
+    Gets arguments passed into aom_service.py
+    Returns: parser object
+    """
+    parser = argparse.ArgumentParser(description="Parameters to start the alerting on metrics service")
+    parser.add_argument('-l', '--log_level', type=int, default=0, help="The log level for the aom_service app"
+                                                                       "[0=Error, 1=Info, 2=Debug]")
+    parser.add_argument('-a', '--alert_configs', default=None,
+                        help="If provided will override the folder location read from the config with the value passed "
+                             "in. Is helpful for testing and troubleshooting alerts")
+    parser.add_argument('-o', '--override', action='store_true', help="Overrides the check leader election value")
+    parser.add_argument('--hostname', help="If provided, will override the actual hostname check with this value")
+    parser.add_argument('-p', '--port', type=int, default=8080, help="The port to run the webapp on")
+    return args_to_dict(parser)
+
+
+def args_to_dict(parsed_args):
+    """
+    Converts the argument parser object to a dict
+    Args:
+        parsed_args: Arg parser object
+    Returns:
+        Dictionary of arguments
+    """
+    try:
+        arg_list = parsed_args.parse_args()
+        # RETURN A DICT OF ARGUMENTS
+        arg_dict = dict()
+        for val in vars(arg_list):
+            arg_dict[val] = getattr(arg_list, val)
+        return arg_dict
+    except argparse.ArgumentError:
+        parsed_args.print_help()
+        sys.exit(1)