diff --git a/wideq/refrigerator.py b/wideq/refrigerator.py index f10b4de..9d8f133 100644 --- a/wideq/refrigerator.py +++ b/wideq/refrigerator.py @@ -64,24 +64,29 @@ class DoorOpenState(enum.Enum): class RefrigeratorDevice(Device): """A higher-level interface for a refrigerator.""" - def set_temp_refrigerator_c(self, temp_refrigerator): + def set_temp_refrigerator_c(self, temp): - temp_refrigerator_value = self.model.enum_value('TempRefrigerator', str(temp_refrigerator)) - # '{"RETM":"{{TempRefrigerator}}", "REFT":"{{TempFreezer}}", "REIP":"{{IcePlus}}", "REEF":"{{EcoFriendly}}" }' - self._set_control('RETM', temp_refrigerator_value) + value = self.model.enum_value('TempRefrigerator', str(temp)) + # { + # "RETM":"{{TempRefrigerator}}", + # "REFT":"{{TempFreezer}}", + # "REIP":"{{IcePlus}}", + # "REEF":"{{EcoFriendly}}" + # } + self._set_control('RETM', value) - def set_temp_freezer_c(self, temp_freezer): + def set_temp_freezer_c(self, temp): - temp_freezer_value = self.model.enum_value('TempFreezer', str(temp_freezer)) - self._set_control('REFT', temp_freezer_value) + value = self.model.enum_value('TempFreezer', str(temp)) + self._set_control('REFT', value) def poll(self) -> Optional['RefrigeratorStatus']: """Poll the device's current state. Monitoring must be started first with `monitor_start`. - :returns: Either a `RefrigeratorStatus` instance or `None` if the status is - not yet available. + :returns: Either a `RefrigeratorStatus` instance or `None` if the + status is not yet available. """ # Abort if monitoring has not started yet. if not hasattr(self, 'mon'): @@ -108,28 +113,33 @@ class RefrigeratorStatus(object): @property def temp_refrigerator_c(self): - return int(lookup_enum('TempRefrigerator', self.data, self.refrigerator)) + temp = lookup_enum('TempRefrigerator', self.data, self.refrigerator) + return int(temp) @property def temp_freezer_c(self): - return int(lookup_enum('TempFreezer', self.data, self.refrigerator)) + temp = lookup_enum('TempFreezer', self.data, self.refrigerator) + return int(temp) @property def ice_plus_status(self): - return IcePlus(lookup_enum('IcePlus', self.data, self.refrigerator)) + status = lookup_enum('IcePlus', self.data, self.refrigerator) + return IcePlus(status) @property def fresh_air_filter_status(self): - return FreshAirFilter(lookup_enum('FreshAirFilter', self.data, self.refrigerator)) + status = lookup_enum('FreshAirFilter', self.data, self.refrigerator) + return FreshAirFilter(status) @property def energy_saving_mode(self): - return SmartSavingMode(lookup_enum('SmartSavingMode', self.data, self.refrigerator)) + mode = lookup_enum('SmartSavingMode', self.data, self.refrigerator) + return SmartSavingMode(mode) @property def door_opened(self): - door = DoorOpenState(lookup_enum('DoorOpenState', self.data, self.refrigerator)) - return door == DoorOpenState.OPEN + door = lookup_enum('DoorOpenState', self.data, self.refrigerator) + return DoorOpenState(door) == DoorOpenState.OPEN @property def temp_unit(self): @@ -137,13 +147,15 @@ class RefrigeratorStatus(object): @property def energy_saving_enabled(self): - status = SmartSavingModeStatus(lookup_enum('SmartSavingModeStatus', self.data, self.refrigerator)) - return status == SmartSavingModeStatus.ON + mode = lookup_enum( + 'SmartSavingModeStatus', self.data, self.refrigerator + ) + return SmartSavingModeStatus(mode) == SmartSavingModeStatus.ON @property def locked(self): - status = LockingStatus(lookup_enum('LockingStatus', self.data, self.refrigerator)) - return status == LockingStatus.LOCK + status = lookup_enum('LockingStatus', self.data, self.refrigerator) + return LockingStatus(status) == LockingStatus.LOCK @property def active_saving_status(self): @@ -151,8 +163,8 @@ class RefrigeratorStatus(object): @property def eco_enabled(self): - eco = EcoFriendly(lookup_enum('EcoFriendly', self.data, self.refrigerator)) - return eco == EcoFriendly.ON + eco = lookup_enum('EcoFriendly', self.data, self.refrigerator) + return EcoFriendly(eco) == EcoFriendly.ON @property def water_filter_used_month(self):