118 lines
4.9 KiB
Python
118 lines
4.9 KiB
Python
|
import logging
|
||
|
|
from .message import (
|
||
|
|
MessageQuery,
|
||
|
|
MessageECResponse
|
||
|
|
)
|
||
|
|
try:
|
||
|
|
from enum import StrEnum
|
||
|
|
except ImportError:
|
||
|
|
from ...backports.enum import StrEnum
|
||
|
|
from ...core.device import MiedaDevice
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class DeviceAttributes(StrEnum):
|
||
|
|
cooking = "cooking"
|
||
|
|
mode = "mode"
|
||
|
|
time_remaining = "time_remaining"
|
||
|
|
keep_warm_time = "keep_warm_time"
|
||
|
|
top_temperature = "top_temperature"
|
||
|
|
bottom_temperature = "bottom_temperature"
|
||
|
|
progress = "progress"
|
||
|
|
with_pressure = "with_pressure"
|
||
|
|
|
||
|
|
|
||
|
|
class MideaECDevice(MiedaDevice):
|
||
|
|
_mode_list = [
|
||
|
|
"smart", "reserve", "cook_rice", "fast_cook_rice", "standard_cook_rice",
|
||
|
|
"gruel", "cook_congee", "stew_soup", "stewing", "heat_rice", "make_cake",
|
||
|
|
"yoghourt", "soup_rice", "coarse_rice", "five_ceeals_rice", "eight_treasures_rice",
|
||
|
|
"crispy_rice", "shelled_rice", "eight_treasures_congee", "infant_congee",
|
||
|
|
"older_rice", "rice_soup", "rice_paste", "egg_custard", "warm_milk",
|
||
|
|
"hot_spring_egg", "millet_congee", "firewood_rice", "few_rice",
|
||
|
|
"red_potato", "corn", "quick_freeze_bun", "steam_ribs", "steam_egg",
|
||
|
|
"coarse_congee", "steep_rice", "appetizing_congee", "corn_congee",
|
||
|
|
"sprout_rice", "luscious_rice", "luscious_boiled", "fast_rice", "fast_boil",
|
||
|
|
"bean_rice_congee", "fast_congee", "baby_congee", "cook_soup", "congee_coup",
|
||
|
|
"steam_corn", "steam_red_potato", "boil_congee", "delicious_steam", "boil_egg",
|
||
|
|
"rice_wine", "fruit_vegetable_paste", "vegetable_porridge", "pork_porridge",
|
||
|
|
"fragrant_rice", "assorte_rice", "steame_fish", "baby_rice", "essence_rice",
|
||
|
|
"fragrant_dense_congee", "one_two_cook", "original_steame", "hot_fast_rice",
|
||
|
|
"online_celebrity_rice", "sushi_rice", "stone_bowl_rice", "no_water_treat",
|
||
|
|
"keep_fresh", "low_sugar_rice", "black_buckwheat_rice", "resveratrol_rice",
|
||
|
|
"yellow_wheat_rice", "green_buckwheat_rice", "roughage_rice", "millet_mixed_rice",
|
||
|
|
"iron_pan_rice", "olla_pan_rice", "vegetable_rice", "baby_side", "regimen_congee",
|
||
|
|
"earthen_pot_congee", "regimen_soup", "pottery_jar_soup", "canton_soup",
|
||
|
|
"nutrition_stew", "northeast_stew", "uncap_boil", "trichromatic_coarse_grain",
|
||
|
|
"four_color_vegetables", "egg", "chop",
|
||
|
|
] + ["unknown"] * 98 + ["clean"] + ["unknown"] * 5 + ["keep_warm", "diy"]
|
||
|
|
_progress = ["Idle", "Cooking", "Delay", "Keep-warm",
|
||
|
|
"Lid-open", "Relieving", "Keep-pressure",
|
||
|
|
"Relieving", "Cooking", "Relieving", "Lid-open"]
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
name: str,
|
||
|
|
device_id: int,
|
||
|
|
ip_address: str,
|
||
|
|
port: int,
|
||
|
|
token: str,
|
||
|
|
key: str,
|
||
|
|
protocol: int,
|
||
|
|
model: str,
|
||
|
|
customize: str
|
||
|
|
):
|
||
|
|
super().__init__(
|
||
|
|
name=name,
|
||
|
|
device_id=device_id,
|
||
|
|
device_type=0xEC,
|
||
|
|
ip_address=ip_address,
|
||
|
|
port=port,
|
||
|
|
token=token,
|
||
|
|
key=key,
|
||
|
|
protocol=protocol,
|
||
|
|
model=model,
|
||
|
|
attributes={
|
||
|
|
DeviceAttributes.cooking: False,
|
||
|
|
DeviceAttributes.mode: 0,
|
||
|
|
DeviceAttributes.time_remaining: None,
|
||
|
|
DeviceAttributes.top_temperature: None,
|
||
|
|
DeviceAttributes.bottom_temperature: None,
|
||
|
|
DeviceAttributes.keep_warm_time: None,
|
||
|
|
DeviceAttributes.progress: "Unknown",
|
||
|
|
DeviceAttributes.with_pressure: None
|
||
|
|
})
|
||
|
|
|
||
|
|
def build_query(self):
|
||
|
|
return [MessageQuery(self._device_protocol_version)]
|
||
|
|
|
||
|
|
def process_message(self, msg):
|
||
|
|
message = MessageECResponse(msg)
|
||
|
|
_LOGGER.debug(f"[{self.device_id}] Received: {message}")
|
||
|
|
new_status = {}
|
||
|
|
for status in self._attributes.keys():
|
||
|
|
if hasattr(message, str(status)):
|
||
|
|
value = getattr(message, str(status))
|
||
|
|
if status == DeviceAttributes.progress:
|
||
|
|
if value < len(MideaECDevice._progress):
|
||
|
|
self._attributes[status] = MideaECDevice._progress[getattr(message, str(status))]
|
||
|
|
else:
|
||
|
|
self._attributes[status] = "Unknown"
|
||
|
|
elif status == DeviceAttributes.mode:
|
||
|
|
if value < len(MideaECDevice._mode_list):
|
||
|
|
self._attributes[status] = MideaECDevice._mode_list[value]
|
||
|
|
else:
|
||
|
|
self._attributes[status] = "Cloud"
|
||
|
|
else:
|
||
|
|
self._attributes[status] = value
|
||
|
|
new_status[str(status)] = self._attributes[status]
|
||
|
|
return new_status
|
||
|
|
|
||
|
|
def set_attribute(self, attr, value):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class MideaAppliance(MideaECDevice):
|
||
|
|
pass
|