Fix entity and device selector TypedDict's (#94510)

pull/94287/head^2
Raman Gupta 2023-06-13 04:52:55 -04:00 committed by GitHub
parent 0d1bcd8a8f
commit aab58ad89c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions

View File

@ -189,8 +189,6 @@ class DeviceFilterSelectorConfig(TypedDict, total=False):
integration: str
manufacturer: str
model: str
entity: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig]
filter: DeviceFilterSelectorConfig | list[DeviceFilterSelectorConfig]
class ActionSelectorConfig(TypedDict):
@ -546,14 +544,12 @@ class DateTimeSelector(Selector[DateTimeSelectorConfig]):
return data
class DeviceSelectorConfig(TypedDict, total=False):
class DeviceSelectorConfig(DeviceFilterSelectorConfig, total=False):
"""Class to represent a device selector config."""
integration: str
manufacturer: str
model: str
entity: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig]
multiple: bool
filter: DeviceFilterSelectorConfig | list[DeviceFilterSelectorConfig]
@SELECTORS.register("device")
@ -622,6 +618,7 @@ class EntitySelectorConfig(EntityFilterSelectorConfig, total=False):
exclude_entities: list[str]
include_entities: list[str]
multiple: bool
filter: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig]
@SELECTORS.register("entity")