]> arthur.barton.de Git - netdata.git/blob - python.d/hddtemp.chart.py
add option to connect to unix socket with `SocketService`
[netdata.git] / 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 = 5
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             return None
50         data = {}
51         for i in range(len(raw) // 5):
52             try:
53                 val = int(raw[i*5+3])
54             except ValueError:
55                 val = 0
56             data[raw[i*5+1].replace("/dev/", "")] = val
57         return data
58
59     def check(self):
60         """
61         Parse configuration, check if hddtemp is available, and dynamically create chart lines data
62         :return: boolean
63         """
64         self._parse_config()
65         data = self._get_data()
66         if data is None:
67             self.error("No data received")
68             return False
69
70         for name in data:
71             self.definitions[ORDER[0]]['lines'].append([name])
72
73         return True
74
75