91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""Fetch hard drive temeperature data from a hddtemp daemon
|
||
|
|
that runs on localhost and default port (7634)
|
||
|
|
"""
|
||
|
|
|
||
|
|
import socket
|
||
|
|
|
||
|
|
import bumblebee.engine
|
||
|
|
import bumblebee.output
|
||
|
|
|
||
|
|
HOST = "localhost"
|
||
|
|
PORT = 7634
|
||
|
|
|
||
|
|
CHUNK_SIZE = 1024
|
||
|
|
RECORD_SIZE = 5
|
||
|
|
SEPARATOR = "|"
|
||
|
|
|
||
|
|
|
||
|
|
class Module(bumblebee.engine.Module):
|
||
|
|
def __init__(self, engine, config):
|
||
|
|
widget = bumblebee.output.Widget(full_text=self.hddtemps)
|
||
|
|
super(Module, self).__init__(engine, config, widget)
|
||
|
|
self._hddtemps = self._get_hddtemps()
|
||
|
|
|
||
|
|
def hddtemps(self, __):
|
||
|
|
return self._hddtemps
|
||
|
|
|
||
|
|
def _fetch_data(self):
|
||
|
|
"""fetch data from hddtemp service"""
|
||
|
|
try:
|
||
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
|
||
|
|
sock.connect((HOST, PORT))
|
||
|
|
data = ""
|
||
|
|
while True:
|
||
|
|
chunk = sock.recv(CHUNK_SIZE)
|
||
|
|
if chunk:
|
||
|
|
data += str(chunk)
|
||
|
|
else:
|
||
|
|
break
|
||
|
|
return data
|
||
|
|
except (AttributeError, socket.error) as e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _get_parts(data):
|
||
|
|
"""
|
||
|
|
split data using | separator and remove first item
|
||
|
|
(because the first item is empty)
|
||
|
|
"""
|
||
|
|
parts = data.split("|")[1:]
|
||
|
|
return parts
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _partition_parts(parts):
|
||
|
|
"""
|
||
|
|
partition parts: one device record is five (5) items
|
||
|
|
"""
|
||
|
|
per_disk = [parts[i:i+RECORD_SIZE]
|
||
|
|
for i in range(len(parts))[::RECORD_SIZE]]
|
||
|
|
return per_disk
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _get_name_and_temp(device_record):
|
||
|
|
"""
|
||
|
|
get device name (without /dev part, to save space on bar)
|
||
|
|
and temperature (in °C) as tuple
|
||
|
|
"""
|
||
|
|
device_name = device_record[0].split("/")[-1]
|
||
|
|
device_temp = device_record[2]
|
||
|
|
return (device_name, device_temp)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _get_hddtemp(device_record):
|
||
|
|
name, temp = device_record
|
||
|
|
hddtemp = "{}+{}°C".format(name, temp)
|
||
|
|
return hddtemp
|
||
|
|
|
||
|
|
def _get_hddtemps(self):
|
||
|
|
data = self._fetch_data()
|
||
|
|
if data is None:
|
||
|
|
return "n/a"
|
||
|
|
parts = self._get_parts(data)
|
||
|
|
per_disk = self._partition_parts(parts)
|
||
|
|
names_and_temps = [self._get_name_and_temp(x) for x in per_disk]
|
||
|
|
hddtemps = [self._get_hddtemp(x) for x in names_and_temps]
|
||
|
|
return SEPARATOR.join(hddtemps)
|
||
|
|
|
||
|
|
def update(self, __):
|
||
|
|
self._hddtemps = self._get_hddtemps()
|