Breakout get_vendor_name()

Author: bouwew

plugwise/common.py

@@ -11,6 +11,16 @@
 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."""
 
@@ -112,10 +122,7 @@ def _get_module_data(
             module = xml_2.find(loc)
             if module is not None:  # pylint: disable=consider-using-assignment-expr
                 model_data["contents"] = True
-                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]
+                get_vendor_name(module, model_data)
                 model_data["vendor_model"] = module.find("vendor_model").text
                 model_data["hardware_version"] = module.find("hardware_version").text
                 model_data["firmware_version"] = module.find("firmware_version").text