set device tracker name and add hostname to attributes

elad-bar · May 2, 2024 · 5e77b5a · 5e77b5a
1 parent d34b53d
commit 5e77b5a
Show file tree

Hide file tree

Showing 8 changed files with 31 additions and 3 deletions.
diff --git a/custom_components/edgeos/common/consts.py b/custom_components/edgeos/common/consts.py
@@ -65,6 +65,7 @@
 ATTR_ACTIONS = "actions"
 ATTR_IS_ON = "is_on"
 ATTR_LAST_ACTIVITY = "last activity"
+ATTR_HOSTNAME = "hostname"
 
 ACTION_ENTITY_TURN_ON = "turn_on"
 ACTION_ENTITY_TURN_OFF = "turn_off"

diff --git a/custom_components/edgeos/data_processors/system_processor.py b/custom_components/edgeos/data_processors/system_processor.py
@@ -63,8 +63,6 @@ def get(self) -> EdgeOSSystemData:
     def get_device_info(self, item_id: str | None = None) -> DeviceInfo:
         name = self._system.hostname.upper()
 
-        _LOGGER.error(self._system)
-
         device_info = DeviceInfo(
             identifiers={(DEFAULT_NAME, name)},
             name=name,

diff --git a/custom_components/edgeos/device_tracker.py b/custom_components/edgeos/device_tracker.py
@@ -15,7 +15,7 @@
 from homeassistant.core import HomeAssistant
 
 from .common.base_entity import IntegrationBaseEntity, async_setup_base_entry
-from .common.consts import ATTR_ATTRIBUTES, ATTR_IS_ON
+from .common.consts import ATTR_ATTRIBUTES, ATTR_HOSTNAME, ATTR_IS_ON
 from .common.entity_descriptions import IntegrationDeviceTrackerEntityDescription
 from .common.enums import DeviceTypes
 from .managers.coordinator import Coordinator
@@ -51,12 +51,18 @@ def __init__(
         self._attr_mac_address: str | None = None
         self._attr_source_type: SourceType | str | None = SourceType.ROUTER
         self._attr_is_connected: bool = False
+        self._attr_hostname: str | None = None
 
     @property
     def ip_address(self) -> str | None:
         """Return the primary ip address of the device."""
         return self._attr_ip_address
 
+    @property
+    def hostname(self) -> str | None:
+        """Return the hostname of the device."""
+        return self._attr_hostname
+
     @property
     def mac_address(self) -> str | None:
         """Return the mac address of the device."""
@@ -82,6 +88,7 @@ def update_component(self, data):
             self._attr_is_connected = is_connected
             self._attr_ip_address = attributes.get(ATTR_IP)
             self._attr_mac_address = attributes.get(ATTR_MAC)
+            self._attr_hostname = attributes.get(ATTR_HOSTNAME)
 
             self._attr_extra_state_attributes = {
                 attribute: attributes[attribute]

diff --git a/custom_components/edgeos/managers/coordinator.py b/custom_components/edgeos/managers/coordinator.py
@@ -24,6 +24,7 @@
     API_RECONNECT_INTERVAL,
     ATTR_ACTIONS,
     ATTR_ATTRIBUTES,
+    ATTR_HOSTNAME,
     ATTR_IS_ON,
     ATTR_LAST_ACTIVITY,
     DOMAIN,
@@ -734,6 +735,7 @@ def _get_device_tracker_data(
                 ATTR_LAST_ACTIVITY: last_activity,
                 ATTR_IP: device.ip,
                 ATTR_MAC: device.mac,
+                ATTR_HOSTNAME: device.hostname,
             },
         }
 

diff --git a/custom_components/edgeos/strings.json b/custom_components/edgeos/strings.json
@@ -58,6 +58,11 @@
         "name": "Connected"
       }
     },
+    "device_tracker": {
+      "device_tracker": {
+        "name": "Status"
+      }
+    },
     "number": {
       "consider_away_interval": {
         "name": "Consider Away Interval"

diff --git a/custom_components/edgeos/translations/en.json b/custom_components/edgeos/translations/en.json
@@ -34,6 +34,11 @@
         "name": "Connected"
       }
     },
+    "device_tracker": {
+      "device_tracker": {
+        "name": "Status"
+      }
+    },
     "number": {
       "consider_away_interval": {
         "name": "Consider Away Interval"

diff --git a/custom_components/edgeos/translations/nb.json b/custom_components/edgeos/translations/nb.json
@@ -34,6 +34,11 @@
         "name": "Tilkoblet"
       }
     },
+    "device_tracker": {
+      "device_tracker": {
+        "name": "Status"
+      }
+    },
     "number": {
       "consider_away_interval": {
         "name": "Vurder bort intervall"

diff --git a/custom_components/edgeos/translations/pt-BR.json b/custom_components/edgeos/translations/pt-BR.json
@@ -34,6 +34,11 @@
         "name": "Conectada Conectado"
       }
     },
+    "device_tracker": {
+      "device_tracker": {
+        "name": "Status"
+      }
+    },
     "number": {
       "consider_away_interval": {
         "name": "Considere o intervalo fora"