roborock-cli 0.1.1__py3-none-any.whl

roborock_cli/_vendor/roborock/const.py

@@ -0,0 +1,84 @@
+# Total time in seconds consumables have before Roborock recommends replacing
+MAIN_BRUSH_REPLACE_TIME = 1080000
+SIDE_BRUSH_REPLACE_TIME = 720000
+FILTER_REPLACE_TIME = 540000
+SENSOR_DIRTY_REPLACE_TIME = 108000
+MOP_ROLLER_REPLACE_TIME = 1080000
+STRAINER_REPLACE_TIME = 150
+CLEANING_BRUSH_REPLACE_TIME = 300
+DUST_COLLECTION_REPLACE_TIME = 90
+FLOOR_CLEANER_REPLACE_TIME = 300
+
+
+ROBOROCK_V1 = "ROBOROCK.vacuum.v1"
+ROBOROCK_S4 = "roborock.vacuum.s4"
+ROBOROCK_S4_MAX = "roborock.vacuum.a19"
+ROBOROCK_S5 = "roborock.vacuum.s5"
+ROBOROCK_S5_MAX = "roborock.vacuum.s5e"
+ROBOROCK_S6 = "roborock.vacuum.s6"
+ROBOROCK_T6 = "roborock.vacuum.t6"  # cn s6
+ROBOROCK_E4 = "roborock.vacuum.a01"
+ROBOROCK_S6_PURE = "roborock.vacuum.a08"
+ROBOROCK_T7 = "roborock.vacuum.a11"  # cn s7
+ROBOROCK_T7S = "roborock.vacuum.a14"
+ROBOROCK_T7SPLUS = "roborock.vacuum.a23"
+ROBOROCK_S7_MAXV = "roborock.vacuum.a27"
+ROBOROCK_S7_MAXV_ULTRA = "roborock.vacuum.a65"
+ROBOROCK_S7_PRO_ULTRA = "roborock.vacuum.a62"
+ROBOROCK_Q5 = "roborock.vacuum.a34"
+ROBOROCK_Q5_PRO = "roborock.vacuum.a72"
+ROBOROCK_Q7 = "roborock.vacuum.a40"
+ROBOROCK_Q7_MAX = "roborock.vacuum.a38"
+ROBOROCK_Q7PLUS = "roborock.vacuum.a40"
+ROBOROCK_QREVO_MASTER = "roborock.vacuum.a117"
+ROBOROCK_QREVO_CURV = "roborock.vacuum.a135"
+ROBOROCK_Q8_MAX = "roborock.vacuum.a73"
+ROBOROCK_Q10 = "roborock.vacuum.ss07"
+ROBOROCK_G10S_PRO = "roborock.vacuum.a26"
+ROBOROCK_G20S_Ultra = "roborock.vacuum.a143"  # cn saros_r10
+ROBOROCK_G10S = "roborock.vacuum.a46"
+ROBOROCK_G10 = "roborock.vacuum.a29"
+ROCKROBO_G10_SG = "roborock.vacuum.a30"  # Variant of the G10, has similar features as S7
+ROBOROCK_S7 = "roborock.vacuum.a15"
+ROBOROCK_S6_MAXV = "roborock.vacuum.a10"
+ROBOROCK_E2 = "roborock.vacuum.e2"
+ROBOROCK_1S = "roborock.vacuum.m1s"
+ROBOROCK_C1 = "roborock.vacuum.c1"
+ROBOROCK_S8_PRO_ULTRA = "roborock.vacuum.a70"
+ROBOROCK_S8 = "roborock.vacuum.a51"
+ROBOROCK_P10 = "roborock.vacuum.a75"  # also known as q_revo
+ROBOROCK_S8_MAXV_ULTRA = "roborock.vacuum.a97"
+ROBOROCK_QREVO_S = "roborock.vacuum.a104"
+ROBOROCK_QREVO_PRO = "roborock.vacuum.a101"
+ROBOROCK_QREVO_MAXV = "roborock.vacuum.a87"
+ROBOROCK_SAROS_10R = "roborock.vacuum.a144"
+ROBOROCK_SAROS_10 = "roborock.vacuum.a147"
+
+ROBOROCK_DYAD_AIR = "roborock.wetdryvac.a107"
+ROBOROCK_DYAD_PRO_COMBO = "roborock.wetdryvac.a83"
+ROBOROCK_DYAD_PRO = "roborock.wetdryvac.a56"
+
+# These are the devices that show up when you add a device - more could be supported and just not show up
+SUPPORTED_VACUUMS = [
+    ROBOROCK_G10,
+    ROBOROCK_G10S_PRO,
+    ROBOROCK_G20S_Ultra,
+    ROBOROCK_Q5,
+    ROBOROCK_Q7,
+    ROBOROCK_Q7_MAX,
+    ROBOROCK_S4,
+    ROBOROCK_S5_MAX,
+    ROBOROCK_S6,
+    ROBOROCK_S6_MAXV,
+    ROBOROCK_S6_PURE,
+    ROBOROCK_S7_MAXV,
+    ROBOROCK_S8_PRO_ULTRA,
+    ROBOROCK_S8,
+    ROBOROCK_S4_MAX,
+    ROBOROCK_S7,
+    ROBOROCK_P10,
+    ROBOROCK_Q10,
+    ROCKROBO_G10_SG,
+]
+
+NO_MAP = 63

