check_apache_drill_encryption_enabled.py

#!/usr/bin/env python3
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2018-05-04 18:34:52 +0100 (Fri, 04 May 2018)
#
#  https://github.com/HariSekhon/Nagios-Plugins
#
#  License: see accompanying Hari Sekhon LICENSE file
#
#  If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback # pylint: disable=line-too-long
#
#  https://www.linkedin.com/in/HariSekhon
#

"""

Nagios Plugin to check an Apache Drill node has encryption enabled via its Rest API

Encryption is only available from Drill 1.11+

Tested on Apache Drill 1.11, 1.12, 1.13, 1.14, 1.15

"""


# Doesn't work on 0.7, the API endpoint isn't found

from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

import os
import sys
import traceback
libdir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'pylib'))
sys.path.append(libdir)
try:
    # pylint: disable=wrong-import-position
    from harisekhon import RestNagiosPlugin
except ImportError:
    print(traceback.format_exc(), end='')
    sys.exit(4)

__author__ = 'Hari Sekhon'
__version__ = '0.1'


# pylint: disable=too-few-public-methods
class CheckApacheDrillEncryptionEnabled(RestNagiosPlugin):

    def __init__(self):
        # Python 2.x
        super(CheckApacheDrillEncryptionEnabled, self).__init__()
        # Python 3.x
        # super().__init__()
        self.name = 'Apache Drill'
        self.path = '/cluster.json'
        self.default_port = 8047
        self.json = True
        self.auth = False
        self.msg = 'Apache Drill message not defined'

    def parse_json(self, json_data):
        user_encryption_enabled = json_data['userEncryptionEnabled']
        bit_encryption_enabled = json_data['bitEncryptionEnabled']
        self.msg = 'Apache Drill user encryption enabled = {}'.format(user_encryption_enabled)
        if not user_encryption_enabled:
            self.critical()
        self.msg += ', drillbit encryption_enabled = {}'.format(bit_encryption_enabled)
        if not bit_encryption_enabled:
            self.critical()


if __name__ == '__main__':
    CheckApacheDrillEncryptionEnabled().main()