fix: issues with helium stats and improve error handling

Author: enes-oerdek

custom_components/helium_solana/sensor.py

@@ -103,17 +103,17 @@ async def get_sensors(wallets, prices):
         for price in prices:
             sensors.append(PriceSensor(price))
 
-    sensors.append(HeliumStats(heliumStatsAPI, 'IOT', 'total_hotspots', 'Total Hotspots', ['data', 'helium_iot', 'total_hotspots'], 'mdi:router-wireless', 'Hotspots'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'IOT', 'active_hotspots', 'Active Hotspots',  ['data', 'helium_iot', 'active_hotspots'], 'mdi:router-wireless', 'Hotspots'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'IOT', 'total_cities', 'Total Cities', ['data', 'helium_iot', 'total_cities'],'mdi:city', 'Cities'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'IOT', 'total_countries', 'Total Countries', ['data', 'helium_iot', 'total_countries'], 'mdi:earth', 'Countries'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'IOT', 'daily_average_rewards', 'Daily Average Rewards', ['data', 'helium_iot', 'daily_average_rewards'], 'mdi:hand-coin-outline', 'IOT', 'float'))
-
-    sensors.append(HeliumStats(heliumStatsAPI, 'MOBILE', 'total_hotspots', 'Total Hotspots', ['data', 'helium_mobile', 'total_hotspots'],'mdi:router-wireless', 'Hotspots'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'MOBILE', 'active_hotspots', 'Active Hotspots', ['data', 'helium_mobile', 'active_hotspots'],'mdi:router-wireless', 'Hotspots'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'MOBILE', 'total_cities', 'Total Cities', ['data', 'helium_mobile', 'total_cities'], 'mdi:city', 'Cities'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'MOBILE', 'total_countries', 'Total Countries', ['data', 'helium_mobile', 'total_countries'], 'mdi:earth', 'Countries'))
-    sensors.append(HeliumStats(heliumStatsAPI, 'MOBILE', 'daily_average_rewards', 'Daily Average Rewards', ['data', 'helium_mobile', 'daily_average_rewards'], 'mdi:hand-coin-outline', 'MOBILE' ,'float'))
+    sensors.append(HeliumStats(api_backend, 'IOT', 'total_hotspots', 'Total Hotspots', ['stats', 'iot', 'total_hotspots'], 'mdi:router-wireless', 'Hotspots'))
+    sensors.append(HeliumStats(api_backend, 'IOT', 'active_hotspots', 'Active Hotspots',  ['stats', 'iot', 'active_hotspots'], 'mdi:router-wireless', 'Hotspots'))
+    sensors.append(HeliumStats(api_backend, 'IOT', 'total_cities', 'Total Cities', ['stats', 'iot', 'total_cities'],'mdi:city', 'Cities'))
+    sensors.append(HeliumStats(api_backend, 'IOT', 'total_countries', 'Total Countries', ['stats', 'iot', 'total_countries'], 'mdi:earth', 'Countries'))
+    sensors.append(HeliumStats(api_backend, 'IOT', 'daily_average_rewards', 'Daily Average Rewards', ['stats', 'iot', 'daily_average_rewards'], 'mdi:hand-coin-outline', 'IOT', 'float'))
+
+    sensors.append(HeliumStats(api_backend, 'MOBILE', 'total_hotspots', 'Total Hotspots', ['stats', 'mobile', 'total_hotspots'],'mdi:router-wireless', 'Hotspots'))
+    sensors.append(HeliumStats(api_backend, 'MOBILE', 'active_hotspots', 'Active Hotspots', ['stats', 'mobile', 'active_hotspots'],'mdi:router-wireless', 'Hotspots'))
+    sensors.append(HeliumStats(api_backend, 'MOBILE', 'total_cities', 'Total Cities', ['stats', 'mobile', 'total_cities'], 'mdi:city', 'Cities'))
+    sensors.append(HeliumStats(api_backend, 'MOBILE', 'total_countries', 'Total Countries', ['stats', 'mobile', 'total_countries'], 'mdi:earth', 'Countries'))
+    sensors.append(HeliumStats(api_backend, 'MOBILE', 'daily_average_rewards', 'Daily Average Rewards', ['stats', 'mobile', 'daily_average_rewards'], 'mdi:hand-coin-outline', 'MOBILE' ,'float'))
 
     for wallet in wallets:
         len_wallet = len(wallet)

custom_components/helium_solana/sensors/HeliumStats.py

@@ -1,3 +1,4 @@
+
 from typing import Any, Callable, Dict, Optional
 from homeassistant.helpers.entity import (
     Entity,
@@ -8,6 +9,9 @@
     DOMAIN
 )
 
+import logging
+_LOGGER = logging.getLogger(__name__)
+
 
 class HeliumStats(Entity):
     """Helium Stats"""
@@ -71,7 +75,7 @@ async def async_update(self):
         try:
 
 
-            response = await self.api.get_data()
+            response = await self.api.get_data('heliumstats')
 
             if response.status_code != 200:
                 return

custom_components/helium_solana/sensors/HotspotReward.py

@@ -9,6 +9,9 @@
     DOMAIN
 )
 
+import logging
+_LOGGER = logging.getLogger(__name__)
+
 class HotspotReward(Entity):
     """Hotspot Reward"""
     def __init__(self, api, wallet, identifier, path, label, uom, icon):

custom_components/helium_solana/sensors/PriceSensor.py

@@ -11,6 +11,8 @@
 )
 import asyncio
 
+import logging
+_LOGGER = logging.getLogger(__name__)
 
 class PriceSensor(Entity):
     """Price Sensor for Solana tokens"""

custom_components/helium_solana/sensors/StakingRewardsPosition.py

@@ -8,6 +8,9 @@
     DOMAIN
 )
 
+import logging
+_LOGGER = logging.getLogger(__name__)
+
 class StakingRewardsPosition(Entity):
     """Staking Reward"""
     def __init__(self, api, wallet, delegated_position_key, data, icon):

custom_components/helium_solana/sensors/StakingRewardsToken.py

@@ -8,6 +8,9 @@
     DOMAIN
 )
 
+import logging
+_LOGGER = logging.getLogger(__name__)
+
 class StakingRewardsToken(Entity):
     """Staking Reward"""
     def __init__(self, api, wallet, token, icon):

custom_components/helium_solana/sensors/WalletBalance.py

@@ -9,6 +9,9 @@
 )
 
 
+import logging
+_LOGGER = logging.getLogger(__name__)
+
 class WalletBalance(Entity):
     """Wallet Balance"""
     def __init__(self, api, address, key, path, uom, icon):