Replace not needed guard in command_line with type check (#123722)

pull/123674/head^2
G Johansson 2024-08-12 21:42:00 +02:00 committed by GitHub
parent 178cb0659a
commit b09c6654ec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 11 deletions

View File

@ -4,7 +4,7 @@ from __future__ import annotations
import asyncio
from datetime import datetime, timedelta
from typing import Any, cast
from typing import TYPE_CHECKING, Any, cast
from homeassistant.components.cover import CoverEntity
from homeassistant.const import (
@ -141,10 +141,10 @@ class CommandCover(ManualTriggerEntity, CoverEntity):
async def _async_query_state(self) -> str | None:
"""Query for the state."""
if self._command_state:
LOGGER.debug("Running state value command: %s", self._command_state)
return await async_check_output_or_log(self._command_state, self._timeout)
return None
if TYPE_CHECKING:
assert self._command_state
LOGGER.debug("Running state value command: %s", self._command_state)
return await async_check_output_or_log(self._command_state, self._timeout)
async def _update_entity_state(self, now: datetime | None = None) -> None:
"""Update the state of the entity."""

View File

@ -4,7 +4,7 @@ from __future__ import annotations
import asyncio
from datetime import datetime, timedelta
from typing import Any, cast
from typing import TYPE_CHECKING, Any, cast
from homeassistant.components.switch import ENTITY_ID_FORMAT, SwitchEntity
from homeassistant.const import (
@ -142,11 +142,11 @@ class CommandSwitch(ManualTriggerEntity, SwitchEntity):
async def _async_query_state(self) -> str | int | None:
"""Query for state."""
if self._command_state:
if self._value_template:
return await self._async_query_state_value(self._command_state)
return await self._async_query_state_code(self._command_state)
return None
if TYPE_CHECKING:
assert self._command_state
if self._value_template:
return await self._async_query_state_value(self._command_state)
return await self._async_query_state_code(self._command_state)
async def _update_entity_state(self, now: datetime | None = None) -> None:
"""Update the state of the entity."""