From cc9410510bd812c668ab464e86ace2c21d176dc7 Mon Sep 17 00:00:00 2001 From: Brian Berg Date: Sun, 12 Jul 2020 21:03:16 +0000 Subject: [PATCH] refactor(sensors): refactor rig temperature - remove rig temperatures attribute --- custom_components/nicehash/nicehash.py | 4 +--- custom_components/nicehash/rig_sensors.py | 4 ++-- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/custom_components/nicehash/nicehash.py b/custom_components/nicehash/nicehash.py index 9e170dc..6e813d0 100644 --- a/custom_components/nicehash/nicehash.py +++ b/custom_components/nicehash/nicehash.py @@ -46,16 +46,14 @@ class MiningRig: self.name = data.get("name") self.status = data.get("minerStatus") self.status_time = data.get("statusTime") - self.devices = dict() - self.temperatures = [] self.profitability = data.get("profitability") self.unpaid_amount = data.get("unpaidAmount") devices = data.get("devices") self.num_devices = len(devices) + self.devices = dict() for device_data in devices: device = MiningRigDevice(device_data) self.devices[f"{device.id}"] = device - self.temperatures.append(device.temperature) class Payout: diff --git a/custom_components/nicehash/rig_sensors.py b/custom_components/nicehash/rig_sensors.py index 984abfb..454e89c 100644 --- a/custom_components/nicehash/rig_sensors.py +++ b/custom_components/nicehash/rig_sensors.py @@ -97,9 +97,9 @@ class RigTemperatureSensor(RigSensor): self._highest_temp = 0 rig = self._get_rig() if rig: - self._temps = rig.temperatures self._num_devices = rig.num_devices - self._highest_temp = max(rig.temperatures) + self._temps = [device.temperature for device in rig.devices.values()] + self._highest_temp = max(self._temps) return self._highest_temp