happyz
/
xiaomi_vacuum_as_data_collector
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bug fix

This commit is contained in:
HappyZ 2019-05-15 12:55:09 -05:00
parent 06f443b30c
commit 7ca01279f9
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libs/vacuum_controller.py
View File

@ -170,7 +170,7 @@ class VacuumController():
.format(i+1, seen_device[0], seen_device[1], seen_device[2])
)
try:
selected = input('Please select one by typing number (1-{}): '.format(len(ips)))
selected = input('Please select one by typing number (1-{}): '.format(len(seen_devices)))
self.set_ip(seen_devices[int(selected)-1][0])
self.vacuum.token = seen_devices[int(selected)-1][2]
self.set_token(codecs.encode(seen_devices[int(selected)-1][2], 'hex').decode())
@ -271,9 +271,26 @@ class VacuumController():
self._control(["move", "auto"])
self._control(["fanspeed", "1"]) # set to lowest fan speed
while 1:
status = self.vacuum.status()
try:
status = self.vacuum.status()
except KeyboardInterrupt:
break
except BaseException as e:
print("Err: {}".format(e))
print("Wait for a bit..")
time.sleep(10)
continue
if status.error_code > 0 or status.state_code == 12:
print("Err: {}".format(status.error))
try:
# try to pause
print("Trying to pause due to the error")
self.vacuum.pause()
except BaseException as e:
print("Err: cannot pause due to {}".format(e))
print("Wait for 30s and restart discovering mode")
self.discover()
continue
break
if status.state_code == 6:
print("Returning home.. stopping..")