zabbix-elasticsearch

1
from urllib.parse import urlparse
2

3
import json
4
import argparse
5
import requests
6
import sys
7

8
def check_es_health(url, metric):
9
    """
10
    Try to get and return the specified metric
11
    """
12
    address = url.scheme + '://' + url.netloc + '/_cluster/health'
13
    result = requests.get(address)
14
    
15
    assert result.status_code == 200
16

17
    state = result.json()    
18
    
19
    if metric in state.keys():
20
        print(result.json()[metric])
21
        sys.exit(0)
22

23
    raise ValueError('The Key %s provided not exists on the response' %metric)
24

25

26
if __name__ == '__main__':
27
    parser = argparse.ArgumentParser()
28
    parser.add_argument('--url', type=urlparse, required=True)
29
    parser.add_argument('--metric', type=str, required=True)
30
    
31
    args = parser.parse_args()
32
    
33
    check_es_health(url=args.url, metric=args.metric)
34