Roxedus/ha-nicehash
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(sensor): rename sensors

Browse Source
This commit is contained in:
Brian Berg 2020-07-09 00:31:02 +00:00
parent a31d524e10
commit 743fd0febf
4 changed files with 36 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_components/nicehash/account_sensors.py
View File

@ -23,7 +23,7 @@ from .const import (
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
class NiceHashBalanceSensor(Entity): class BalanceSensor(Entity):
""" """
Displays [available|pending|total] balance of an account for a currency Displays [available|pending|total] balance of an account for a currency
""" """

14
custom_components/nicehash/device_sensors.py
View File

@ -23,7 +23,7 @@ from .const import (
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
class NiceHashDeviceSensor(Entity): class DeviceSensor(Entity):
""" """
Mining rig device sensor Mining rig device sensor
""" """
@ -129,7 +129,7 @@ class NiceHashDeviceSensor(Entity):
self._algorithm = None self._algorithm = None
class NiceHashDeviceStatusSensor(NiceHashDeviceSensor): class DeviceStatusSensor(DeviceSensor):
""" """
Displays status of a mining rig device Displays status of a mining rig device
""" """
@ -155,7 +155,7 @@ class NiceHashDeviceStatusSensor(NiceHashDeviceSensor):
return ICON_PULSE return ICON_PULSE
class NiceHashDeviceSpeedSensor(NiceHashDeviceSensor): class DeviceSpeedSensor(DeviceSensor):
""" """
Displays speed of a mining rig device Displays speed of a mining rig device
""" """
@ -186,7 +186,7 @@ class NiceHashDeviceSpeedSensor(NiceHashDeviceSensor):
return f"{self._speed_unit}/s" return f"{self._speed_unit}/s"
class NiceHashDeviceAlgorithmSensor(NiceHashDeviceSensor): class DeviceAlgorithmSensor(DeviceSensor):
""" """
Displays algorithm of a mining rig device Displays algorithm of a mining rig device
""" """
@ -212,7 +212,7 @@ class NiceHashDeviceAlgorithmSensor(NiceHashDeviceSensor):
return ICON_PICKAXE return ICON_PICKAXE
class NiceHashDeviceTemperatureSensor(NiceHashDeviceSensor): class DeviceTemperatureSensor(DeviceSensor):
""" """
Displays temperature of a mining rig device Displays temperature of a mining rig device
""" """
@ -244,7 +244,7 @@ class NiceHashDeviceTemperatureSensor(NiceHashDeviceSensor):
return "C" return "C"
class NiceHashDeviceLoadSensor(NiceHashDeviceSensor): class DeviceLoadSensor(DeviceSensor):
""" """
Displays load of a mining rig device Displays load of a mining rig device
""" """
@ -275,7 +275,7 @@ class NiceHashDeviceLoadSensor(NiceHashDeviceSensor):
return "%" return "%"
class NiceHashDeviceRPMSensor(NiceHashDeviceSensor): class DeviceRPMSensor(DeviceSensor):
""" """
Displays RPM of a mining rig device Displays RPM of a mining rig device
""" """

6
custom_components/nicehash/rig_sensors.py
View File

@ -20,7 +20,7 @@ from .const import (
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
class NiceHashRigTemperatureSensor(Entity): class RigTemperatureSensor(Entity):
""" """
Displays highest temperature of active mining rig devices Displays highest temperature of active mining rig devices
""" """
@ -112,7 +112,7 @@ class NiceHashRigTemperatureSensor(Entity):
await self.coordinator.async_request_refresh() await self.coordinator.async_request_refresh()
class NiceHashRigStatusSensor(Entity): class RigStatusSensor(Entity):
""" """
Displays status of a mining rig Displays status of a mining rig
""" """
@ -199,7 +199,7 @@ class NiceHashRigStatusSensor(Entity):
await self.coordinator.async_request_refresh() await self.coordinator.async_request_refresh()
class NiceHashRigProfitabilitySensor(Entity): class RigProfitabilitySensor(Entity):
""" """
Displays profitability of a mining rig Displays profitability of a mining rig
""" """

54
custom_components/nicehash/sensor.py
View File

@ -22,19 +22,19 @@ from .const import (
DEVICE_SPEED_ALGORITHM, DEVICE_SPEED_ALGORITHM,
) )
from .nicehash import NiceHashPrivateClient, NiceHashPublicClient from .nicehash import NiceHashPrivateClient, NiceHashPublicClient
from .account_sensors import NiceHashBalanceSensor from .account_sensors import BalanceSensor
from .rig_sensors import ( from .rig_sensors import (
NiceHashRigStatusSensor, RigStatusSensor,
NiceHashRigTemperatureSensor, RigTemperatureSensor,
NiceHashRigProfitabilitySensor, RigProfitabilitySensor,
) )
from .device_sensors import ( from .device_sensors import (
NiceHashDeviceAlgorithmSensor, DeviceAlgorithmSensor,
NiceHashDeviceSpeedSensor, DeviceSpeedSensor,
NiceHashDeviceStatusSensor, DeviceStatusSensor,
NiceHashDeviceLoadSensor, DeviceLoadSensor,
NiceHashDeviceRPMSensor, DeviceRPMSensor,
NiceHashDeviceTemperatureSensor, DeviceTemperatureSensor,
) )
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -80,19 +80,19 @@ async def async_setup_platform(
def create_balance_sensors(organization_id, currency, coordinator): def create_balance_sensors(organization_id, currency, coordinator):
balance_sensors = [ balance_sensors = [
NiceHashBalanceSensor( BalanceSensor(
coordinator, coordinator,
organization_id, organization_id,
currency=CURRENCY_BTC, currency=CURRENCY_BTC,
balance_type=BALANCE_TYPE_AVAILABLE, balance_type=BALANCE_TYPE_AVAILABLE,
), ),
NiceHashBalanceSensor( BalanceSensor(
coordinator, coordinator,
organization_id, organization_id,
currency=CURRENCY_BTC, currency=CURRENCY_BTC,
balance_type=BALANCE_TYPE_PENDING, balance_type=BALANCE_TYPE_PENDING,
), ),
NiceHashBalanceSensor( BalanceSensor(
coordinator, coordinator,
organization_id, organization_id,
currency=CURRENCY_BTC, currency=CURRENCY_BTC,
@ -101,7 +101,7 @@ def create_balance_sensors(organization_id, currency, coordinator):
] ]
if currency == CURRENCY_USD or currency == CURRENCY_EUR: if currency == CURRENCY_USD or currency == CURRENCY_EUR:
balance_sensors.append( balance_sensors.append(
NiceHashBalanceSensor( BalanceSensor(
coordinator, coordinator,
organization_id, organization_id,
currency=currency, currency=currency,
@ -109,7 +109,7 @@ def create_balance_sensors(organization_id, currency, coordinator):
) )
) )
balance_sensors.append( balance_sensors.append(
NiceHashBalanceSensor( BalanceSensor(
coordinator, coordinator,
organization_id, organization_id,
currency=currency, currency=currency,
@ -117,7 +117,7 @@ def create_balance_sensors(organization_id, currency, coordinator):
) )
) )
balance_sensors.append( balance_sensors.append(
NiceHashBalanceSensor( BalanceSensor(
coordinator, coordinator,
organization_id, organization_id,
currency=currency, currency=currency,
@ -133,9 +133,9 @@ def create_balance_sensors(organization_id, currency, coordinator):
def create_rig_sensors(mining_rigs, coordinator): def create_rig_sensors(mining_rigs, coordinator):
rig_sensors = [] rig_sensors = []
for rig in mining_rigs: for rig in mining_rigs:
rig_sensors.append(NiceHashRigStatusSensor(coordinator, rig)) rig_sensors.append(RigStatusSensor(coordinator, rig))
rig_sensors.append(NiceHashRigTemperatureSensor(coordinator, rig)) rig_sensors.append(RigTemperatureSensor(coordinator, rig))
rig_sensors.append(NiceHashRigProfitabilitySensor(coordinator, rig)) rig_sensors.append(RigProfitabilitySensor(coordinator, rig))
return rig_sensors return rig_sensors
@ -146,15 +146,11 @@ def create_device_sensors(mining_rigs, coordinator):
devices = rig.get("devices") devices = rig.get("devices")
for i in range(len(devices)): for i in range(len(devices)):
device = devices[i] device = devices[i]
device_sensors.append( device_sensors.append(DeviceAlgorithmSensor(coordinator, rig, device))
NiceHashDeviceAlgorithmSensor(coordinator, rig, device) device_sensors.append(DeviceSpeedSensor(coordinator, rig, device))
) device_sensors.append(DeviceStatusSensor(coordinator, rig, device))
device_sensors.append(NiceHashDeviceSpeedSensor(coordinator, rig, device)) device_sensors.append(DeviceTemperatureSensor(coordinator, rig, device))
device_sensors.append(NiceHashDeviceStatusSensor(coordinator, rig, device)) device_sensors.append(DeviceLoadSensor(coordinator, rig, device))
device_sensors.append( device_sensors.append(DeviceRPMSensor(coordinator, rig, device))
NiceHashDeviceTemperatureSensor(coordinator, rig, device)
)
device_sensors.append(NiceHashDeviceLoadSensor(coordinator, rig, device))
device_sensors.append(NiceHashDeviceRPMSensor(coordinator, rig, device))
return device_sensors return device_sensors