roborock_cli/_vendor/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 *

roborock_cli/_vendor/roborock/data/b01_q10/__init__.py

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

roborock_cli/_vendor/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)
+    RECENT_CLEAN_RECORD = ("dpRecentCleanRecord", 53)  # NOTE: typo "dpRecendCleanRecord" in source code
+    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)
+    REQUEST_NOT_DISTURB_DATA = ("dpRequestNotDisturbData", 75)  # NOTE: typo "dpRequsetNotDisturbData" in source code
+    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_PROGRESS = ("dpCleanProgress", 87)  # NOTE: typo "dpCleanProgess" in source code
+    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)
+    REQUEST_DPS = ("dpRequestDps", 102)  # NOTE: typo "dpRequetdps" 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)
+    FAN_LEVEL = ("dpFanLevel", 123)  # NOTE: typo "dpfunLevel" in source code
+    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
+    QUIET = "quiet", 1
+    NORMAL = "normal", 2
+    STRONG = "strong", 3
+    MAX = "max", 4
+    SUPER = "super", 8
+
+
+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

roborock_cli/_vendor/roborock/data/b01_q10/b01_q10_containers.py

@@ -0,0 +1,102 @@
+"""Data container classes for Q10 B01 devices.
+
+Many of these classes use the `field(metadata={"dps": ...})` convention to map
+dataclass fields to device Data Points (DPS). This metadata is utilized by the
+`update_from_dps` helper in `roborock.devices.traits.b01.q10.common` to
+automatically update objects from raw device responses.
+"""
+
+from dataclasses import dataclass, field
+
+from ..containers import RoborockBase
+from .b01_q10_code_mappings import (
+    B01_Q10_DP,
+    YXBackType,
+    YXDeviceCleanTask,
+    YXDeviceState,
+    YXDeviceWorkMode,
+    YXFanLevel,
+    YXWaterLevel,
+)
+
+
+@dataclass
+class dpCleanRecord(RoborockBase):
+    op: str
+    result: int
+    id: str
+    data: list
+
+
+@dataclass
+class dpMultiMap(RoborockBase):
+    op: str
+    result: int
+    data: list
+
+
+@dataclass
+class dpGetCarpet(RoborockBase):
+    op: str
+    result: int
+    data: str
+
+
+@dataclass
+class dpSelfIdentifyingCarpet(RoborockBase):
+    op: str
+    result: int
+    data: str
+
+
+@dataclass
+class dpNetInfo(RoborockBase):
+    wifiName: str
+    ipAdress: str
+    mac: str
+    signal: int
+
+
+@dataclass
+class dpNotDisturbExpand(RoborockBase):
+    disturb_dust_enable: int
+    disturb_light: int
+    disturb_resume_clean: int
+    disturb_voice: int
+
+
+@dataclass
+class dpCurrentCleanRoomIds(RoborockBase):
+    room_id_list: list
+
+
+@dataclass
+class dpVoiceVersion(RoborockBase):
+    version: int
+
+
+@dataclass
+class dpTimeZone(RoborockBase):
+    timeZoneCity: str
+    timeZoneSec: int
+
+
+@dataclass
+class Q10Status(RoborockBase):
+    """Status for Q10 devices.
+
+    Fields are mapped to DPS values using metadata. Objects of this class can be
+    automatically updated using the `update_from_dps` helper.
+    """
+
+    clean_time: int | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_TIME})
+    clean_area: int | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_AREA})
+    battery: int | None = field(default=None, metadata={"dps": B01_Q10_DP.BATTERY})
+    status: YXDeviceState | None = field(default=None, metadata={"dps": B01_Q10_DP.STATUS})
+    fan_level: YXFanLevel | None = field(default=None, metadata={"dps": B01_Q10_DP.FAN_LEVEL})
+    water_level: YXWaterLevel | None = field(default=None, metadata={"dps": B01_Q10_DP.WATER_LEVEL})
+    clean_count: int | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_COUNT})
+    clean_mode: YXDeviceWorkMode | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_MODE})
+    clean_task_type: YXDeviceCleanTask | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_TASK_TYPE})
+    back_type: YXBackType | None = field(default=None, metadata={"dps": B01_Q10_DP.BACK_TYPE})
+    cleaning_progress: int | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEANING_PROGRESS})

roborock_cli/_vendor/roborock/data/b01_q7/__init__.py

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