Просмотр исходного кода

fix(services): use a plain dictionary schema, validate entity id

Jason Rumney 1 неделя назад
Родитель
Сommit
3e5c5dd12d
1 измененных файлов с 6 добавлено и 7 удалено
  1. 6 7
      custom_components/tuya_local/services.py

+ 6 - 7
custom_components/tuya_local/services.py

@@ -12,18 +12,17 @@ from homeassistant.components.remote import (
 from homeassistant.components.remote import DOMAIN as REMOTE_DOMAIN
 from homeassistant.core import HomeAssistant, ServiceCall
 from homeassistant.helpers import service
+from homeassistant.helpers import config_validation as cv
 
 from .const import DOMAIN
 from .infrared import TuyaRemoteCommand
 from .remote import TuyaLocalRemote
 
-REMOTE_SEND_IR_COMMAND_SCHEMA = vol.Schema(
-    {
-        vol.Required("emitter_entity_id"): str,
-        vol.Required("code"): str,
-        vol.Optional("device"): str,
-    }
-)
+REMOTE_SEND_IR_COMMAND_SCHEMA = {
+    vol.Required("emitter_entity_id"): cv.entity_id,
+    vol.Required("code"): str,
+    vol.Optional("device"): str,
+}
 
 _LOGGER = logging.getLogger(__name__)