netdata/python.d/hddtemp.chart.py at master · unixapple/netdata · GitHub

History

120 lines (104 loc) · 3.76 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

# -*- 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`)

#update_every = 2

priority = 60000

retries = 60

# default job configuration (overridden by python.d.plugin)

# config = {'local': {

# 'update_every': update_every,

# 'retries': retries,

# 'priority': priority,

# 'host': 'localhost',

# 'port': 7634

# }}

ORDER = ['temperatures']

class Service(SocketService):

def __init__(self, configuration=None, name=None):

SocketService.__init__(self, configuration=configuration, name=name)

self._keep_alive = False

self.request = ""

self.host = "127.0.0.1"

self.port = 7634

self.order = ORDER

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):

"""

Get data from TCP/IP socket

:return: dict

"""

try:

raw = self._get_raw_data().split("|")[:-1]

except AttributeError:

self.error("no data received")

return None

data = {}

for i in range(len(raw) // 5):

if not raw[i*5+1] in self.disks:

continue

try:

val = self.calc(int(raw[i*5+3]))

except ValueError:

val = 0

data[raw[i*5+1].replace("/dev/", "")] = val

if len(data) == 0:

self.error("received data doesn't have needed records")

return None

else:

return data

def check(self):

"""

Parse configuration, check if hddtemp is available, and dynamically create chart lines data

: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

netdata/python.d/hddtemp.chart.py at master · unixapple/netdata · GitHub

Expand file tree

Breadcrumbs

hddtemp.chart.py

Latest commit

History

Breadcrumbs

hddtemp.chart.py

File metadata and controls