feat: shorten zone name if necessary

signalkraft · Dec 29, 2023 · dd65147 · dd65147
1 parent 35bdabe
commit dd65147
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/custom_components/mypyllant/climate.py b/custom_components/mypyllant/climate.py
@@ -45,6 +45,8 @@
     ZoneTimeProgram,
 )
 
+from custom_components.mypyllant.utils import shorten_zone_name
+
 from . import SystemCoordinator
 from .const import (
     DEFAULT_TIME_PROGRAM_OVERWRITE,
@@ -277,7 +279,7 @@ def circuit_name_suffix(self) -> str:
 
     @property
     def name_prefix(self) -> str:
-        return f"{self.system.home.home_name or self.system.home.nomenclature} Zone {self.zone.name}{self.circuit_name_suffix}"
+        return f"{self.system.home.home_name or self.system.home.nomenclature} Zone {shorten_zone_name(self.zone.name)}{self.circuit_name_suffix}"
 
     @property
     def id_infix(self) -> str:

diff --git a/custom_components/mypyllant/sensor.py b/custom_components/mypyllant/sensor.py
@@ -25,6 +25,8 @@
     Zone,
 )
 
+from custom_components.mypyllant.utils import shorten_zone_name
+
 from . import DailyDataCoordinator, SystemCoordinator
 from .const import DOMAIN
 
@@ -333,7 +335,7 @@ def circuit_name_suffix(self) -> str:
 
     @property
     def name_prefix(self) -> str:
-        return f"{self.system.home.home_name or self.system.home.nomenclature} Zone {self.zone.name}{self.circuit_name_suffix}"
+        return f"{self.system.home.home_name or self.system.home.nomenclature} Zone {shorten_zone_name(self.zone.name)}{self.circuit_name_suffix}"
 
     @property
     def id_infix(self) -> str:

diff --git a/custom_components/mypyllant/utils.py b/custom_components/mypyllant/utils.py
@@ -3,6 +3,12 @@
 from aiohttp.client_exceptions import ClientResponseError
 
 
+def shorten_zone_name(zone_name: str) -> str:
+    if zone_name.startswith("Zone "):
+        return zone_name[5:]
+    return zone_name
+
+
 def is_quota_exceeded_exception(exc_info: Exception) -> bool:
     """
     Returns True if the exception is a quota exceeded ClientResponseError