cmk/special_agents/utils.py

   1 #!/usr/bin/env python
   2 # -*- encoding: utf-8; py-indent-offset: 4 -*-
   3 # +------------------------------------------------------------------+
   4 # |             ____ _               _        __  __ _  __           |
   5 # |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   6 # |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   7 # |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   8 # |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   9 # |                                                                  |
  10 # | Copyright Mathias Kettner 2016             mk@mathias-kettner.de |
  11 # +------------------------------------------------------------------+
  12 #
  13 # This file is part of Check_MK.
  14 # The official homepage is at http://mathias-kettner.de/check_mk.
  15 #
  16 # check_mk is free software;  you can redistribute it and/or modify it
  17 # under the  terms of the  GNU General Public License  as published by
  18 # the Free Software Foundation in version 2.  check_mk is  distributed
  19 # in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
  20 # out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
  21 # PARTICULAR PURPOSE. See the  GNU General Public License for more de-
  22 # tails. You should have  received  a copy of the  GNU  General Public
  23 # License along with GNU Make; see the file  COPYING.  If  not,  write
  24 # to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
  25 # Boston, MA 02110-1301 USA.
  26 """Place for common code shared among different Check_MK special agents"""
  27
  28 import getopt
  29 import json
  30 import pprint
  31 import sys
  32
  33 import requests
  34
  35
  36 class AgentJSON(object):
  37     def __init__(self, key, title):
  38         self._key = key
  39         self._title = title
  40
  41     def usage(self):
  42         sys.stderr.write("""
  43 Check_MK %s Agent
  44
  45 USAGE: agent_%s --section_url [{section_name},{url}]
  46
  47     Parameters:
  48         --section_url   Pair of section_name and url
  49                         Can be defined multiple times
  50         --debug         Output json data with pprint
  51
  52 """ % (self._title, self._key))
  53
  54     def get_content(self):
  55         short_options = "h"
  56         long_options = ["section_url=", "help", "newline_replacement=", "debug"]
  57
  58         try:
  59             opts, _args = getopt.getopt(sys.argv[1:], short_options, long_options)
  60         except getopt.GetoptError as err:
  61             sys.stderr.write("%s\n" % err)
  62             sys.exit(1)
  63
  64         sections = []
  65         newline_replacement = "\\n"
  66         opt_debug = False
  67
  68         for o, a in opts:
  69             if o in ["--section_url"]:
  70                 sections.append(a.split(",", 1))
  71             elif o in ["--newline_replacement"]:
  72                 newline_replacement = a
  73             elif o in ["--debug"]:
  74                 opt_debug = True
  75             elif o in ['-h', '--help']:
  76                 self.usage()
  77                 sys.exit(0)
  78
  79         if not sections:
  80             self.usage()
  81             sys.exit(0)
  82
  83         content = {}
  84         for section_name, url in sections:
  85             content.setdefault(section_name, [])
  86             content[section_name].append(requests.get(url).text.replace("\n", newline_replacement))
  87
  88         if opt_debug:
  89             for line in content:
  90                 try:
  91                     pprint.pprint(json.loads(line))
  92                 except:
  93                     print line
  94         else:
  95             return content