A few fixes

This commit is contained in:
Jesse Hills 2024-04-10 04:09:29 +00:00
parent b5aa9029a0
commit fbeca664e3
No known key found for this signature in database
GPG Key ID: BEAAE804EFD8E83A
3 changed files with 4 additions and 11 deletions

View File

@ -1164,17 +1164,13 @@ class APIClient:
position: float | None = None, position: float | None = None,
stop: bool = False, stop: bool = False,
) -> None: ) -> None:
connection = self._get_connection()
req = ValveCommandRequest(key=key) req = ValveCommandRequest(key=key)
apiv = self.api_version
if TYPE_CHECKING:
assert apiv is not None
if position is not None: if position is not None:
req.has_position = True req.has_position = True
req.position = position req.position = position
if stop: if stop:
req.stop = stop req.stop = stop
connection.send_message(req) self._get_connection().send_message(req)
def media_player_command( def media_player_command(
self, self,

View File

@ -372,6 +372,6 @@ MESSAGE_TYPE_TO_PROTO = {
105: TimeCommandRequest, 105: TimeCommandRequest,
106: VoiceAssistantAudio, 106: VoiceAssistantAudio,
109: ListEntitiesValveResponse, 109: ListEntitiesValveResponse,
110: ValveCommandRequest, 110: ValveStateResponse,
111: ValveStateResponse, 111: ValveCommandRequest,
} }

View File

@ -756,10 +756,10 @@ class LockEntityState(EntityState):
# ==================== VALVE ==================== # ==================== VALVE ====================
@_frozen_dataclass_decorator @_frozen_dataclass_decorator
class ValveInfo(EntityInfo): class ValveInfo(EntityInfo):
device_class: str = ""
assumed_state: bool = False assumed_state: bool = False
supports_stop: bool = False supports_stop: bool = False
supports_position: bool = False supports_position: bool = False
device_class: str = ""
class ValveOperation(APIIntEnum): class ValveOperation(APIIntEnum):
@ -777,9 +777,6 @@ class ValveState(EntityState):
default=ValveOperation.IDLE, converter=ValveOperation.convert default=ValveOperation.IDLE, converter=ValveOperation.convert
) )
def is_closed(self) -> bool:
return self.position == 0.0
# ==================== MEDIA PLAYER ==================== # ==================== MEDIA PLAYER ====================
class MediaPlayerState(APIIntEnum): class MediaPlayerState(APIIntEnum):