diff --git a/.gitignore b/.gitignore index a3b4c0c07..9e02bc771 100644 --- a/.gitignore +++ b/.gitignore @@ -28,4 +28,5 @@ test.py # rust so *.cpython*.so -.codspeed \ No newline at end of file +.codspeed +models_map.json \ No newline at end of file diff --git a/deebot_client/hardware/deebot/34vhpm.py b/deebot_client/hardware/deebot/34vhpm.py new file mode 120000 index 000000000..7e53f007c --- /dev/null +++ b/deebot_client/hardware/deebot/34vhpm.py @@ -0,0 +1 @@ +tpnwyu.py \ No newline at end of file diff --git a/deebot_client/hardware/deebot/cuoipb.py b/deebot_client/hardware/deebot/cuoipb.py new file mode 120000 index 000000000..aa242599b --- /dev/null +++ b/deebot_client/hardware/deebot/cuoipb.py @@ -0,0 +1 @@ +elrxgb.py \ No newline at end of file diff --git a/scripts/check_for_similar_models.py b/scripts/check_for_similar_models.py index 7c0deb96c..614f0aca0 100644 --- a/scripts/check_for_similar_models.py +++ b/scripts/check_for_similar_models.py @@ -3,8 +3,10 @@ from __future__ import annotations import asyncio +import json import logging import os +from pathlib import Path import time import aiohttp @@ -15,6 +17,12 @@ from deebot_client.util import md5 +def _save_models(models_map: dict[str, list[str]]) -> None: + """Save models to file.""" + with Path("models_map.json").open("w") as f: + f.write(json.dumps(models_map, indent=4)) + + async def main() -> None: """Execute script.""" async with aiohttp.ClientSession() as session: @@ -34,6 +42,8 @@ async def main() -> None: for key, value in (await api_client.get_product_iot_map()).items(): name_map.setdefault(value["name"], []).append(key) + await asyncio.get_event_loop().run_in_executor(None, _save_models, name_map) + # Load current models await asyncio.get_event_loop().run_in_executor(None, _load)