Upstream
/
esphome
mirror of https://github.com/esphome/esphome.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Housecleaning: Use walrus operator in light (#6556)

This commit is contained in:
Jesse Hills 2024-04-17 18:26:36 +12:00 committed by GitHub
parent 2fc2d5839f
commit 3f015562d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
esphome/components/light/__init__.py
View File

@ -137,18 +137,16 @@ async def setup_light_core_(light_var, output_var, config):
cg.add(light_var.set_restore_mode(config[CONF_RESTORE_MODE])) cg.add(light_var.set_restore_mode(config[CONF_RESTORE_MODE]))
if CONF_DEFAULT_TRANSITION_LENGTH in config: if (
cg.add( default_transition_length := config.get(CONF_DEFAULT_TRANSITION_LENGTH)
light_var.set_default_transition_length( ) is not None:
config[CONF_DEFAULT_TRANSITION_LENGTH] cg.add(light_var.set_default_transition_length(default_transition_length))
) if (
) flash_transition_length := config.get(CONF_FLASH_TRANSITION_LENGTH)
if CONF_FLASH_TRANSITION_LENGTH in config: ) is not None:
cg.add( cg.add(light_var.set_flash_transition_length(flash_transition_length))
light_var.set_flash_transition_length(config[CONF_FLASH_TRANSITION_LENGTH]) if (gamma_correct := config.get(CONF_GAMMA_CORRECT)) is not None:
) cg.add(light_var.set_gamma_correct(gamma_correct))
if CONF_GAMMA_CORRECT in config:
cg.add(light_var.set_gamma_correct(config[CONF_GAMMA_CORRECT]))
effects = await cg.build_registry_list( effects = await cg.build_registry_list(
EFFECTS_REGISTRY, config.get(CONF_EFFECTS, []) EFFECTS_REGISTRY, config.get(CONF_EFFECTS, [])
) )
@ -164,15 +162,15 @@ async def setup_light_core_(light_var, output_var, config):
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], light_var) trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], light_var)
await auto.build_automation(trigger, [], conf) await auto.build_automation(trigger, [], conf)
if CONF_COLOR_CORRECT in config: if (color_correct := config.get(CONF_COLOR_CORRECT)) is not None:
cg.add(output_var.set_correction(*config[CONF_COLOR_CORRECT])) cg.add(output_var.set_correction(*color_correct))
if CONF_POWER_SUPPLY in config: if (power_supply_id := config.get(CONF_POWER_SUPPLY)) is not None:
var_ = await cg.get_variable(config[CONF_POWER_SUPPLY]) var_ = await cg.get_variable(power_supply_id)
cg.add(output_var.set_power_supply(var_)) cg.add(output_var.set_power_supply(var_))
if CONF_MQTT_ID in config: if (mqtt_id := config.get(CONF_MQTT_ID)) is not None:
mqtt_ = cg.new_Pvariable(config[CONF_MQTT_ID], light_var) mqtt_ = cg.new_Pvariable(mqtt_id, light_var)
await mqtt.register_mqtt_component(mqtt_, config) await mqtt.register_mqtt_component(mqtt_, config)