python-roborock 2.61.0__tar.gz → 3.1.2__tar.gz

{python_roborock-2.61.0 → python_roborock-3.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-roborock
-Version: 2.61.0
+Version: 3.1.2
 Summary: A package to control Roborock vacuums.
 Project-URL: Repository, https://github.com/humbertogontijo/python-roborock
 Project-URL: Documentation, https://python-roborock.readthedocs.io/

{python_roborock-2.61.0 → python_roborock-3.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-roborock"
-version = "2.61.0"
+version = "3.1.2"
 description = "A package to control Roborock vacuums."
 authors = [{ name = "humbertogontijo", email = "humbertogontijo@users.noreply.github.com" }, {name="Lash-L"}, {name="allenporter"}]
 requires-python = ">=3.11, <4"
@@ -44,7 +44,7 @@ dev = [
     "pytest",
     "pre-commit>=3.5,<5.0",
     "mypy",
-    "ruff==0.14.0",
+    "ruff==0.14.1",
     "codespell",
     "pyshark>=0.6,<0.7",
     "aioresponses>=0.7.7,<0.8",
@@ -52,6 +52,8 @@ dev = [
     "pytest-timeout>=2.3.1,<3",
     "syrupy>=4.9.1,<5",
     "pdoc>=15.0.4,<16",
+    "pyyaml>=6.0.3",
+    "pyshark>=0.6",
 ]
 
 [tool.hatch.build.targets.sdist]
@@ -67,7 +69,18 @@ build-backend = "hatchling.build"
 [tool.semantic_release]
 branch = "main"
 version_toml = ["pyproject.toml:project.version"]
-build_command = "pip install uv && uv build"
+build_command = "pip install uv && uv lock --upgrade-package python-roborock && git add uv.lock && uv build"
+changelog_file = 'CHANGELOG.md'
+commit = true
+
+[tool.semantic_release.branches.main]
+match = "main"
+prerelease = false
+
+[tool.semantic_release.branches.temp-main-branch]
+match = "temp-main-branch"
+prerelease = false
+
 
 [tool.semantic_release.commit_parser_options]
 allowed_tags = [
@@ -92,9 +105,3 @@ asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "function"
 timeout = 30
 log_format = "%(asctime)s.%(msecs)03d %(levelname)s (%(threadName)s) [%(name)s] %(message)s"
-
-[tool.uv]
-dev-dependencies = [
-    "pyyaml>=6.0.3",
-    "pyshark>=0.6",
-]

{python_roborock-2.61.0 → python_roborock-3.1.2}/roborock/__init__.py

@@ -3,19 +3,14 @@
 .. include:: ../README.md
 """
 
-from roborock.b01_containers import *
-from roborock.code_mappings import *
-from roborock.containers import *
+from roborock.data import *
 from roborock.exceptions import *
 from roborock.roborock_typing import *
 
 from . import (
-    b01_containers,
-    clean_modes,
     cloud_api,
-    code_mappings,
     const,
-    containers,
+    data,
     exceptions,
     roborock_typing,
     version_1_apis,
@@ -27,13 +22,10 @@ __all__ = [
     "web_api",
     "version_1_apis",
     "version_a01_apis",
-    "containers",
-    "b01_containers",
     "const",
     "cloud_api",
-    "clean_modes",
-    "code_mappings",
     "roborock_typing",
     "exceptions",
+    "data",
     # Add new APIs here in the future when they are public e.g. devices/
 ]

{python_roborock-2.61.0 → python_roborock-3.1.2}/roborock/api.py

@@ -8,7 +8,7 @@ import time
 from abc import ABC, abstractmethod
 from typing import Any
 
-from .containers import (
+from .data import (
     DeviceData,
 )
 from .exceptions import (

{python_roborock-2.61.0 → python_roborock-3.1.2}/roborock/broadcast_protocol.py

@@ -19,7 +19,7 @@ from construct import (  # type: ignore
 from Crypto.Cipher import AES
 
 from roborock import RoborockException
-from roborock.containers import BroadcastMessage
+from roborock.data import BroadcastMessage
 from roborock.protocol import EncryptionAdapter, Utils, _Parser
 
 _LOGGER = logging.getLogger(__name__)

{python_roborock-2.61.0 → python_roborock-3.1.2}/roborock/cli.py

@@ -42,7 +42,7 @@ from pyshark.capture.live_capture import LiveCapture, UnknownInterfaceException
 from pyshark.packet.packet import Packet  # type: ignore
 
 from roborock import SHORT_MODEL_TO_ENUM, RoborockCommand
-from roborock.containers import CombinedMapInfo, DeviceData, HomeData, NetworkInfo, RoborockBase, UserData
+from roborock.data import CombinedMapInfo, DeviceData, HomeData, NetworkInfo, RoborockBase, UserData
 from roborock.device_features import DeviceFeatures
 from roborock.devices.cache import Cache, CacheData
 from roborock.devices.device import RoborockDevice
@@ -731,6 +731,16 @@ async def home(ctx, device_id: str, refresh: bool):
         click.echo("No maps discovered")
 
 
+@session.command()
+@click.option("--device_id", required=True)
+@click.pass_context
+@async_command
+async def network_info(ctx, device_id: str):
+    """Get device network information."""
+    context: RoborockContext = ctx.obj
+    await _display_v1_trait(context, device_id, lambda v1: v1.network_info)
+
+
 @click.command()
 @click.option("--device_id", required=True)
 @click.option("--cmd", required=True)
@@ -979,6 +989,7 @@ cli.add_command(child_lock)
 cli.add_command(dnd)
 cli.add_command(flow_led_status)
 cli.add_command(led_status)
+cli.add_command(network_info)
 
 
 def main():

{python_roborock-2.61.0 → python_roborock-3.1.2}/roborock/cloud_api.py

@@ -14,7 +14,7 @@ from paho.mqtt.enums import MQTTErrorCode
 from paho.mqtt.reasoncodes import ReasonCode  # type: ignore
 
 from .api import KEEPALIVE, RoborockClient
-from .containers import DeviceData, UserData
+from .data import DeviceData, UserData
 from .exceptions import RoborockException, VacuumError
 from .protocol import (
     Decoder,

python_roborock-3.1.2/roborock/data/__init__.py

@@ -0,0 +1,9 @@
+"""This module is meant to hold dataclasses and codemappings for various devices and protocols."""
+
+from .b01_q7 import *
+from .b01_q10 import *
+from .code_mappings import *
+from .containers import *
+from .dyad import *
+from .v1 import *
+from .zeo import *

python_roborock-3.1.2/roborock/data/b01_q10/__init__.py

@@ -0,0 +1,2 @@
+from .b01_q10_code_mappings import *
+from .b01_q10_containers import *

python_roborock-3.1.2/roborock/data/b01_q10/b01_q10_code_mappings.py

@@ -0,0 +1,213 @@
+from ..code_mappings import RoborockModeEnum
+
+
+class B01_Q10_DP(RoborockModeEnum):
+    CLEAN_TIME = ("dpCleanTime", 6)
+    CLEAN_AREA = ("dpCleanArea", 7)
+    SEEK = ("dpSeek", 11)
+    REMOTE = ("dpRemote", 12)
+    MAP_RESET = ("dpMapReset", 13)
+    REQUEST = ("dpRequest", 16)
+    RESET_SIDE_BRUSH = ("dpResetSideBrush", 18)
+    RESET_MAIN_BRUSH = ("dpResetMainBrush", 20)
+    RESET_FILTER = ("dpResetFilter", 22)
+    RAG_LIFE = ("dpRagLife", 23)
+    RESET_RAG_LIFE = ("dpResetRagLife", 24)
+    NOT_DISTURB = ("dpNotDisturb", 25)
+    VOLUME = ("dpVolume", 26)
+    BEAK_CLEAN = ("dpBeakClean", 27)
+    TOTAL_CLEAN_AREA = ("dpTotalCleanArea", 29)
+    TOTAL_CLEAN_COUNT = ("dpTotalCleanCount", 30)
+    TOTAL_CLEAN_TIME = ("dpTotalCleanTime", 31)
+    TIMER = ("dpTimer", 32)
+    NOT_DISTURB_DATA = ("dpNotDisturbData", 33)
+    DEVICE_INFO = ("dpDeviceInfo", 34)
+    VOICE_PACKAGE = ("dpVoicePackage", 35)
+    VOICE_LANGUAGE = ("dpVoiceLanguage", 36)
+    DUST_SWITCH = ("dpDustSwitch", 37)
+    CUSTOM_MODE = ("dpCustomMode", 39)
+    MOP_STATE = ("dpMopState", 40)
+    UNIT = ("dpUnit", 42)
+    CARPET_CLEAN_PREFER = ("dpCarpetCleanPrefer", 44)
+    AUTO_BOOST = ("dpAutoBoost", 45)
+    CHILD_LOCK = ("dpChildLock", 47)
+    DUST_SETTING = ("dpDustSetting", 50)
+    MAP_SAVE_SWITCH = ("dpMapSaveSwitch", 51)
+    CLEAN_RECORD = ("dpCleanRecord", 52)
+    RECEND_CLEAN_RECORD = ("dpRecendCleanRecord", 53)
+    RESTRICTED_ZONE = ("dpRestrictedZone", 54)
+    RESTRICTED_ZONE_UP = ("dpRestrictedZoneUp", 55)
+    VIRTUAL_WALL = ("dpVirtualWall", 56)
+    VIRTUAL_WALL_UP = ("dpVirtualWallUp", 57)
+    ZONED = ("dpZoned", 58)
+    ZONED_UP = ("dpZonedUp", 59)
+    MULTI_MAP_SWITCH = ("dpMultiMapSwitch", 60)
+    MULTI_MAP = ("dpMultiMap", 61)
+    CUSTOMER_CLEAN = ("dpCustomerClean", 62)
+    CUSTOMER_CLEAN_REQUEST = ("dpCustomerCleanRequest", 63)
+    GET_CARPET = ("dpGetCarpet", 64)
+    CARPET_UP = ("dpCarpetUp", 65)
+    SELF_IDENTIFYING_CARPET = ("dpSelfIdentifyingCarpet", 66)
+    SENSOR_LIFE = ("dpSensorLife", 67)
+    RESET_SENSOR = ("dpResetSensor", 68)
+    REQUEST_TIMER = ("dpRequestTimer", 69)
+    REMOVE_ZONED = ("dpRemoveZoned", 70)
+    REMOVE_ZONED_UP = ("dpRemoveZonedUp", 71)
+    ROOM_MERGE = ("dpRoomMerge", 72)
+    ROOM_SPLIT = ("dpRoomSplit", 73)
+    RESET_ROOM_NAME = ("dpResetRoomName", 74)
+    REQUSET_NOT_DISTURB_DATA = ("dpRequsetNotDisturbData", 75)
+    CARPET_CLEAN_TYPE = ("dpCarpetCleanType", 76)
+    BUTTON_LIGHT_SWITCH = ("dpButtonLightSwitch", 77)
+    CLEAN_LINE = ("dpCleanLine", 78)
+    TIME_ZONE = ("dpTimeZone", 79)
+    AREA_UNIT = ("dpAreaUnit", 80)
+    NET_INFO = ("dpNetInfo", 81)
+    CLEAN_ORDER = ("dpCleanOrder", 82)
+    ROBOT_TYPE = ("dpRobotType", 83)
+    LOG_SWITCH = ("dpLogSwitch", 84)
+    FLOOR_MATERIAL = ("dpFloorMaterial", 85)
+    LINE_LASER_OBSTACLE_AVOIDANCE = ("dpLineLaserObstacleAvoidance", 86)
+    CLEAN_PROGESS = ("dpCleanProgess", 87)
+    GROUND_CLEAN = ("dpGroundClean", 88)
+    IGNORE_OBSTACLE = ("dpIgnoreObstacle", 89)
+    FAULT = ("dpFault", 90)
+    CLEAN_EXPAND = ("dpCleanExpand", 91)
+    NOT_DISTURB_EXPAND = ("dpNotDisturbExpand", 92)
+    TIMER_TYPE = ("dpTimerType", 93)
+    CREATE_MAP_FINISHED = ("dpCreateMapFinished", 94)
+    ADD_CLEAN_AREA = ("dpAddCleanArea", 95)
+    ADD_CLEAN_STATE = ("dpAddCleanState", 96)
+    RESTRICTED_AREA = ("dpRestrictedArea", 97)
+    RESTRICTED_AREA_UP = ("dpRestrictedAreaUp", 98)
+    SUSPECTED_THRESHOLD = ("dpSuspectedThreshold", 99)
+    SUSPECTED_THRESHOLD_UP = ("dpSuspectedThresholdUp", 100)
+    COMMON = ("dpCommon", 101)
+    JUMP_SCAN = ("dpJumpScan", 101)
+    REQUETDPS = ("dpRequetdps", 102)  # NOTE: THIS TYPO IS FOUND IN SOURCE CODE
+    CLIFF_RESTRICTED_AREA = ("dpCliffRestrictedArea", 102)
+    CLIFF_RESTRICTED_AREA_UP = ("dpCliffRestrictedAreaUp", 103)
+    BREAKPOINT_CLEAN = ("dpBreakpointClean", 104)
+    VALLEY_POINT_CHARGING = ("dpValleyPointCharging", 105)
+    VALLEY_POINT_CHARGING_DATA_UP = ("dpValleyPointChargingDataUp", 106)
+    VALLEY_POINT_CHARGING_DATA = ("dpValleyPointChargingData", 107)
+    VOICE_VERSION = ("dpVoiceVersion", 108)
+    ROBOT_COUNTRY_CODE = ("dpRobotCountryCode", 109)
+    HEARTBEAT = ("dpHeartbeat", 110)
+    STATUS = ("dpStatus", 121)
+    BATTERY = ("dpBattery", 122)
+    FUN_LEVEL = ("dpfunLevel", 123)
+    WATER_LEVEL = ("dpWaterLevel", 124)
+    MAIN_BRUSH_LIFE = ("dpMainBrushLife", 125)
+    SIDE_BRUSH_LIFE = ("dpSideBrushLife", 126)
+    FILTER_LIFE = ("dpFilterLife", 127)
+    TASK_CANCEL_IN_MOTION = ("dpTaskCancelInMotion", 132)
+    OFFLINE = ("dpOffline", 135)
+    CLEAN_COUNT = ("dpCleanCount", 136)
+    CLEAN_MODE = ("dpCleanMode", 137)
+    CLEAN_TASK_TYPE = ("dpCleanTaskType", 138)
+    BACK_TYPE = ("dpBackType", 139)
+    CLEANING_PROGRESS = ("dpCleaningProgress", 141)
+    FLEEING_GOODS = ("dpFleeingGoods", 142)
+    START_CLEAN = ("dpStartClean", 201)
+    START_BACK = ("dpStartBack", 202)
+    START_DOCK_TASK = ("dpStartDockTask", 203)
+    PAUSE = ("dpPause", 204)
+    RESUME = ("dpResume", 205)
+    STOP = ("dpStop", 206)
+    USER_PLAN = ("dpUserPlan", 207)
+
+
+class YXFanLevel(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    CLOSE = "close", 0
+    QUITE = "quite", 1
+    NORMAL = "normal", 2
+    STRONG = "strong", 3
+    MAX = "max", 4
+    SUPER = "super", 5
+
+
+class YXWaterLevel(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    CLOSE = "close", 0
+    LOW = "low", 1
+    MIDDLE = "middle", 2
+    HIGH = "high", 3
+
+
+class YXCleanLine(RoborockModeEnum):
+    FAST = "fast", 0
+    DAILY = "daily", 1
+    FINE = "fine", 2
+
+
+class YXRoomMaterial(RoborockModeEnum):
+    HORIZONTAL_FLOOR_BOARD = "horizontalfloorboard", 0
+    VERTICAL_FLOOR_BOARD = "verticalfloorboard", 1
+    CERAMIC_TILE = "ceramictile", 2
+    OTHER = "other", 255
+
+
+class YXCleanType(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    BOTH_WORK = "bothwork", 1
+    ONLY_SWEEP = "onlysweep", 2
+    ONLY_MOP = "onlymop", 3
+
+
+class YXDeviceState(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    SLEEP_STATE = "sleepstate", 2
+    STANDBY_STATE = "standbystate", 3
+    CLEANING_STATE = "cleaningstate", 5
+    TO_CHARGE_STATE = "tochargestate", 6
+    REMOTEING_STATE = "remoteingstate", 7
+    CHARGING_STATE = "chargingstate", 8
+    PAUSE_STATE = "pausestate", 10
+    FAULT_STATE = "faultstate", 12
+    UPGRADE_STATE = "upgradestate", 14
+    DUSTING = "dusting", 22
+    CREATING_MAP_STATE = "creatingmapstate", 29
+    MAP_SAVE_STATE = "mapsavestate", 99
+    RE_LOCATION_STATE = "relocationstate", 101
+    ROBOT_SWEEPING = "robotsweeping", 102
+    ROBOT_MOPING = "robotmoping", 103
+    ROBOT_SWEEP_AND_MOPING = "robotsweepandmoping", 104
+    ROBOT_TRANSITIONING = "robottransitioning", 105
+    ROBOT_WAIT_CHARGE = "robotwaitcharge", 108
+
+
+class YXBackType(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    IDLE = "idle", 0
+    BACK_DUSTING = "backdusting", 4
+    BACK_CHARGING = "backcharging", 5
+
+
+class YXDeviceWorkMode(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    BOTH_WORK = "bothwork", 1
+    ONLY_SWEEP = "onlysweep", 2
+    ONLY_MOP = "onlymop", 3
+    CUSTOMIZED = "customized", 4
+    SAVE_WORRY = "saveworry", 5
+    SWEEP_MOP = "sweepmop", 6
+
+
+class YXDeviceCleanTask(RoborockModeEnum):
+    UNKNOWN = "unknown", -1
+    IDLE = "idle", 0
+    SMART = "smart", 1
+    ELECTORAL = "electoral", 2
+    DIVIDE_AREAS = "divideareas", 3
+    CREATING_MAP = "creatingmap", 4
+    PART = "part", 5
+
+
+class YXDeviceDustCollectionFrequency(RoborockModeEnum):
+    DAILY = "daily", 0
+    INTERVAL_15 = "interval_15", 15
+    INTERVAL_30 = "interval_30", 30
+    INTERVAL_45 = "interval_45", 45
+    INTERVAL_60 = "interval_60", 60

python_roborock-3.1.2/roborock/data/b01_q10/b01_q10_containers.py

@@ -0,0 +1,53 @@
+from ..containers import RoborockBase
+
+
+class dpCleanRecord(RoborockBase):
+    op: str
+    result: int
+    id: str
+    data: list
+
+
+class dpMultiMap(RoborockBase):
+    op: str
+    result: int
+    data: list
+
+
+class dpGetCarpet(RoborockBase):
+    op: str
+    result: int
+    data: str
+
+
+class dpSelfIdentifyingCarpet(RoborockBase):
+    op: str
+    result: int
+    data: str
+
+
+class dpNetInfo(RoborockBase):
+    wifiName: str
+    ipAdress: str
+    mac: str
+    signal: int
+
+
+class dpNotDisturbExpand(RoborockBase):
+    disturb_dust_enable: int
+    disturb_light: int
+    disturb_resume_clean: int
+    disturb_voice: int
+
+
+class dpCurrentCleanRoomIds(RoborockBase):
+    room_id_list: list
+
+
+class dpVoiceVersion(RoborockBase):
+    version: int
+
+
+class dpTimeZone(RoborockBase):
+    timeZoneCity: str
+    timeZoneSec: int

python_roborock-3.1.2/roborock/data/b01_q7/__init__.py

@@ -0,0 +1,2 @@
+from .b01_q7_code_mappings import *
+from .b01_q7_containers import *

python_roborock-2.61.0/roborock/b01_containers.py → python_roborock-3.1.2/roborock/data/b01_q7/b01_q7_code_mappings.py

@@ -1,7 +1,4 @@
-from dataclasses import dataclass, field
-
-from .code_mappings import RoborockModeEnum
-from .containers import RoborockBase
+from ..code_mappings import RoborockModeEnum
 
 
 class WorkStatusMapping(RoborockModeEnum):
@@ -125,64 +122,6 @@ class CarpetModeMapping(RoborockModeEnum):
     OFF = ("off", 2)
 
 
-@dataclass
-class NetStatus(RoborockBase):
-    """Represents the network status of the device."""
-
-    rssi: str
-    loss: int
-    ping: int
-    ip: str
-    mac: str
-    ssid: str
-    frequency: int
-    bssid: str
-
-
-@dataclass
-class OrderTotal(RoborockBase):
-    """Represents the order total information."""
-
-    total: int
-    enable: int
-
-
-@dataclass
-class Privacy(RoborockBase):
-    """Represents the privacy settings of the device."""
-
-    ai_recognize: int
-    dirt_recognize: int
-    pet_recognize: int
-    carpet_turbo: int
-    carpet_avoid: int
-    carpet_show: int
-    map_uploads: int
-    ai_agent: int
-    ai_avoidance: int
-    record_uploads: int
-    along_floor: int
-    auto_upgrade: int
-
-
-@dataclass
-class PvCharging(RoborockBase):
-    """Represents the photovoltaic charging status."""
-
-    status: int
-    begin_time: int
-    end_time: int
-
-
-@dataclass
-class Recommend(RoborockBase):
-    """Represents cleaning recommendations."""
-
-    sill: int
-    wall: int
-    room_id: list[int] = field(default_factory=list)
-
-
 class B01Fault(RoborockModeEnum):
     """B01 fault codes and their descriptions."""
 
@@ -347,71 +286,3 @@ class B01Fault(RoborockModeEnum):
     F_2113 = ("fault_2113", 2113)
     F_2114 = ("fault_2114", 2114)
     F_2115 = ("fault_2115", 2115)
-
-
-@dataclass
-class B01Props(RoborockBase):
-    """
-    Represents the complete properties and status for a Roborock B01 model.
-    This dataclass is generated based on the device's status JSON object.
-    """
-
-    status: WorkStatusMapping
-    fault: B01Fault
-    wind: SCWindMapping
-    water: int
-    mode: int
-    quantity: int
-    alarm: int
-    volume: int
-    hypa: int
-    main_brush: int
-    side_brush: int
-    mop_life: int
-    main_sensor: int
-    net_status: NetStatus
-    repeat_state: int
-    tank_state: int
-    sweep_type: int
-    clean_path_preference: int
-    cloth_state: int
-    time_zone: int
-    time_zone_info: str
-    language: int
-    cleaning_time: int
-    real_clean_time: int
-    cleaning_area: int
-    custom_type: int
-    sound: int
-    work_mode: WorkModeMapping
-    station_act: int
-    charge_state: int
-    current_map_id: int
-    map_num: int
-    dust_action: int
-    quiet_is_open: int
-    quiet_begin_time: int
-    quiet_end_time: int
-    clean_finish: int
-    voice_type: int
-    voice_type_version: int
-    order_total: OrderTotal
-    build_map: int
-    privacy: Privacy
-    dust_auto_state: int
-    dust_frequency: int
-    child_lock: int
-    multi_floor: int
-    map_save: int
-    light_mode: int
-    green_laser: int
-    dust_bag_used: int
-    order_save_mode: int
-    manufacturer: str
-    back_to_wash: int
-    charge_station_type: int
-    pv_cut_charge: int
-    pv_charging: PvCharging
-    serial_number: str
-    recommend: Recommend
-    add_sweep_status: int