Skip to content

Commit 89e1f28

Browse files
fix: issue building roborock message
1 parent 4956efc commit 89e1f28

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

roborock/protocol.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@
99
from asyncio import BaseTransport, Lock
1010
from typing import Callable
1111

12+
from Crypto.Cipher import AES
13+
from Crypto.Util.Padding import pad, unpad
1214
from construct import ( # type: ignore
1315
Bytes,
1416
Checksum,
@@ -28,8 +30,6 @@
2830
stream_seek,
2931
stream_tell,
3032
)
31-
from Crypto.Cipher import AES
32-
from Crypto.Util.Padding import pad, unpad
3333

3434
from roborock import BroadcastMessage, RoborockException
3535
from roborock.roborock_message import RoborockMessage
@@ -339,7 +339,9 @@ def build(
339339
},
340340
}
341341
)
342-
return self.con.build({"messages": [message for message in messages]}, local_key=local_key, prefixed=prefixed)
342+
return self.con.build(
343+
{"messages": [message for message in messages], "remaining": b""}, local_key=local_key, prefixed=prefixed
344+
)
343345

344346

345347
MessageParser: _Parser = _Parser(_Messages, True)

0 commit comments

Comments
 (0)