137 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			137 lines
		
	
	
		
			3.7 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()
 | |
| 
 | |
| 	status_icon = ui.status_icon.create(ui.main_window.toggle_all, ui.main_window.popup)
 | |
| 	assert status_icon
 | |
| 
 | |
| 	listeners = {}
 | |
| 	from solaar.listener import ReceiverListener
 | |
| 
 | |
| 	def handle_receivers_events(action, device_info):
 | |
| 		assert action is not None
 | |
| 		assert device_info is not None
 | |
| 
 | |
| 		# whatever the action, stop any previous receivers at this path
 | |
| 		l = listeners.pop(device_info.path, None)
 | |
| 		if l is not None:
 | |
| 			assert isinstance(l, ReceiverListener)
 | |
| 			l.stop()
 | |
| 
 | |
| 		if action == 'add':
 | |
| 			# a new receiver device was detected
 | |
| 			try:
 | |
| 				l = ReceiverListener.open(device_info, status_changed)
 | |
| 				if l is not None:
 | |
| 					listeners[device_info.path] = l
 | |
| 			except OSError:
 | |
| 				# permission error, blacklist this path for now
 | |
| 				listeners.pop(device_info.path, None)
 | |
| 				GLib.idle_add(ui.error_dialog, 'Permissions error',
 | |
| 					'Found a Logitech Receiver, but did not have permission to open it.\n'
 | |
| 					'\n'
 | |
| 					'If you\'ve just installed Solaar, try removing the receiver\n'
 | |
| 					'and plugging it back in.')
 | |
| 
 | |
| 		# 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:
 | |
| 		# 		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)
 | |
| 
 | |
| 		GLib.idle_add(ui.status_icon.update, status_icon, device)
 | |
| 		GLib.idle_add(ui.main_window.update, device, alert & ALERT.SHOW_WINDOW, status_icon)
 | |
| 
 | |
| 		if alert & ALERT.NOTIFICATION:
 | |
| 			GLib.idle_add(ui.notify.show, device, reason)
 | |
| 
 | |
| 	# ugly...
 | |
| 	def _startup_check_receiver():
 | |
| 		if not listeners:
 | |
| 			ui.notify.alert('No receiver found.')
 | |
| 			ui.status_icon.update(status_icon)
 | |
| 	GLib.timeout_add(1000, _startup_check_receiver)
 | |
| 
 | |
| 	from logitech.unifying_receiver import base as _base
 | |
| 	GLib.timeout_add(10, _base.notify_on_receivers, handle_receivers_events)
 | |
| 	from gi.repository import Gtk
 | |
| 	Gtk.main()
 | |
| 	# ui.status_icon.destroy(status_icon)
 | |
| 
 | |
| 	for l in listeners.values():
 | |
| 		l.stop()
 | |
| 	ui.notify.uninit()
 | |
| 	for l in listeners.values():
 | |
| 		l.join()
 | |
| 
 | |
| 
 | |
| 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()
 |