Improve autodiscovered yeelights model detection (#24671)

* Improve autodiscovered yeelights model detection

* Lint fixes

* Logger warn fix

homeassistant/components/yeelight/__init__.py

@@ -122,17 +122,12 @@ def setup(hass, config):
     def device_discovered(_, info):
         _LOGGER.debug("Adding autodetected %s", info['hostname'])
 
-        device_type = info['device_type']
-
-        name = "yeelight_%s_%s" % (device_type,
+        name = "yeelight_%s_%s" % (info['device_type'],
                                    info['properties']['mac'])
-        ipaddr = info[CONF_HOST]
-        device_config = DEVICE_SCHEMA({
-            CONF_NAME: name,
-            CONF_MODEL: device_type
-        })
 
-        _setup_device(hass, config, ipaddr, device_config)
+        device_config = DEVICE_SCHEMA({CONF_NAME: name})
+
+        _setup_device(hass, config, info[CONF_HOST], device_config)
 
     discovery.listen(hass, SERVICE_YEELIGHT, device_discovered)
 

homeassistant/components/yeelight/light.py

@@ -178,8 +178,10 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
         _lights_setup_helper(YeelightWithAmbientLight)
         _lights_setup_helper(YeelightAmbientLight)
     else:
-        _LOGGER.error("Cannot determine device type for %s, %s",
-                      device.ipaddr, device.name)
+        _lights_setup_helper(YeelightGenericLight)
+        _LOGGER.warning("Cannot determine device type for %s, %s. "
+                        "Falling back to white only", device.ipaddr,
+                        device.name)
 
     hass.data[data_key] += lights
     add_entities(lights, True)