class ParrotZikIndicator(SysIndicator):
def __init__(self):
self.menu = Menu()
self.info_item = MenuItem("Parrot Zik Not connected",
None, sensitive=False)
self.menu.append(self.info_item)
self.version_1_interface = ParrotZikVersion1Interface(self)
self.version_2_interface = ParrotZikVersion2Interface(self)
self.quit_item = MenuItem("Quit", self.quit, checkitem=True)
self.menu.append(self.quit_item)
SysIndicator.__init__(self, icon="zik-audio-headset", menu=self.menu)
self.active_interface = None
@repeat
def reconnect(self):
if self.active_interface:
self.reconnect.stop()
else:
self.info("Trying to connect")
try:
manager = bluetooth_paired_devices.connect()
except bluetooth_paired_devices.BluetoothIsNotOn:
self.info("Bluetooth is turned off")
except bluetooth_paired_devices.DeviceNotConnected:
self.info("Parrot Zik Not connected")
except bluetooth_paired_devices.ConnectionFailure:
self.info("Failed to connect")
else:
if manager.api_version.startswith('1'):
interface = self.version_1_interface
else:
interface = self.version_2_interface
try:
interface.activate(manager)
except resource_manager.DeviceDisconnected:
interface.deactivate()
else:
self.autorefresh(self)
self.autorefresh.start(self, REFRESH_FREQUENCY)
self.reconnect.stop()
def info(self, message):
self.info_item.set_label(message)
print(message)
@repeat
def autorefresh(self):
if self.active_interface:
self.active_interface.refresh()
else:
self.reconnect.start(self, RECONNECT_FREQUENCY)
self.autorefresh.stop()
@classmethod
def main(cls):
try:
indicator = cls()
cls.reconnect.start(indicator, RECONNECT_FREQUENCY)
super(ParrotZikIndicator, cls).main()
except KeyboardInterrupt:
pass
class ParrotZikBaseInterface(object):
def __init__(self, indicator):
self.indicator = indicator
self.parrot = None
self.battery_level = MenuItem("Battery Level:",
None,
sensitive=False,
visible=False)
self.battery_state = MenuItem("Battery State:",
None,
sensitive=False,
visible=False)
self.firmware_version = MenuItem("Firmware Version:",
None,
sensitive=False,
visible=False)
self.settings = MenuItem("Settings", None, visible=False)
self.settings_submenu = Menu()
self.settings.set_submenu(self.settings_submenu)
self.auto_connection = MenuItem("Auto Connection",
self.toggle_auto_connection,
checkitem=True)
self.settings_submenu.append(self.auto_connection)
self.indicator.menu.append(self.battery_level)
self.indicator.menu.append(self.battery_state)
self.indicator.menu.append(self.firmware_version)
self.indicator.menu.append(self.settings)
def activate(self, manager):
self.parrot = self.parrot_class(manager)
self.read_battery()
self.indicator.info("Connected to: " + self.parrot.friendly_name)
self.firmware_version.set_label("Firmware version: " +
self.parrot.version)
self.auto_connection.set_active(self.parrot.auto_connect)
self.battery_level.show()
self.battery_state.show()
self.firmware_version.show()
self.settings.show()
self.indicator.active_interface = self
@property
def parrot_class(self):
raise NotImplementedError
def deactivate(self):
self.parrot = None
self.battery_level.hide()
self.battery_state.hide()
self.firmware_version.hide()
self.settings.hide()
self.indicator.menu.reposition()
self.indicator.active_interface = None
self.indicator.setIcon("zik-audio-headset")
self.indicator.info('Lost Connection')
self.indicator.reconnect.start(self.indicator, RECONNECT_FREQUENCY)
def toggle_auto_connection(self, widget):
try:
self.parrot.auto_connect = self.auto_connection.get_active()
self.auto_connection.set_active(self.parrot.auto_connect)
except resource_manager.DeviceDisconnected:
self.deactivate()
def refresh(self):
self.read_battery()
def read_battery(self):
try:
self.parrot.refresh_battery()
battery_level = self.parrot.battery_level
battery_state = self.parrot.battery_state
except resource_manager.DeviceDisconnected:
self.deactivate()
else:
if battery_state == BatteryStates.CHARGING:
self.indicator.setIcon("zik-battery-charging")
elif battery_level > 80:
self.indicator.setIcon("zik-battery-100")
elif battery_level > 60:
self.indicator.setIcon("zik-battery-080")
elif battery_level > 40:
self.indicator.setIcon("zik-battery-060")
elif battery_level > 20:
self.indicator.setIcon("zik-battery-040")
else:
self.indicator.setIcon("zik-battery-low")
self.battery_state.set_label(
"State: " + BatteryStates.representation[battery_state])
self.battery_level.set_label("Battery Level: " +
str(battery_level))
class ParrotZikBaseInterface(object):
def __init__(self, indicator):
self.indicator = indicator
self.parrot = None
self.battery_level = MenuItem("Battery Level:", None, sensitive=False,
visible=False)
self.battery_state = MenuItem("Battery State:", None, sensitive=False,
visible=False)
self.firmware_version = MenuItem("Firmware Version:", None,
sensitive=False, visible=False)
self.settings = MenuItem("Settings", None, visible=False)
self.settings_submenu = Menu()
self.settings.set_submenu(self.settings_submenu)
self.auto_connection = MenuItem("Auto Connection", self.toggle_auto_connection,
checkitem=True)
self.settings_submenu.append(self.auto_connection)
self.indicator.menu.append(self.battery_level)
self.indicator.menu.append(self.battery_state)
self.indicator.menu.append(self.firmware_version)
self.indicator.menu.append(self.settings)
def activate(self, manager):
self.parrot = self.parrot_class(manager)
self.read_battery()
self.indicator.info("Connected to: " + self.parrot.friendly_name)
self.firmware_version.set_label(
"Firmware version: " + self.parrot.version)
self.auto_connection.set_active(self.parrot.auto_connect)
self.battery_level.show()
self.battery_state.show()
self.firmware_version.show()
self.settings.show()
self.indicator.active_interface = self
@property
def parrot_class(self):
raise NotImplementedError
def deactivate(self):
self.parrot = None
self.battery_level.hide()
self.battery_state.hide()
self.firmware_version.hide()
self.settings.hide()
self.indicator.menu.reposition()
self.indicator.active_interface = None
self.indicator.setIcon("zik-audio-headset")
self.indicator.info('Lost Connection')
self.indicator.reconnect.start(self.indicator, RECONNECT_FREQUENCY)
def toggle_auto_connection(self, widget):
try:
self.parrot.auto_connect = self.auto_connection.get_active()
self.auto_connection.set_active(self.parrot.auto_connect)
except resource_manager.DeviceDisconnected:
self.deactivate()
def refresh(self):
self.read_battery()
def read_battery(self):
try:
self.parrot.refresh_battery()
battery_level = self.parrot.battery_level
battery_state = self.parrot.battery_state
except resource_manager.DeviceDisconnected:
self.deactivate()
else:
if battery_state == BatteryStates.CHARGING:
self.indicator.setIcon("zik-battery-charging")
elif battery_level > 80:
self.indicator.setIcon("zik-battery-100")
elif battery_level > 60:
self.indicator.setIcon("zik-battery-080")
elif battery_level > 40:
self.indicator.setIcon("zik-battery-060")
elif battery_level > 20:
self.indicator.setIcon("zik-battery-040")
elif battery_level > 5:
self.indicator.setIcon("zik-battery-020")
else:
self.indicator.setIcon("zik-battery-low")
self.battery_state.set_label(
"State: " + BatteryStates.representation[battery_state])
self.battery_level.set_label(
"Battery Level: " + str(battery_level))
class ParrotZikBaseInterface(object):
def __init__(self, indicator):
self.indicator = indicator
self.parrot = None
self.battery_level = MenuItem("Battery Level:", None, sensitive=False,
visible=False)
self.battery_state = MenuItem("Battery State:", None, sensitive=False,
visible=False)
self.firmware_version = MenuItem("Firmware Version:", None,
sensitive=False, visible=False)
self.settings = MenuItem("Settings", None, visible=False)
self.settings_submenu = Menu()
self.settings.set_submenu(self.settings_submenu)
self.auto_connection = MenuItem("Auto Connection", self.toggle_auto_connection,
checkitem=True)
self.settings_submenu.append(self.auto_connection)
self.indicator.menu.append(self.battery_level)
self.indicator.menu.append(self.battery_state)
self.indicator.menu.append(self.firmware_version)
self.indicator.menu.append(self.settings)
def activate(self, manager):
self.parrot = self.parrot_class(manager)
self.read_battery()
self.indicator.info("Connected to: " + self.parrot.friendly_name)
self.firmware_version.set_label(
"Firmware version: " + self.parrot.version)
self.auto_connection.set_active(self.parrot.auto_connect)
self.battery_level.show()
self.battery_state.show()
self.firmware_version.show()
self.settings.show()
self.indicator.active_interface = self
@property
def parrot_class(self):
raise NotImplementedError
def deactivate(self):
self.parrot = None
self.battery_level.hide()
self.battery_state.hide()
self.firmware_version.hide()
self.settings.hide()
self.indicator.menu.reposition()
self.indicator.active_interface = None
self.indicator.setIcon("zik-audio-headset")
self.indicator.info('Lost Connection')
self.indicator.reconnect.start(self.indicator, RECONNECT_FREQUENCY)
def toggle_auto_connection(self, widget):
try:
self.parrot.auto_connect = self.auto_connection.get_active()
self.auto_connection.set_active(self.parrot.auto_connect)
except resource_manager.DeviceDisconnected:
self.deactivate()
def refresh(self):
self.read_battery()
def read_battery(self):
try:
self.parrot.refresh_battery()
battery_level = self.parrot.battery_level
battery_state = self.parrot.battery_state
except resource_manager.DeviceDisconnected:
self.deactivate()
else:
icon_battery_level = int(battery_level/20)*20
icon_name = "zik-battery-" + format(icon_battery_level, '03')
if True:
icon_name += "-headset"
if battery_state == 'charging':
icon_name += "-charging"
elif battery_state == 'charged':
icon_name += "-charging"
if self.parrot.version.startswith("2"):
from parrot_zik.model.version2 import NoiseControlTypes
if self.parrot.noise_control in [NoiseControlTypes.NOISE_CONTROL_ON,NoiseControlTypes.NOISE_CONTROL_MAX]:
icon_name += "-nac"
if self.parrot.version.startswith("1"):
if self.parrot.cancel_noise:
icon_name += "-nac"
self.indicator.setIcon(icon_name)
self.battery_state.set_label(
"State: " + BatteryStates.representation[battery_state])
self.battery_level.set_label(
"Battery Level: " + str(battery_level))
