remote/adaptive-lighting
1
0
Fork 0
mirror of https://github.com/basnijholt/adaptive-lighting.git synced 2026-09-11 22:34:04 +02:00
Code Issues Projects Releases Packages Wiki Activity

simplify _split_service_data

Browse source
This commit is contained in:
Bas Nijholt 2020-12-19 14:06:48 +01:00
commit ff2321ece9
1 changed files with 8 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

22
custom_components/adaptive_lighting/switch.py
View file

@ -184,25 +184,19 @@ def is_our_context(context: Optional[Context]) -> bool:
return context.id.startswith(_DOMAIN_SHORT)
def _copy_and_pop(dct, keys):
"""Copy a dictionary and remove 'keys' if they exist."""
copy = dct.copy()
for key in keys:
copy.pop(key, None)
return copy
def _split_service_data(service_data, adapt_brightness, adapt_color):
"""Split service_data into two dictionaries (for color and brightness)."""
service_datas = []
if adapt_color:
service_datas.append(
_copy_and_pop(service_data, (ATTR_WHITE_VALUE, ATTR_BRIGHTNESS))
)
service_data_color = service_data.copy()
service_data_color.pop(ATTR_WHITE_VALUE, None)
service_data_color.pop(ATTR_BRIGHTNESS, None)
service_datas.append(service_data_color)
if adapt_brightness:
service_datas.append(
_copy_and_pop(service_data, (ATTR_RGB_COLOR, ATTR_COLOR_TEMP))
)
service_data_brightness = service_data.copy()
service_data_brightness.pop(ATTR_RGB_COLOR, None)
service_data_brightness.pop(ATTR_COLOR_TEMP, None)
service_datas.append(service_data_brightness)
return service_datas