Skip to content

Commit 49a6f5b

Browse files
authored
Add support for T50 Pro (cuoipb) and DEEBOT T8 AIVI + (34vhpm) by similarity (#943)
1 parent 46edf16 commit 49a6f5b

4 files changed

Lines changed: 14 additions & 1 deletion

File tree

.gitignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,5 @@ test.py
2828
# rust so
2929
*.cpython*.so
3030

31-
.codspeed
31+
.codspeed
32+
models_map.json
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
tpnwyu.py
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
elrxgb.py

scripts/check_for_similar_models.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,10 @@
33
from __future__ import annotations
44

55
import asyncio
6+
import json
67
import logging
78
import os
9+
from pathlib import Path
810
import time
911

1012
import aiohttp
@@ -15,6 +17,12 @@
1517
from deebot_client.util import md5
1618

1719

20+
def _save_models(models_map: dict[str, list[str]]) -> None:
21+
"""Save models to file."""
22+
with Path("models_map.json").open("w") as f:
23+
f.write(json.dumps(models_map, indent=4))
24+
25+
1826
async def main() -> None:
1927
"""Execute script."""
2028
async with aiohttp.ClientSession() as session:
@@ -34,6 +42,8 @@ async def main() -> None:
3442
for key, value in (await api_client.get_product_iot_map()).items():
3543
name_map.setdefault(value["name"], []).append(key)
3644

45+
await asyncio.get_event_loop().run_in_executor(None, _save_models, name_map)
46+
3747
# Load current models
3848
await asyncio.get_event_loop().run_in_executor(None, _load)
3949

0 commit comments

Comments
 (0)