From 0873a58dc0873b779244ff02a8300864ddc34646 Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Tue, 31 Mar 2020 21:03:07 +0200
Subject: [PATCH] [modules] Re-added hddtemp

---
 modules/contrib/hddtemp.py | 90 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 modules/contrib/hddtemp.py

diff --git a/modules/contrib/hddtemp.py b/modules/contrib/hddtemp.py
new file mode 100644
index 0000000..86800c1
--- /dev/null
+++ b/modules/contrib/hddtemp.py
@@ -0,0 +1,90 @@
+# -*- 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()