more reliable friendly names handling, fixes #20 #21

benleb · Jun 9, 2020 · 26ddf2c · 26ddf2c
1 parent e7aff47
commit 26ddf2c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/apps/ench/ench.py b/apps/ench/ench.py
@@ -49,8 +49,10 @@ def hl(text: Union[int, float, str]) -> str:
 
 
 def hl_entity(entity: str) -> str:
-    domain, entity = entity.split(".")
-    return f"{domain}.{hl(entity)}"
+    if (len(splitted := entity.split(".")) > 1):
+        return f"{splitted[0]}.{hl(splitted[1])}"
+    else:
+        return f"{hl(entity)}"
 
 
 class EnCh(hass.Hass):  # type: ignore