82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
|
"""
|
||
|
|
Sensor platform for NiceHash
|
||
|
|
"""
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from homeassistant.config_entries import ConfigEntry
|
||
|
|
from homeassistant.const import ATTR_ATTRIBUTION
|
||
|
|
from homeassistant.core import Config, HomeAssistant
|
||
|
|
|
||
|
|
from .const import (
|
||
|
|
BALANCE_TYPE_AVAILABLE,
|
||
|
|
BALANCE_TYPE_PENDING,
|
||
|
|
BALANCE_TYPE_TOTAL,
|
||
|
|
CURRENCY_BTC,
|
||
|
|
CURRENCY_EUR,
|
||
|
|
CURRENCY_USD,
|
||
|
|
DOMAIN,
|
||
|
|
DEVICE_LOAD,
|
||
|
|
DEVICE_RPM,
|
||
|
|
DEVICE_SPEED_RATE,
|
||
|
|
DEVICE_SPEED_ALGORITHM,
|
||
|
|
)
|
||
|
|
from .nicehash import (
|
||
|
|
MiningRig,
|
||
|
|
MiningRigDevice,
|
||
|
|
NiceHashPrivateClient,
|
||
|
|
NiceHashPublicClient,
|
||
|
|
)
|
||
|
|
from .control_switches import (
|
||
|
|
GPUSwitch
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_setup_platform(
|
||
|
|
hass: HomeAssistant, config: Config, async_add_entities, discovery_info=None
|
||
|
|
):
|
||
|
|
"""Setup NiceHash sensor platform"""
|
||
|
|
_LOGGER.debug("Creating new NiceHash switch components")
|
||
|
|
|
||
|
|
data = hass.data[DOMAIN]
|
||
|
|
# Configuration
|
||
|
|
organization_id = data.get("organization_id")
|
||
|
|
client = data.get("client")
|
||
|
|
# Options
|
||
|
|
currency = data.get("currency")
|
||
|
|
balances_enabled = data.get("balances_enabled")
|
||
|
|
payouts_enabled = data.get("payouts_enabled")
|
||
|
|
rigs_enabled = data.get("rigs_enabled")
|
||
|
|
devices_enabled = data.get("devices_enabled")
|
||
|
|
|
||
|
|
|
||
|
|
# Mining rig and device sensors
|
||
|
|
if rigs_enabled or devices_enabled:
|
||
|
|
rigs_coordinator = data.get("rigs_coordinator")
|
||
|
|
rig_data = await client.get_mining_rigs()
|
||
|
|
mining_rigs = rig_data.get("miningRigs")
|
||
|
|
_LOGGER.debug(f"Found {len(mining_rigs)} rigs")
|
||
|
|
|
||
|
|
if devices_enabled:
|
||
|
|
_LOGGER.debug("Device sensors enabled")
|
||
|
|
device_switches = create_device_switches(mining_rigs, rigs_coordinator,client)
|
||
|
|
async_add_entities(device_switches, True)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def create_device_switches(mining_rigs, coordinator, client):
|
||
|
|
device_switches = []
|
||
|
|
for rig_data in mining_rigs:
|
||
|
|
rig = MiningRig(rig_data)
|
||
|
|
devices = rig.devices.values()
|
||
|
|
_LOGGER.debug(
|
||
|
|
f"Found {len(devices)} device switches(s) for {rig.name} ({rig.id})"
|
||
|
|
)
|
||
|
|
for device in devices:
|
||
|
|
_LOGGER.debug(f"Creating {device.name} ({device.id}) switches")
|
||
|
|
device_switches.append(GPUSwitch(coordinator, rig, device, client))
|
||
|
|
|
||
|
|
return device_switches
|
||
|
|
|