Skip to content

Commit c74feca

Browse files
committed
fix: add error for web calls and saros dock
1 parent f2d0c39 commit c74feca

File tree

4 files changed

+9
-0
lines changed

4 files changed

+9
-0
lines changed

roborock/code_mappings.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,7 @@ class RoborockDockTypeCode(RoborockEnum):
461461
s8_maxv_ultra_dock = 10
462462
qrevo_master_dock = 14
463463
qrevo_s_dock = 15
464+
saros_r10_dock = 16
464465
qrevo_curv_dock = 17
465466

466467

roborock/exceptions.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,3 +68,7 @@ class RoborockTooFrequentCodeRequests(RoborockException):
6868

6969
class RoborockMissingParameters(RoborockException):
7070
"""Class for Roborock missing parameters exceptions."""
71+
72+
73+
class RoborockTooManyRequest(RoborockException):
74+
"""Class for Roborock too many request exceptions."""

roborock/version_1_apis/roborock_client_v1.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@
7272
RoborockDockTypeCode.p10_pro_dock,
7373
RoborockDockTypeCode.s8_maxv_ultra_dock,
7474
RoborockDockTypeCode.qrevo_s_dock,
75+
RoborockDockTypeCode.saros_r10_dock,
7576
RoborockDockTypeCode.qrevo_curv_dock,
7677
]
7778
RT = TypeVar("RT", bound=RoborockBase)

roborock/web_api.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
RoborockMissingParameters,
2323
RoborockNoUserAgreement,
2424
RoborockTooFrequentCodeRequests,
25+
RoborockTooManyRequest,
2526
RoborockUrlException,
2627
)
2728

@@ -54,6 +55,8 @@ async def _get_base_url(self) -> str:
5455
raise RoborockMissingParameters(
5556
"You are missing parameters for this request, are you sure you " "entered your username?"
5657
)
58+
elif response_code == 9002:
59+
raise RoborockTooManyRequest("Please temporarily disable making requests and try again later.")
5760
raise RoborockUrlException(f"error code: {response_code} msg: {response.get('error')}")
5861
response_data = response.get("data")
5962
if response_data is None:

0 commit comments

Comments
 (0)