Move get_vendor_name() to util.py

Author: bouwew

plugwise/common.py

@@ -5,22 +5,12 @@
 from __future__ import annotations
 
 from plugwise.constants import ModelData
-from plugwise.util import check_heater_central, check_model
+from plugwise.util import check_heater_central, check_model, get_vendor_name
 
 from defusedxml import ElementTree as etree
 from munch import Munch
 
 
-def get_vendor_name(module: etree, model_data: ModelData) -> ModelData:
-    """Helper-function for _get_model_data()."""
-    if (vendor_name := module.find("vendor_name").text) is not None:
-        model_data["vendor_name"] = vendor_name
-        if "Plugwise" in vendor_name:
-            model_data["vendor_name"] = vendor_name.split(" ", 1)[0]
-
-    return model_data
-
-
 class SmileCommon:
     """The SmileCommon class."""
 

plugwise/util.py

@@ -13,6 +13,7 @@
     SPECIAL_FORMAT,
     TEMP_CELSIUS,
     DeviceData,
+    ModelData,
 )
 
 from defusedxml import ElementTree as etree
@@ -89,6 +90,16 @@ def format_measure(measure: str, unit: str) -> float | int:
     return result
 
 
+def get_vendor_name(module: etree, model_data: ModelData) -> ModelData:
+    """Helper-function for _get_model_data()."""
+    if (vendor_name := module.find("vendor_name").text) is not None:
+        model_data["vendor_name"] = vendor_name
+        if "Plugwise" in vendor_name:
+            model_data["vendor_name"] = vendor_name.split(" ", 1)[0]
+
+    return model_data
+
+
 def power_data_local_format(
     attrs: dict[str, str], key_string: str, val: str
 ) -> float | int: