blob: 75f368c9852f54e68d7b5085670b98ddafd39fa7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
from parrot_zik import resource_manager
from parrot_zik.model.base import BatteryStates
RECONNECT_FREQUENCY = 5
class ParrotZikBaseInterface(object):
def __init__(self, indicator):
self.indicator = indicator
self.parrot = None
self.battery_level = ""
self.battery_state = ""
def activate(self, manager):
self.parrot = self.parrot_class(manager)
self.indicator.info({"info": "Connected to: " + self.parrot.friendly_name})
self.firmware_version = self.parrot.version
self.auto_connection = self.parrot.auto_connect
self.indicator.active_interface = self
@property
def parrot_class(self):
raise NotImplementedError
def deactivate(self):
self.parrot = None
self.battery_level = ""
self.battery_mode = ""
self.firmware_version = ""
self.indicator.active_interface = None
self.indicator.info({"error": "Lost Connection"})
self.indicator.reconnect(RECONNECT_FREQUENCY)
def read_auto_connection(self):
"""
Return auto connection status.
"""
try:
data = {"data": {"auto_connection": self.parrot.auto_connect}}
self.indicator.info(data)
return data
except resource_manager.DeviceDisconnected:
self.deactivate()
def toggle_auto_connection(self, auto_connection):
"""
Set auto connection mode. Must be 'true' or 'false'.
"""
try:
self.parrot.auto_connect = auto_connection
return self.read_auto_connection()
except resource_manager.DeviceDisconnected:
self.deactivate()
def read_battery(self):
"""
Return battery level and battery state
"""
try:
self.parrot.refresh_battery()
battery_level = self.parrot.battery_level
battery_state = self.parrot.battery_state
except AssertionError as e:
print(e)
except resource_manager.DeviceDisconnected:
self.deactivate()
else:
return {"data": {"state": BatteryStates.representation[battery_state], "level": battery_level}}
|
