python.d/hddtemp.chart.py

   1 # -*- coding: utf-8 -*-
   2 # Description: hddtemp netdata python.d module
   3 # Author: Pawel Krupa (paulfantom)
   4
   5 from base import SocketService
   6
   7 # default module values (can be overridden per job in `config`)
   8 #update_every = 2
   9 priority = 60000
  10 retries = 60
  11
  12 # default job configuration (overridden by python.d.plugin)
  13 # config = {'local': {
  14 #             'update_every': update_every,
  15 #             'retries': retries,
  16 #             'priority': priority,
  17 #             'host': 'localhost',
  18 #             'port': 7634
  19 #          }}
  20
  21 ORDER = ['temperatures']
  22
  23 CHARTS = {
  24     'temperatures': {
  25         'options': ['disks_temp', 'temperature', 'Celsius', 'Disks temperature', 'hddtemp.temp', 'line'],
  26         'lines': [
  27             # lines are created dynamically in `check()` method
  28         ]}
  29 }
  30
  31
  32 class Service(SocketService):
  33     def __init__(self, configuration=None, name=None):
  34         SocketService.__init__(self, configuration=configuration, name=name)
  35         self.request = ""
  36         self.host = "127.0.0.1"
  37         self.port = 7634
  38         self.order = ORDER
  39         self.definitions = CHARTS
  40
  41     def _get_data(self):
  42         """
  43         Get data from TCP/IP socket
  44         :return: dict
  45         """
  46         try:
  47             raw = self._get_raw_data().split("|")[:-1]
  48         except AttributeError:
  49             self.error("no data received")
  50             return None
  51         data = {}
  52         for i in range(len(raw) // 5):
  53             try:
  54                 val = int(raw[i*5+3])
  55             except ValueError:
  56                 val = 0
  57             data[raw[i*5+1].replace("/dev/", "")] = val
  58
  59         if len(data) == 0:
  60             self.error("received data doesn't have needed records")
  61             return None
  62         else:
  63             return data
  64
  65     def check(self):
  66         """
  67         Parse configuration, check if hddtemp is available, and dynamically create chart lines data
  68         :return: boolean
  69         """
  70         self._parse_config()
  71         data = self._get_data()
  72         if data is None:
  73             return False
  74
  75         for name in data:
  76             self.definitions[ORDER[0]]['lines'].append([name])
  77
  78         return True
  79
  80