Skip to content

Commit 4d978f0

Browse files
committed
Add init to StickNetworkInfoRequest
1 parent 0ed8baf commit 4d978f0

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

plugwise_usb/messages/requests.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -355,8 +355,18 @@ class StickNetworkInfoRequest(PlugwiseRequest):
355355
_identifier = b"0001"
356356
_reply_identifier = b"0002"
357357

358+
def __init__(
359+
self,
360+
send_fn: Callable[[PlugwiseRequest, bool], Awaitable[PlugwiseResponse | None]],
361+
) -> None:
362+
"""Initialize StickInitRequest message object."""
363+
super().__init__(send_fn, None)
364+
self._max_retries = 1
365+
358366
async def send(self) -> StickNetworkInfoResponse | None:
359367
"""Send request."""
368+
if self._send_fn is None:
369+
raise MessageError("Send function missing")
360370
result = await self._send_request()
361371
if isinstance(result, StickNetworkInfoResponse):
362372
return result

0 commit comments

Comments
 (0)