2 # -*- coding: utf-8 -*-
4 # indicator-keyboard-led - simulate keyboard lock keys LED
5 # Copyright (c) 2016 Adrian I Lam <adrianiainlam@gmail.com>
7 # Permission is hereby granted, free of charge, to any person obtaining a
8 # copy of this software and associated documentation files (the "Software"),
9 # to deal in the Software without restriction, including without limitation
10 # the rights to use, copy, modify, merge, publish, distribute, sublicense,
11 # and/or sell copies of the Software, and to permit persons to whom the
12 # Software is furnished to do so, subject to the following conditions:
14 # The above copyright notice and this permission notice shall be included in
15 # all copies or substantial portions of the Software.
17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20 # THE AUTHOR OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
22 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
25 # I would like to thank Tobias Schlitt <toby@php.net>, who wrote
26 # indicator-chars <https://github.com/tobyS/indicator-chars> which I used
27 # as a reference when writing this software.
35 gi
.require_version('Gdk', '3.0')
36 gi
.require_version('Gtk', '3.0')
37 gi
.require_version('AppIndicator3', '0.1')
38 from gi
.repository
import Gdk
, Gtk
, AppIndicator3
40 SCRIPT_DIR
= path
.dirname(path
.realpath(__file__
))
42 t
= gettext
.translation('default', path
.join(SCRIPT_DIR
, 'locale'))
45 class IndicatorKeyboardLED
:
46 locks
= { 'N': _('Num'), 'C': _('Caps'), 'S': _('Scroll') }
48 def __init__(self
, short
=False, order
='NCS'):
49 self
.indicator
= AppIndicator3
.Indicator
.new(
50 'indicator-keyboard-led',
51 path
.join(SCRIPT_DIR
, 'icon.svg'),
52 AppIndicator3
.IndicatorCategory
.APPLICATION_STATUS
)
53 self
.indicator
.set_status(AppIndicator3
.IndicatorStatus
.ACTIVE
)
56 self
.locks
= { 'N': _('N'), 'C': _('C'), 'S': _('S') }
58 keymap
= Gdk
.Keymap
.get_default()
59 keymap
.connect('state-changed', self
.update_indicator
, order
)
60 self
.update_indicator(keymap
, order
)
64 'N': Gtk
.MenuItem
.new_with_label(_('Num Lock')),
65 'C': Gtk
.MenuItem
.new_with_label(_('Caps Lock')),
66 'S': Gtk
.MenuItem
.new_with_label(_('Scroll Lock'))
68 items
['N'].connect('activate', self
.send_keypress
, 'Num_Lock')
69 items
['C'].connect('activate', self
.send_keypress
, 'Caps_Lock')
70 items
['S'].connect('activate', self
.send_keypress
, 'Scroll_Lock')
75 quit_item
= Gtk
.MenuItem
.new_with_label(_('Quit'))
76 menu
.append(Gtk
.SeparatorMenuItem())
77 menu
.append(quit_item
)
78 quit_item
.connect('activate', Gtk
.main_quit
)
80 self
.indicator
.set_menu(menu
)
83 def update_indicator(self
, keymap
, order
):
87 state
= keymap
.get_num_lock_state()
89 state
= keymap
.get_caps_lock_state()
91 state
= keymap
.get_scroll_lock_state()
93 raise ValueError('Invalid value in ORDER')
94 labels
+= [('⚫' if state
else '⚪') + self
.locks
[i
]]
95 self
.indicator
.set_label(' '.join(labels
), '')
97 def send_keypress(self
, menuitem
, keystroke
):
98 subprocess
.call(['xdotool', 'key', keystroke
])
100 def validate_order(args
):
101 args
.order
= args
.order
.upper()
103 if i
not in ['N', 'C', 'S']:
104 sys
.exit('Illegal character in ORDER. (Choices: [N, C, S])')
105 if len(args
.order
) != len(set(args
.order
)):
106 sys
.exit('Repeated character in ORDER. '
107 'Please specify each lock at most once.')
109 if __name__
== '__main__':
110 signal
.signal(signal
.SIGINT
, lambda signum
, frame
: Gtk
.main_quit())
112 parser
= argparse
.ArgumentParser(
113 description
='indicator-keyboard-led - simulate keyboard lock keys LED',
114 formatter_class
=argparse
.ArgumentDefaultsHelpFormatter
)
115 parser
.add_argument('-s', '--short', required
=False, action
='store_true',
116 help='use short label, i.e. ⚫N ⚫C ⚫S instead of ⚫Num ⚫Caps ⚫Scroll')
117 parser
.add_argument('-o', '--order', required
=False, default
='NCS',
118 help='specify the order of the locks displayed, e.g. CSN for '
119 '⚫Caps ⚫Scroll ⚫Num, or NC for ⚫Num ⚫Caps without Scroll lock')
120 args
= parser
.parse_args()
123 IndicatorKeyboardLED(short
=args
.short
, order
=args
.order
)