check_couchdb_database_data_size.py

#!/usr/bin/env python
#  coding=utf-8
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2017-10-29 16:16:55 +0100 (Sun, 29 Oct 2017)
#
#  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 to help steer this or other code I publish
#
#  https://www.linkedin.com/in/harisekhon
#

"""

Nagios Plugin to check a CouchDB database's data size via its API

- Optional thresholds are applied to size in bytes
- Outputs perfdata for graphing

Tested on CouchDB 1.6.1 and 2.1.0

"""

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

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

__author__ = 'Hari Sekhon'
__version__ = '0.1'


class CheckCouchDBDatabaseDataSize(CheckCouchDBDatabaseStats):

    def __init__(self):
        # Python 2.x
        super(CheckCouchDBDatabaseDataSize, self).__init__()
        # Python 3.x
        # super().__init__()
        self.has_thresholds = True

    def check_couchdb_stats(self, json_data):
        data_size = json_data['data_size']
        self.msg += "data_size = {0}".format(humanize.naturalsize(data_size))
        self.check_thresholds(data_size)
        self.msg += ' | data_size={0}b{1}'.format(data_size, self.get_perf_thresholds())


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