1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
|
# -*- coding: utf-8 -*-
# Description: hddtemp netdata python.d module
# Author: Pawel Krupa (paulfantom)
# Author: Ilya Mashchenko (ilyam8)
# SPDX-License-Identifier: GPL-3.0-or-later
import re
from copy import deepcopy
from bases.FrameworkServices.SocketService import SocketService
ORDER = [
'temperatures',
]
CHARTS = {
'temperatures': {
'options': ['disks_temp', 'Disks Temperatures', 'Celsius', 'temperatures', 'hddtemp.temperatures', 'line'],
'lines': [
# lines are created dynamically in `check()` method
]}}
RE = re.compile(r'\/dev\/([^|]+)\|([^|]+)\|([0-9]+|SLP|UNK)\|')
class Disk:
def __init__(self, id_, name, temp):
self.id = id_.split('/')[-1]
self.name = name.replace(' ', '_')
self.temp = temp if temp.isdigit() else None
def __repr__(self):
return self.id
class Service(SocketService):
def __init__(self, configuration=None, name=None):
SocketService.__init__(self, configuration=configuration, name=name)
self.order = ORDER
self.definitions = deepcopy(CHARTS)
self.do_only = self.configuration.get('devices')
self._keep_alive = False
self.request = ""
self.host = "127.0.0.1"
self.port = 7634
def get_disks(self):
r = self._get_raw_data()
if not r:
return None
m = RE.findall(r)
if not m:
self.error("received data doesn't have needed records")
return None
rv = [Disk(*d) for d in m]
self.debug('available disks: {0}'.format(rv))
if self.do_only:
return [v for v in rv if v.id in self.do_only]
return rv
def get_data(self):
"""
Get data from TCP/IP socket
:return: dict
"""
disks = self.get_disks()
if not disks:
return None
return dict((d.id, d.temp) for d in disks)
def check(self):
"""
Parse configuration, check if hddtemp is available, and dynamically create chart lines data
:return: boolean
"""
self._parse_config()
disks = self.get_disks()
if not disks:
return False
for d in disks:
dim = [d.id]
self.definitions['temperatures']['lines'].append(dim)
return True
@staticmethod
def _check_raw_data(data):
return not bool(data)
|