aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMarek Siarkowicz <mareksiarkowicz@gmail.com>2015-06-12 19:03:25 +0200
committerMarek Siarkowicz <mareksiarkowicz@gmail.com>2015-06-12 19:03:25 +0200
commit215e013edafbf2688cf5dc1742716be919502ee5 (patch)
tree67a1230230df02689dc4aba63010ad6d1d82f67e
parent0d692102cd94f70e78cf036cb9f64c485ea8b6a5 (diff)
downloadpyParrotZikTCP-215e013edafbf2688cf5dc1742716be919502ee5.tar.xz
pyParrotZikTCP-215e013edafbf2688cf5dc1742716be919502ee5.zip
Allow adding different zik version interfaces.
Diffstat (limited to '')
-rwxr-xr-xParrotZikTray232
-rw-r--r--SysIndicator.py24
2 files changed, 160 insertions, 96 deletions
diff --git a/ParrotZikTray b/ParrotZikTray
index 1f934ea..934062c 100755
--- a/ParrotZikTray
+++ b/ParrotZikTray
@@ -1,12 +1,33 @@
#!/usr/bin/env python
-import re
import ParrotZik
import BluetoothPairedDevices
from SysIndicator import *
-# seconds
-UPDATE_FREQUENCY = 30
+UPDATE_FREQUENCY = 1000
+RECONNECT_FREQUENCY = 5000
+
+class repeat(object):
+ def __init__(self, f):
+ self.f = f
+ self.id = None
+
+ def start(self, cls, frequency):
+ if not self.id:
+ def run():
+ self.f(cls)
+ return True
+
+ self.id = gtk.timeout_add(frequency, run)
+
+ def stop(self):
+ if self.id:
+ gtk.timeout_remove(self.id)
+ self.id = None
+
+
+class Stop(Exception):
+ pass
class ParrotZikIndicator(SysIndicator):
@@ -18,91 +39,128 @@ class ParrotZikIndicator(SysIndicator):
None, sensitive=False)
self.menu.append(self.info_item)
- self.battery_level = MenuItem("Battery Level:",
- None, sensitive=False)
- self.menu.append(self.battery_level)
-
- self.battery_state = MenuItem("Battery State:",
- None, sensitive=False)
- self.menu.append(self.battery_state)
-
- self.firmware_version = MenuItem("Firmware Version:",
- None, sensitive=False)
- self.menu.append(self.firmware_version)
-
- self.check = MenuItem("Noise Cancellation", self.toggleANC,
- sensitive=False, checkitem=True)
- self.menu.append(self.check)
-
- self.check2 = MenuItem("Auto Connection", self.toggleAuto,
- sensitive=False, checkitem=True)
- self.menu.append(self.check2)
-
- self.check3 = MenuItem("Lou Reed Mode", self.toggleLouReedMode,
- sensitive=False, checkitem=True)
- self.menu.append(self.check3)
-
- self.check4 = MenuItem(
- "Concert Hall Mode", self.toggleParrotConcertHall,
- sensitive=False, checkitem=True)
- self.menu.append(self.check4)
-
- self.quit = MenuItem("Quit", sys.exit, sensitive=True, checkitem=True)
+ self.version_1_interface = ParrotZikVersion1Interface(self)
+ self.quit = MenuItem("Quit", gtk.main_quit, checkitem=True)
self.menu.append(self.quit)
SysIndicator.__init__(self, icon="zik-audio-headset", menu=self.menu)
- self.connected = False
- self.p = re.compile(
- '90:03:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}'
- )
+ self.active_interface = None
+ @repeat
def EstablishConnection(self):
- if self.connected:
- if not self.parrot.sock:
+ if self.active_interface:
+ if not self.active_interface.parrot.sock:
print "Lost connection"
- self.connected = False
+ self.active_interface.deactivate()
else:
- print "Connection already established"
+ self.EstablishConnection.stop()
else:
mac = BluetoothPairedDevices.ParrotZikMac()
if mac:
- self.parrot = ParrotZik.ParrotZik(mac)
- if not self.parrot.sock:
+ parrot = ParrotZik.ParrotZik(mac)
+ if parrot.sock:
+ self.version_1_interface.activate(parrot)
+ self.CheckBattery.start(self, UPDATE_FREQUENCY)
+ self.EstablishConnection.stop()
+ print "Connected"
+ else:
print "Failed to connect to Parrot Zik %s" % mac
- return False
-
- self.connected = True
- self.name = self.parrot.friendly_name
- self.version = self.parrot.version
-
- self.check.set_sensitive(True)
- self.check2.set_sensitive(True)
- self.check3.set_sensitive(True)
- self.check4.set_sensitive(True)
- if self.parrot.noise_cancel == "true":
- self.check.set_active(True)
- else:
- self.check.set_active(False)
+ @repeat
+ def CheckBattery(self):
+ if self.active_interface:
+ self.active_interface.CheckBattery()
+ else:
+ self.EstablishConnection.start(self, RECONNECT_FREQUENCY)
+ self.CheckBattery.stop()
- if self.parrot.auto_connect == "true":
- self.check2.set_active(True)
- else:
- self.check2.set_active(False)
+ def main(self):
+ self.EstablishConnection.start(self, RECONNECT_FREQUENCY)
+ SysIndicator.main(self)
+
+class ParrotZikVersion1Interface:
+ def __init__(self, indicator):
+ self.indicator = indicator
+ 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.check = MenuItem("Noise Cancellation", self.toggleANC,
+ checkitem=True, visible=False)
+ self.check2 = MenuItem("Auto Connection", self.toggleAuto,
+ checkitem=True, visible=False)
+ self.check3 = MenuItem("Lou Reed Mode", self.toggleLouReedMode,
+ checkitem=True, visible=False)
+ self.check4 = MenuItem(
+ "Concert Hall Mode", self.toggleParrotConcertHall,
+ checkitem=True, visible=False)
+ 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.check)
+ self.indicator.menu.append(self.check2)
+ self.indicator.menu.append(self.check3)
+ self.indicator.menu.append(self.check4)
+
+ @property
+ def connected(self):
+ if self.parrot:
+ return self.parrot.sock
+ else:
+ return False
+
+ def activate(self, parrot):
+ self.parrot = parrot
+ self.battery_level.show()
+ self.battery_state.show()
+ self.firmware_version.show()
+ self.check.show()
+ self.check2.show()
+ self.check3.show()
+ self.check4.show()
+
+ self.name = self.parrot.friendly_name
+ self.version = self.parrot.version
+
+ if self.parrot.noise_cancel == "true":
+ self.check.set_active(True)
+ else:
+ self.check.set_active(False)
- if self.parrot.lou_reed_mode == "true":
- self.check3.set_active(True)
- else:
- self.check3.set_active(False)
+ if self.parrot.auto_connect == "true":
+ self.check2.set_active(True)
+ else:
+ self.check2.set_active(False)
- if self.parrot.concert_hall == "true":
- self.check4.set_active(True)
- else:
- self.check4.set_active(False)
+ if self.parrot.lou_reed_mode == "true":
+ self.check3.set_active(True)
+ else:
+ self.check3.set_active(False)
- self.CheckBattery()
- return True
+ if self.parrot.concert_hall == "true":
+ self.check4.set_active(True)
+ else:
+ self.check4.set_active(False)
+
+ self.indicator.menu.reposition()
+ self.indicator.active_interface = self
+
+ def deactivate(self):
+ self.parrot = None
+ self.battery_level.hide()
+ self.battery_state.hide()
+ self.firmware_version.hide()
+ self.check.hide()
+ self.check2.hide()
+ self.check3.hide()
+ self.check4.hide()
+ self.indicator.setIcon("zik-audio-headset")
+ self.indicator.info_item.set_label("Parrot Zik Not connected..")
+ self.indicator.menu.reposition()
+ self.indicator.active_interface = None
def toggleANC(self, widget):
if self.connected:
@@ -126,7 +184,7 @@ class ParrotZikIndicator(SysIndicator):
else:
self.parrot.lou_reed_mode = "false"
self.check4.set_sensitive(True)
-
+
def toggleParrotConcertHall(self, widget):
if self.connected:
if self.check4.get_active():
@@ -138,48 +196,36 @@ class ParrotZikIndicator(SysIndicator):
if self.connected:
print "Updating battery"
self.batteryLevel = int(self.parrot.battery_level)
-
+
if self.parrot.BatteryCharging:
self.batteryLevel = "Charging"
- self.setIcon("zik-battery-charging")
+ self.indicator.setIcon("zik-battery-charging")
self.batteryLevel = "Unknown"
self.batteryState = "Charging"
elif self.batteryLevel > 80:
- self.setIcon("zik-battery-100")
+ self.indicator.setIcon("zik-battery-100")
self.batteryState = "In Use"
elif self.batteryLevel > 60:
- self.setIcon("zik-battery-080")
+ self.indicator.setIcon("zik-battery-080")
self.batteryState = "In Use"
elif self.batteryLevel > 40:
- self.setIcon("zik-battery-060")
+ self.indicator.setIcon("zik-battery-060")
self.batteryState = "In Use"
elif self.batteryLevel > 20:
- self.setIcon("zik-battery-040")
+ self.indicator.setIcon("zik-battery-040")
self.batteryState = "In Use"
else:
- self.setIcon("zik-battery-low")
+ self.indicator.setIcon("zik-battery-low")
self.batteryState = "In Use"
- self.info_item.set_label("Connected to: " + self.name)
+ self.indicator.info_item.set_label("Connected to: " + self.name)
self.firmware_version.set_label(
"Firmware version: " + self.version)
self.battery_state.set_label("State: " + self.batteryState)
self.battery_level.set_label(
"Battery Level: " + str(self.batteryLevel))
else:
- self.setIcon("zik-audio-headset")
- self.info_item.set_label("Parrot Zik Not connected..")
- self.check.set_sensitive(False)
- self.check2.set_sensitive(False)
- self.check3.set_sensitive(False)
- self.check4.set_sensitive(False)
- return True
-
- def main(self):
- self.EstablishConnection()
- gtk.timeout_add(UPDATE_FREQUENCY * 1000, self.EstablishConnection)
- gtk.timeout_add(UPDATE_FREQUENCY * 1000, self.CheckBattery)
- SysIndicator.main(self)
+ self.deactivate()
if __name__ == "__main__":
indicator = ParrotZikIndicator()
diff --git a/SysIndicator.py b/SysIndicator.py
index ea93677..e8d59be 100644
--- a/SysIndicator.py
+++ b/SysIndicator.py
@@ -109,14 +109,17 @@ class UniversalMenu:
self.actions[MenuItem.title] = MenuItem.action
self.menubarMenu.addItem_(MenuItem.nsmenu_item)
+ def reposition(self):
+ if sys.platform == "linux2" or sys.platform == "win32":
+ self.gtk_menu.reposition()
+
class MenuItem:
- def __init__(self, name, action, sensitive=True, checkitem=False):
+ def __init__(self, name, action, sensitive=True, checkitem=False, visible=True):
if sys.platform == "linux2" or sys.platform == "win32":
if checkitem:
self.gtk_item = gtk.CheckMenuItem(name)
else:
self.gtk_item = gtk.MenuItem(name)
- self.gtk_item.show()
if action:
self.gtk_item.connect("activate", action)
@@ -126,8 +129,11 @@ class MenuItem:
self.nsmenu_item = (
NSMenuItem.alloc().initWithTitle_action_keyEquivalent_(
name, 'clicked:', ''))
-
self.set_sensitive(sensitive)
+ if visible:
+ self.show()
+ else:
+ self.hide()
def set_sensitive(self, option):
if sys.platform == "linux2" or sys.platform == "win32":
@@ -156,6 +162,18 @@ class MenuItem:
self.nsmenu_item.setTitle_(option)
#self.rumps_item.title=option
+ def show(self):
+ if sys.platform == "linux2" or sys.platform == "win32":
+ self.gtk_item.show()
+ elif sys.platform == "darwin":
+ self.nsmenu_item.show()
+
+ def hide(self):
+ if sys.platform == "linux2" or sys.platform == "win32":
+ self.gtk_item.hide()
+ elif sys.platform == "darwin":
+ self.nsmenu_item.hide()
+
if __name__ == "__main__":
quit_item = MenuItem("Quit", sys.exit, True)