Update config options and use GRPC if possible.

Schnitzel · Nov 19, 2023 · 71cc42e · 71cc42e
1 parent e67f559
commit 71cc42e
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 17 deletions.
diff --git a/custom_components/miner/coordinator.py b/custom_components/miner/coordinator.py
@@ -20,7 +20,7 @@
 
 
 class MinerCoordinator(DataUpdateCoordinator):
-    """Class to manage fetching update data from the IoTaWatt Energy Device."""
+    """Class to manage fetching update data from the Miner."""
 
     miner: pyasic.AnyMiner | None = None
 
@@ -53,7 +53,19 @@ async def _async_update_data(self):
                 self.miner.username = miner_username
                 self.miner.pwd = miner_password
 
-            miner_data = await self.miner.get_data()
+            miner_data = await self.miner.get_data(
+                include=[
+                    "hostname",
+                    "mac",
+                    "is_mining",
+                    "fw_ver",
+                    "hashrate",
+                    "nominal_hashrate",
+                    "hashboards",
+                    "wattage",
+                    "wattage_limit",
+                ]
+            )
 
         except pyasic.APIError as err:
             raise UpdateFailed("API Error") from err

diff --git a/custom_components/miner/number.py b/custom_components/miner/number.py
@@ -4,6 +4,7 @@
 import logging
 
 import pyasic
+from pyasic.miners.backends import BOSMiner
 from homeassistant.components.number import NumberEntity
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import callback
@@ -35,13 +36,14 @@ def _create_entity(key: str):
         created.add(key)
 
     await coordinator.async_config_entry_first_refresh()
-    async_add_entities(
-        [
-            MinerPowerLimitNumber(
-                coordinator=coordinator,
-            )
-        ]
-    )
+    if coordinator.miner.supports_autotuning:
+        async_add_entities(
+            [
+                MinerPowerLimitNumber(
+                    coordinator=coordinator,
+                )
+            ]
+        )
 
     # @callback
     # def new_data_received():
@@ -118,7 +120,14 @@ async def async_set_native_value(self, value):
                 f"{self.coordinator.entry.title} does not support setting power limit."
             )
 
-        result = await miner.set_power_limit(int(value))
+        if isinstance(miner, BOSMiner):
+            try:
+                result = await miner.web.grpc.set_power_target(int(value))
+            except pyasic.APIError:
+                result = await miner.set_power_limit(int(value))
+        else:
+            result = await miner.set_power_limit(int(value)) # noqa: ignore miner being assumed to be None
+
         if not result:
             raise pyasic.APIError("Failed to set wattage.")
 

diff --git a/custom_components/miner/switch.py b/custom_components/miner/switch.py
@@ -44,13 +44,14 @@ def _create_entity(key: str) -> SwitchEntity:
         created.add(key)
 
     await coordinator.async_config_entry_first_refresh()
-    async_add_entities(
-        [
-            MinerActiveSwitch(
-                coordinator=coordinator,
-            )
-        ]
-    )
+    if coordinator.miner.supports_shutdown:
+        async_add_entities(
+            [
+                MinerActiveSwitch(
+                    coordinator=coordinator,
+                )
+            ]
+        )
 
     # @callback
     # def new_data_received():