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._keep_alive = False
  36         self.request = ""
  37         self.host = "127.0.0.1"
  38         self.port = 7634
  39         self.order = ORDER
  40         self.definitions = CHARTS
  41
  42     def _get_data(self):
  43         """
  44         Get data from TCP/IP socket
  45         :return: dict
  46         """
  47         try:
  48             raw = self._get_raw_data().split("|")[:-1]
  49         except AttributeError:
  50             self.error("no data received")
  51             return None
  52         data = {}
  53         for i in range(len(raw) // 5):
  54             try:
  55                 val = int(raw[i*5+3])
  56             except ValueError:
  57                 val = 0
  58             data[raw[i*5+1].replace("/dev/", "")] = val
  59
  60         if len(data) == 0:
  61             self.error("received data doesn't have needed records")
  62             return None
  63         else:
  64             return data
  65
  66     def check(self):
  67         """
  68         Parse configuration, check if hddtemp is available, and dynamically create chart lines data
  69         :return: boolean
  70         """
  71         self._parse_config()
  72         data = self._get_data()
  73         if data is None:
  74             return False
  75
  76         for name in data:
  77             self.definitions[ORDER[0]]['lines'].append([name])
  78
  79         return True
  80
  81