Source code for temelio_monitoring.cli_parser.client.blocked_client_parser

"""
This module manage blocked client arguments
"""

from argparse import ArgumentParser


class BlockedClientParser(ArgumentParser):
    """
    Manage blocked client thresholds argument parser
    """

    def __init__(self, **kwargs):

        super().__init__(add_help=False)

        # Argument group
        group = self.add_argument_group('Thresholds')

        # Argument settings
        group.add_argument(
            '--warn-blocked-client',
            action=kwargs.get('action', 'store'),
            type=kwargs.get('type', int),
            default=kwargs.get('default', 25),
            help=kwargs.get('help', 'Blocked client warning threshold'),
            required=kwargs.get('required', False))

        # Argument settings
        group.add_argument(
            '--crit-blocked-client',
            action=kwargs.get('action', 'store'),
            type=kwargs.get('type', int),
            default=kwargs.get('default', 50),
            help=kwargs.get('help', 'Blocked client critical threshold'),
            required=kwargs.get('required', False))