# -*- coding: utf-8 -*-
# Description: hddtemp netdata python.d module
# Author: Pawel Krupa (paulfantom)
+# Modified by l2isbad
+import os
from base import SocketService
# default module values (can be overridden per job in `config`)
ORDER = ['temperatures']
-CHARTS = {
- 'temperatures': {
- 'options': ['disks_temp', 'temperature', 'Celsius', 'Disks temperature', 'hddtemp.temp', 'line'],
- 'lines': [
- # lines are created dynamically in `check()` method
- ]}
-}
-
-
class Service(SocketService):
def __init__(self, configuration=None, name=None):
SocketService.__init__(self, configuration=configuration, name=name)
self.host = "127.0.0.1"
self.port = 7634
self.order = ORDER
- self.definitions = CHARTS
+ self.fahrenheit = ('Fahrenheit', lambda x: x * 9 / 5 + 32) if self.configuration.get('fahrenheit') else False
+ self.whatever = ('Whatever', lambda x: x * 33 / 22 + 11) if self.configuration.get('whatever') else False
+ self.choice = (choice for choice in [self.fahrenheit, self.whatever] if choice)
+ self.calc = lambda x: x
+ self.disks = []
+
+ def _get_disks(self):
+ try:
+ disks = self.configuration['devices']
+ self.info("Using configured disks" + str(disks))
+ except (KeyError, TypeError) as e:
+ self.info("Autodetecting disks")
+ return ["/dev/" + f for f in os.listdir("/dev") if len(f) == 3 and f.startswith("sd")]
+
+ ret = []
+ for disk in disks:
+ if not disk.startswith('/dev/'):
+ disk = "/dev/" + disk
+ ret.append(disk)
+ if len(ret) == 0:
+ self.error("Provided disks cannot be found in /dev directory.")
+ return ret
+
+ def _check_raw_data(self, data):
+ if not data.endswith('|'):
+ return False
+
+ if all(disk in data for disk in self.disks):
+ return True
+
+ return False
def _get_data(self):
"""
return None
data = {}
for i in range(len(raw) // 5):
+ if not raw[i*5+1] in self.disks:
+ continue
try:
- val = int(raw[i*5+3])
+ val = self.calc(int(raw[i*5+3]))
except ValueError:
val = 0
data[raw[i*5+1].replace("/dev/", "")] = val
:return: boolean
"""
self._parse_config()
+ self.disks = self._get_disks()
+
data = self._get_data()
if data is None:
return False
+ self.definitions = {
+ 'temperatures': {
+ 'options': ['disks_temp', 'Disks Temperatures', 'temperatures', 'hddtemp.temperatures', 'line'],
+ 'lines': [
+ # lines are created dynamically in `check()` method
+ ]}
+ }
+ try:
+ self.choice = next(self.choice)
+ except StopIteration:
+ self.definitions[ORDER[0]]['options'].insert(2, 'Celsius')
+ else:
+ self.calc = self.choice[1]
+ self.definitions[ORDER[0]]['options'].insert(2, self.choice[0])
+
for name in data:
self.definitions[ORDER[0]]['lines'].append([name])
-
return True
-
-