Solaar/lib/solaar/gtk.py

144 lines
3.9 KiB
Python

#
#
#
from __future__ import absolute_import, division, print_function, unicode_literals
from solaar import __version__, NAME
#
#
#
def _require(module, os_package):
try:
__import__(module)
except ImportError:
import sys
sys.exit("%s: missing required package '%s'" % (NAME, os_package))
def _parse_arguments():
import argparse
arg_parser = argparse.ArgumentParser(prog=NAME.lower())
arg_parser.add_argument('-d', '--debug', action='count', default=0,
help='print logging messages, for debugging purposes (may be repeated for extra verbosity)')
arg_parser.add_argument('-V', '--version', action='version', version='%(prog)s ' + __version__)
args = arg_parser.parse_args()
import logging
if args.debug > 0:
log_level = logging.WARNING - 10 * args.debug
log_format='%(asctime)s %(levelname)8s [%(threadName)s] %(name)s: %(message)s'
logging.basicConfig(level=max(log_level, logging.DEBUG), format=log_format)
else:
logging.root.addHandler(logging.NullHandler())
logging.root.setLevel(logging.ERROR)
return args
def _run(args):
import solaar.ui as ui
ui.notify.init()
window = ui.main_window.create(NAME)
assert window
icon = ui.status_icon.create(window)
assert icon
listeners = {}
from logitech.unifying_receiver import base as _base
from solaar.listener import ReceiverListener
def handle_receivers_events(action, device):
assert action is not None
assert device is not None
if action == 'add':
# a new receiver device was detected
if not listeners:
# handle only one receiver for now, the rest are ignored
try:
l = ReceiverListener.open(device.path, status_changed)
if l is not None:
listeners[device.path] = l
except OSError:
# permission error, blacklist this path for now
listeners.pop(device.path, None)
import logging
logging.exception("failed to open %s", device.path)
# ui.error_dialog(window, 'Permissions error',
# 'Found a possible Unifying Receiver device,\n'
# 'but did not have permission to open it.')
elif action == 'remove':
# we'll be receiving remove events for any hidraw devices,
# not just Logitech receivers, so it's okay if the device is not
# already in our listeners map
l = listeners.pop(device.path, None)
if l is not None:
assert isinstance(l, ReceiverListener)
l.stop()
# print ("****", action, device, listeners)
# callback delivering status notifications from the receiver/devices to the UI
from gi.repository import GLib
from logitech.unifying_receiver.status import ALERT
def status_changed(device, alert=ALERT.NONE, reason=None):
assert device is not None
# print ("status changed", device, reason)
if alert & ALERT.SHOW_WINDOW:
GLib.idle_add(window.present)
GLib.idle_add(ui.main_window.update, window, device)
GLib.idle_add(ui.status_icon.update, icon, device)
if ui.notify.available:
if alert & ALERT.NOTIFICATION:
GLib.idle_add(ui.notify.show, device, reason)
elif device.kind is None and not device:
# notify when a receiver was removed
GLib.idle_add(ui.notify.show, device, reason)
# if device.kind is None and not device:
# # a receiver was removed
# listeners.clear()
# ugly...
def _startup_check_receiver():
from solaar.listener import DUMMY_RECEIVER
if not listeners:
status_changed(DUMMY_RECEIVER, ALERT.NOTIFICATION)
GLib.timeout_add(1000, _startup_check_receiver)
GLib.timeout_add(10, _base.notify_on_receivers, handle_receivers_events)
from gi.repository import Gtk
Gtk.main()
map(ReceiverListener.stop, listeners.values())
ui.notify.uninit()
map(ReceiverListener.join, listeners.values())
def main():
_require('pyudev', 'python-pyudev')
_require('gi.repository', 'python-gi')
_require('gi.repository.Gtk', 'gir1.2-gtk-3.0')
args = _parse_arguments()
from . import appinstance
appid = appinstance.check()
try:
_run(args)
finally:
appinstance.close(appid)
if __name__ == '__main__':
main()