github.com/eufy_robovac
1
0
Fork 1
mirror of https://github.com/mitchellrj/eufy_robovac.git synced 2025-11-25 20:52:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix attributes

Browse Source
This commit is contained in:
Richard Mitchell
2019-04-18 13:03:58 +01:00
parent 53248a9013
commit 8e97b42616
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
eufy_robovac/vacuum.py
View File

@@ -21,10 +21,10 @@ FAN_SPEED_STANDARD = 'Standard'
FAN_SPEED_BOOST_IQ = 'Boost IQ' FAN_SPEED_BOOST_IQ = 'Boost IQ'
FAN_SPEED_MAX = 'Max' FAN_SPEED_MAX = 'Max'
FAN_SPEEDS = { FAN_SPEEDS = {
robovac.Robovac.CleanSpeed.NO_SUCTION: FAN_SPEED_OFF, robovac.CleanSpeed.NO_SUCTION: FAN_SPEED_OFF,
robovac.Robovac.CleanSpeed.STANDARD: FAN_SPEED_STANDARD, robovac.CleanSpeed.STANDARD: FAN_SPEED_STANDARD,
robovac.Robovac.CleanSpeed.BOOST_IQ: FAN_SPEED_BOOST_IQ, robovac.CleanSpeed.BOOST_IQ: FAN_SPEED_BOOST_IQ,
robovac.Robovac.CleanSpeed.MAX: FAN_SPEED_MAX, robovac.CleanSpeed.MAX: FAN_SPEED_MAX,
} }
@@ -87,7 +87,7 @@ class EufyVacuum(VacuumDevice):
@property @property
def is_on(self): def is_on(self):
"""Return true if device is on.""" """Return true if device is on."""
return self.robovac.work_status == robovac.Robovac.WorkStatus.RUNNING return self.robovac.work_status == robovac.WorkStatus.RUNNING
@property @property
def supported_features(self): def supported_features(self):
@@ -113,21 +113,21 @@ class EufyVacuum(VacuumDevice):
@property @property
def status(self): def status(self):
"""Return the status of the vacuum cleaner.""" """Return the status of the vacuum cleaner."""
if self.robovac.error_code != robovac.Robovac.ErrorCode.NO_ERROR: if self.robovac.error_code != robovac.ErrorCode.NO_ERROR:
return STATE_ERROR return STATE_ERROR
elif self.robovac.work_status == robovac.Robovac.WorkStatus.RECHARGE: elif self.robovac.work_status == robovac.WorkStatus.RECHARGE:
return STATE_ERROR return STATE_ERROR
elif self.robovac.go_home: elif self.robovac.go_home:
return STATE_RETURNING return STATE_RETURNING
elif self.robovac.work_status == robovac.Robovac.WorkStatus.RUNNING: elif self.robovac.work_status == robovac.WorkStatus.RUNNING:
return STATE_CLEANING return STATE_CLEANING
elif self.robovac.work_status == robovac.Robovac.WorkStatus.CHARGING: elif self.robovac.work_status == robovac.WorkStatus.CHARGING:
return STATE_DOCKED return STATE_DOCKED
elif self.robovac.work_status == robovac.Robovac.WorkStatus.RECHARGING: elif self.robovac.work_status == robovac.WorkStatus.RECHARGING:
return STATE_DOCKED return STATE_DOCKED
elif self.robovac.work_status == robovac.Robovac.WorkStatus.SLEEPING: elif self.robovac.work_status == robovac.WorkStatus.SLEEPING:
return STATE_DOCKED return STATE_DOCKED
elif self.robovac.work_status == robovac.Robovac.WorkStatus.STAND_BY: elif self.robovac.work_status == robovac.WorkStatus.STAND_BY:
return STATE_IDLE return STATE_IDLE
@property @property
@@ -141,7 +141,7 @@ class EufyVacuum(VacuumDevice):
async def async_clean_spot(self, **kwargs): async def async_clean_spot(self, **kwargs):
"""Perform a spot clean-up.""" """Perform a spot clean-up."""
await self.robovac.async_set_work_mode(robovac.Robovac.WorkMode.SPOT) await self.robovac.async_set_work_mode(robovac.WorkMode.SPOT)
async def async_locate(self, **kwargs): async def async_locate(self, **kwargs):
"""Locate the vacuum cleaner.""" """Locate the vacuum cleaner."""
@@ -154,7 +154,7 @@ class EufyVacuum(VacuumDevice):
async def async_turn_on(self, **kwargs): async def async_turn_on(self, **kwargs):
"""Turn the vacuum on.""" """Turn the vacuum on."""
await self.robovac.async_set_work_mode(robovac.Robovac.WorkMode.AUTO) await self.robovac.async_set_work_mode(robovac.WorkMode.AUTO)
async def async_turn_off(self, **kwargs): async def async_turn_off(self, **kwargs):
"""Turn the vacuum off and return to home.""" """Turn the vacuum off and return to home."""