Improve code readability (#119558)

This commit is contained in:
Matrix 2024-06-13 15:31:29 +08:00 committed by GitHub
parent 440771bdea
commit 55f8a36572
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -70,18 +70,22 @@ DEVICE_TYPES: tuple[YoLinkSwitchEntityDescription, ...] = (
translation_key="plug_1",
device_class=SwitchDeviceClass.OUTLET,
exists_fn=lambda device: device.device_type == ATTR_DEVICE_MULTI_OUTLET,
plug_index_fn=lambda device: 1
if device.device_model_name.startswith(DEV_MODEL_MULTI_OUTLET_YS6801)
else 0,
plug_index_fn=lambda device: (
1
if device.device_model_name.startswith(DEV_MODEL_MULTI_OUTLET_YS6801)
else 0
),
),
YoLinkSwitchEntityDescription(
key="multi_outlet_plug_2",
translation_key="plug_2",
device_class=SwitchDeviceClass.OUTLET,
exists_fn=lambda device: device.device_type == ATTR_DEVICE_MULTI_OUTLET,
plug_index_fn=lambda device: 2
if device.device_model_name.startswith(DEV_MODEL_MULTI_OUTLET_YS6801)
else 1,
plug_index_fn=lambda device: (
2
if device.device_model_name.startswith(DEV_MODEL_MULTI_OUTLET_YS6801)
else 1
),
),
YoLinkSwitchEntityDescription(
key="multi_outlet_plug_3",