from __future__ import annotations from dataclasses import dataclass, field from enum import Enum, StrEnum from .containers import ( CleanRecord, CleanSummary, Consumable, DustCollectionMode, RoborockBase, SmartWashParams, Status, WashTowelMode, ) class RoborockCommand(str, Enum): ADD_MOP_TEMPLATE_PARAMS = "add_mop_template_params" APP_AMETHYST_SELF_CHECK = "app_amethyst_self_check" APP_CHARGE = "app_charge" APP_DELETE_WIFI = "app_delete_wifi" APP_GET_AMETHYST_STATUS = "app_get_amethyst_status" APP_GET_CARPET_DEEP_CLEAN_STATUS = "app_get_carpet_deep_clean_status" APP_GET_CLEAN_ESTIMATE_INFO = "app_get_clean_estimate_info" APP_GET_DRYER_SETTING = "app_get_dryer_setting" APP_GET_INIT_STATUS = "app_get_init_status" APP_GET_LOCALE = "app_get_locale" APP_GET_WIFI_LIST = "app_get_wifi_list" APP_GOTO_TARGET = "app_goto_target" APP_KEEP_EASTER_EGG = "app_keep_easter_egg" APP_PAUSE = "app_pause" APP_RC_END = "app_rc_end" APP_RC_MOVE = "app_rc_move" APP_RC_START = "app_rc_start" APP_RC_STOP = "app_rc_stop" APP_RESUME_BUILD_MAP = "app_resume_build_map" APP_RESUME_PATROL = "app_resume_patrol" APP_SEGMENT_CLEAN = "app_segment_clean" APP_SET_AMETHYST_STATUS = "app_set_amethyst_status" APP_SET_CARPET_DEEP_CLEAN_STATUS = "app_set_carpet_deep_clean_status" APP_SET_CROSS_CARPET_CLEANING_STATUS = "app_set_cross_carpet_cleaning_status" APP_SET_DOOR_SILL_BLOCKS = "app_set_door_sill_blocks" APP_SET_DIRTY_REPLENISH_CLEAN_STATUS = "app_set_dirty_replenish_clean_status" APP_SET_DRYER_SETTING = "app_set_dryer_setting" APP_SET_DRYER_STATUS = "app_set_dryer_status" APP_SET_DYNAMIC_CONFIG = "app_set_dynamic_config" APP_SET_IGNORE_STUCK_POINT = "app_set_ignore_stuck_point" APP_SET_SMART_CLIFF_FORBIDDEN = "app_set_smart_cliff_forbidden" APP_SET_SMART_DOOR_SILL = "app_set_smart_door_sill" APP_SPOT = "app_spot" APP_START = "app_start" APP_START_BUILD_MAP = "app_start_build_map" APP_START_COLLECT_DUST = "app_start_collect_dust" APP_START_EASTER_EGG = "app_start_easter_egg" APP_START_PATROL = "app_start_patrol" APP_START_PET_PATROL = "app_start_pet_patrol" APP_START_WASH = "app_start_wash" APP_STAT = "app_stat" APP_STOP = "app_stop" APP_STOP_COLLECT_DUST = "app_stop_collect_dust" APP_STOP_WASH = "app_stop_wash" APP_UPDATE_UNSAVE_MAP = "app_update_unsave_map" APP_WAKEUP_ROBOT = "app_wakeup_robot" APP_ZONED_CLEAN = "app_zoned_clean" CHANGE_SOUND_VOLUME = "change_sound_volume" CHECK_HOMESEC_PASSWORD = "check_homesec_password" CLOSE_DND_TIMER = "close_dnd_timer" CLOSE_VALLEY_ELECTRICITY_TIMER = "close_valley_electricity_timer" DEL_CLEAN_RECORD = "del_clean_record" DEL_CLEAN_RECORD_MAP_V2 = "del_clean_record_map_v2" DEL_MAP = "del_map" DEL_MOP_TEMPLATE_PARAMS = "del_mop_template_params" DEL_SERVER_TIMER = "del_server_timer" DEL_TIMER = "del_timer" DNLD_INSTALL_SOUND = "dnld_install_sound" ENABLE_HOMESEC_VOICE = "enable_homesec_voice" ENABLE_LOG_UPLOAD = "enable_log_upload" END_EDIT_MAP = "end_edit_map" FIND_ME = "find_me" GET_AUTO_DELIVERY_CLEANING_FLUID = "get_auto_delivery_cleaning_fluid" GET_CAMERA_STATUS = "get_camera_status" GET_CARPET_CLEAN_MODE = "get_carpet_clean_mode" GET_CARPET_MODE = "get_carpet_mode" GET_CHILD_LOCK_STATUS = "get_child_lock_status" GET_CLEAN_FOLLOW_GROUND_MATERIAL_STATUS = "get_clean_follow_ground_material_status" GET_CLEAN_MOTOR_MODE = "get_clean_motor_mode" GET_CLEAN_RECORD = "get_clean_record" GET_CLEAN_RECORD_MAP = "get_clean_record_map" GET_CLEAN_SEQUENCE = "get_clean_sequence" GET_CLEAN_SUMMARY = "get_clean_summary" GET_COLLISION_AVOID_STATUS = "get_collision_avoid_status" GET_CONSUMABLE = "get_consumable" GET_CURRENT_SOUND = "get_current_sound" GET_CUSTOM_MODE = "get_custom_mode" GET_CUSTOMIZE_CLEAN_MODE = "get_customize_clean_mode" GET_DEVICE_ICE = "get_device_ice" GET_DEVICE_SDP = "get_device_sdp" GET_DND_TIMER = "get_dnd_timer" GET_DOCK_INFO = "get_dock_info" GET_DUST_COLLECTION_MODE = "get_dust_collection_mode" GET_DUST_COLLECTION_SWITCH_STATUS = "get_dust_collection_switch_status" GET_DYNAMIC_DATA = "get_dynamic_data" GET_DYNAMIC_MAP_DIFF = "get_dynamic_map_diff" GET_FAN_MOTOR_WORK_TIMEOUT = "get_fan_motor_work_timeout" GET_FLOW_LED_STATUS = "get_flow_led_status" GET_FRESH_MAP = "get_fresh_map" GET_FW_FEATURES = "get_fw_features" GET_HOMESEC_CONNECT_STATUS = "get_homesec_connect_status" GET_IDENTIFY_FURNITURE_STATUS = "get_identify_furniture_status" GET_IDENTIFY_GROUND_MATERIAL_STATUS = "get_identify_ground_material_status" GET_LED_STATUS = "get_led_status" GET_LOG_UPLOAD_STATUS = "get_log_upload_status" GET_MAP = "get_map" GET_MAP_BEAUTIFICATION_STATUS = "get_map_beautification_status" GET_MAP_STATUS = "get_map_status" GET_MAP_V1 = "get_map_v1" GET_MAP_V2 = "get_map_v2" GET_MAP_CALIBRATION = "get_map_calibration" # Custom command GET_MOP_MOTOR_STATUS = "get_mop_motor_status" GET_MOP_TEMPLATE_PARAMS_BY_ID = "get_mop_template_params_by_id" GET_MOP_TEMPLATE_PARAMS_SUMMARY = "get_mop_template_params_summary" GET_MULTI_MAP = "get_multi_map" GET_MULTI_MAPS_LIST = "get_multi_maps_list" GET_NETWORK_INFO = "get_network_info" GET_OFFLINE_MAP_STATUS = "get_offline_map_status" GET_PERSIST = "get_persist_map" GET_PROP = "get_prop" GET_RANDOM_PKEY = "get_random_pkey" GET_RECOVER_MAP = "get_recover_map" GET_RECOVER_MAPS = "get_recover_maps" GET_ROOM_MAPPING = "get_room_mapping" GET_SCENES_VALID_TIDS = "get_scenes_valid_tids" GET_SEGMENT_STATUS = "get_segment_status" GET_SERIAL_NUMBER = "get_serial_number" GET_SERVER_TIMER = "get_server_timer" GET_SMART_WASH_PARAMS = "get_smart_wash_params" GET_SOUND_PROGRESS = "get_sound_progress" GET_SOUND_VOLUME = "get_sound_volume" GET_STATUS = "get_status" GET_TESTID = "get_testid" GET_TIMER = "get_timer" GET_TIMER_DETAIL = "get_timer_detail" GET_TIMER_SUMMARY = "get_timer_summary" GET_TIMEZONE = "get_timezone" GET_TURN_SERVER = "get_turn_server" GET_VALLEY_ELECTRICITY_TIMER = "get_valley_electricity_timer" GET_WASH_DEBUG_PARAMS = "get_wash_debug_params" GET_WASH_TOWEL_MODE = "get_wash_towel_mode" GET_WASH_TOWEL_PARAMS = "get_wash_towel_params" GET_WATER_BOX_CUSTOM_MODE = "get_water_box_custom_mode" LOAD_MULTI_MAP = "load_multi_map" MANUAL_BAK_MAP = "manual_bak_map" MANUAL_SEGMENT_MAP = "manual_segment_map" MERGE_SEGMENT = "merge_segment" MOP_MODE = "mop_mode" MOP_TEMPLATE_ID = "mop_template_id" NAME_MULTI_MAP = "name_multi_map" NAME_SEGMENT = "name_segment" PLAY_AUDIO = "play_audio" RECOVER_MAP = "recover_map" RECOVER_MULTI_MAP = "recover_multi_map" RESET_CONSUMABLE = "reset_consumable" RESET_HOMESEC_PASSWORD = "reset_homesec_password" RESET_MAP = "reset_map" RESOLVE_ERROR = "resolve_error" RESUME_SEGMENT_CLEAN = "resume_segment_clean" RESUME_ZONED_CLEAN = "resume_zoned_clean" RETRY_REQUEST = "retry_request" REUNION_SCENES = "reunion_scenes" SAVE_FURNITURES = "save_furnitures" SAVE_MAP = "save_map" SEND_ICE_TO_ROBOT = "send_ice_to_robot" SEND_SDP_TO_ROBOT = "send_sdp_to_robot" SET_AIRDRY_HOURS = "set_airdry_hours" SET_APP_TIMEZONE = "set_app_timezone" SET_AUTO_DELIVERY_CLEANING_FLUID = "set_auto_delivery_cleaning_fluid" SET_CAMERA_STATUS = "set_camera_status" SET_CARPET_AREA = "set_carpet_area" SET_CARPET_CLEAN_MODE = "set_carpet_clean_mode" SET_CARPET_MODE = "set_carpet_mode" SET_CHILD_LOCK_STATUS = "set_child_lock_status" SET_CLEAN_FOLLOW_GROUND_MATERIAL_STATUS = "set_clean_follow_ground_material_status" SET_CLEAN_MOTOR_MODE = "set_clean_motor_mode" SET_CLEAN_SEQUENCE = "set_clean_sequence" SET_CLEAN_REPEAT_TIMES = "set_clean_repeat_times" SET_COLLISION_AVOID_STATUS = "set_collision_avoid_status" SET_CUSTOM_MODE = "set_custom_mode" SET_CUSTOMIZE_CLEAN_MODE = "set_customize_clean_mode" SET_DND_TIMER = "set_dnd_timer" SET_DND_TIMER_ACTIONS = "set_dnd_timer_actions" SET_DUST_COLLECTION_MODE = "set_dust_collection_mode" SET_DUST_COLLECTION_SWITCH_STATUS = "set_dust_collection_switch_status" SET_FAN_MOTOR_WORK_TIMEOUT = "set_fan_motor_work_timeout" SET_FDS_ENDPOINT = "set_fds_endpoint" SET_FLOW_LED_STATUS = "set_flow_led_status" SET_HOMESEC_PASSWORD = "set_homesec_password" SET_IDENTIFY_FURNITURE_STATUS = "set_identify_furniture_status" SET_IDENTIFY_GROUND_MATERIAL_STATUS = "set_identify_ground_material_status" SET_IGNORE_CARPET_ZONE = "set_ignore_carpet_zone" SET_IGNORE_IDENTIFY_AREA = "set_ignore_identify_area" SET_LAB_STATUS = "set_lab_status" SET_LED_STATUS = "set_led_status" SET_MAP_BEAUTIFICATION_STATUS = "set_map_beautification_status" SET_MOP_MODE = "set_mop_mode" SET_MOP_MOTOR_STATUS = "set_mop_motor_status" SET_MOP_TEMPLATE_ID = "set_mop_template_id" SET_OFFLINE_MAP_STATUS = "set_offline_map_status" SET_SCENES_SEGMENTS = "set_scenes_segments" SET_SCENES_ZONES = "set_scenes_zones" SET_SEGMENT_GROUND_MATERIAL = "set_segment_ground_material" SET_SERVER_TIMER = "set_server_timer" SET_SMART_WASH_PARAMS = "set_smart_wash_params" SET_SWITCH_MOP_MODE = "set_switch_map_mode" SET_TIMER = "set_timer" SET_TIMEZONE = "set_timezone" SET_VALLEY_ELECTRICITY_TIMER = "set_valley_electricity_timer" SET_VOICE_CHAT_VOLUME = "set_voice_chat_volume" SET_WASH_DEBUG_PARAMS = "set_wash_debug_params" SET_WASH_TOWEL_MODE = "set_wash_towel_mode" SET_WASH_TOWEL_PARAMS = "set_wash_towel_params" SET_WATER_BOX_CUSTOM_MODE = "set_water_box_custom_mode" SET_WATER_BOX_DISTANCE_OFF = "set_water_box_distance_off" SORT_MOP_TEMPLATE_PARAMS = "sort_mop_template_params" SPLIT_SEGMENT = "split_segment" START_CAMERA_PREVIEW = "start_camera_preview" START_CLEAN = "start_clean" START_EDIT_MAP = "start_edit_map" START_VOICE_CHAT = "start_voice_chat" START_WASH_THEN_CHARGE = "start_wash_then_charge" STOP_CAMERA_PREVIEW = "stop_camera_preview" STOP_FAN_MOTOR_WORK = "stop_fan_motor_work" STOP_GOTO_TARGET = "stop_goto_target" STOP_SEGMENT_CLEAN = "stop_segment_clean" STOP_VOICE_CHAT = "stop_voice_chat" STOP_ZONED_CLEAN = "stop_zoned_clean" SWITCH_VIDEO_QUALITY = "switch_video_quality" SWITCH_WATER_MARK = "switch_water_mark" TEST_SOUND_VOLUME = "test_sound_volume" UPD_SERVER_TIMER = "upd_server_timer" UPD_TIMER = "upd_timer" UPDATE_DOCK = "update_dock" UPDATE_MOP_TEMPLATE_PARAMS = "update_mop_template_params" UPLOAD_DATA_FOR_DEBUG_MODE = "upload_data_for_debug_mode" UPLOAD_PHOTO = "upload_photo" USE_NEW_MAP = "use_new_map" USE_OLD_MAP = "use_old_map" USER_UPLOAD_LOG = "user_upload_log" SET_STRETCH_TAG_STATUS = "set_stretch_tag_status" GET_STRETCH_TAG_STATUS = "get_stretch_tag_status" SET_RIGHT_BRUSH_STRETCH_STATUS = "set_right_brush_stretch_status" GET_RIGHT_BRUSH_STRETCH_STATUS = "get_right_brush_stretch_status" SET_DIRTY_OBJECT_DETECT_STATUS = "set_dirty_object_detect_status" GET_DIRTY_OBJECT_DETECT_STATUS = "get_dirty_object_detect_status" SET_WASH_WATER_TEMPERATURE = "set_wash_water_temperature" GET_WASH_WATER_TEMPERATURE = "get_wash_water_temperature" APP_EMPTY_RINSE_TANK_WATER = "app_empty_rinse_tank_water" SET_PET_SUPPLIES_DEEP_CLEAN_STATUS = "set_pet_supplies_deep_clean_status" GET_PET_SUPPLIES_DEEP_CLEAN_STATUS = "get_pet_supplies_deep_clean_status" SET_AP_MIC_LED_STATUS = "set_ap_mic_led_status" GET_AP_MIC_LED_STATUS = "get_ap_mic_led_status" SET_HANDLE_LEAK_WATER_STATUS = "set_handle_leak_water_status" GET_HANDLE_LEAK_WATER_STATUS = "get_handle_leak_water_status" APP_IGNORE_DIRTY_OBJECTS = "app_ignore_dirty_objects" MATTER_GET_STATUS = "matter.get_status" MATTER_DNLD_KEY = "matter.dnld_key" MATTER_RESET = "matter.reset" SET_GAP_DEEP_CLEAN_STATUS = "set_gap_deep_clean_status" GET_GAP_DEEP_CLEAN_STATUS = "get_gap_deep_clean_status" APP_SET_ROBOT_SETTING = "app_set_robot_setting" APP_GET_ROBOT_SETTING = "app_get_robot_setting" class RoborockB01Methods(StrEnum): """Methods used by the Roborock B01 model.""" GET_PROP = "prop.get" GET_MAP_LIST = "service.get_map_list" UPLOAD_BY_MAPTYPE = "service.upload_by_maptype" SET_PROP = "prop.set" GET_PREFERENCE = "service.get_preference" GET_RECORD_LIST = "service.get_record_list" GET_ORDER = "service.get_order" EVENT_ORDER_LIST_POST = "event.order_list.post" POST_PROP = "prop.post" @dataclass class DockSummary(RoborockBase): dust_collection_mode: DustCollectionMode | None = None wash_towel_mode: WashTowelMode | None = None smart_wash_params: SmartWashParams | None = None @dataclass class DeviceProp(RoborockBase): status: Status = field(default_factory=Status) clean_summary: CleanSummary = field(default_factory=CleanSummary) consumable: Consumable = field(default_factory=Consumable) last_clean_record: CleanRecord | None = None dock_summary: DockSummary | None = None dust_collection_mode_name: str | None = None def __post_init__(self) -> None: if ( self.dock_summary and self.dock_summary.dust_collection_mode is not None and self.dock_summary.dust_collection_mode.mode is not None ): self.dust_collection_mode_name = self.dock_summary.dust_collection_mode.mode.name def update(self, device_prop: DeviceProp) -> None: if device_prop.status: self.status = device_prop.status if device_prop.clean_summary: self.clean_summary = device_prop.clean_summary if device_prop.consumable: self.consumable = device_prop.consumable if device_prop.last_clean_record: self.last_clean_record = device_prop.last_clean_record if device_prop.dock_summary: self.dock_summary = device_prop.dock_summary self.__post_init__()