checks/aws.include

   1 #!/usr/bin/python
   2 # -*- encoding: utf-8; py-indent-offset: 4 -*-
   3 # +------------------------------------------------------------------+
   4 # |             ____ _               _        __  __ _  __           |
   5 # |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   6 # |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   7 # |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   8 # |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   9 # |                                                                  |
  10 # | Copyright Mathias Kettner 2018             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
  27
  28 def parse_aws(info):
  29     import json
  30     loaded = []
  31     for row in info:
  32         try:
  33             loaded.extend(json.loads(" ".join(row)))
  34         except (TypeError, IndexError):
  35             pass
  36     return loaded
  37
  38
  39 def _extract_aws_metrics(expected_metric_names, parsed):
  40     values = {}
  41     for row in parsed:
  42         for expected_metric_name in expected_metric_names:
  43             if row['Id'].startswith(expected_metric_name.lower()):
  44                 try:
  45                     values.setdefault(expected_metric_name, row['Values'][0])
  46                 except IndexError:
  47                     pass
  48                 break
  49     return values
  50
  51
  52 def _extract_aws_metrics_by_labels(expected_metric_names, parsed):
  53     values_by_labels = {}
  54     for row in parsed:
  55         row_id = row['Id']
  56         row_label = row['Label']
  57         row_values = row['Values']
  58         for expected_metric_name in expected_metric_names:
  59             if not row_id.startswith(expected_metric_name.lower()):
  60                 continue
  61
  62             try:
  63                 value = row_values[0]
  64             except IndexError:
  65                 continue
  66             else:
  67                 values_by_labels.setdefault(row_label, {})\
  68                                 .setdefault(expected_metric_name, value)
  69     return values_by_labels