bumblebee-status/modules/contrib/layout.py
2020-04-24 16:16:14 +02:00

72 lines
2.2 KiB
Python

# pylint: disable=C0111,R0903
"""Displays and changes the current keyboard layout
Requires the following executable:
* setxkbmap
"""
import bumblebee.util
import bumblebee.input
import bumblebee.output
import bumblebee.engine
class Module(bumblebee.engine.Module):
def __init__(self, engine, config):
super(Module, self).__init__(engine, config,
bumblebee.output.Widget(full_text=self.current_layout)
)
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
cmd=self._next_keymap)
engine.input.register_callback(self, button=bumblebee.input.RIGHT_MOUSE,
cmd=self._prev_keymap)
def _next_keymap(self, event):
self._set_keymap(1)
def _prev_keymap(self, event):
self._set_keymap(-1)
def _set_keymap(self, rotation):
layouts = self.get_layouts()
if len(layouts) == 1: return # nothing to do
layouts = layouts[rotation:] + layouts[:rotation]
layout_list = []
variant_list = []
for l in layouts:
tmp = l.split(':')
layout_list.append(tmp[0])
variant_list.append(tmp[1] if len(tmp) > 1 else '')
try:
bumblebee.util.execute('setxkbmap -layout {} -variant {}'.format(','.join(layout_list), ','.join(variant_list)))
except RuntimeError:
pass
def get_layouts(self):
try:
res = bumblebee.util.execute('setxkbmap -query')
except RuntimeError:
return ['n/a']
layouts = []
variants = []
for line in res.split('\n'):
if not line: continue
if 'layout' in line:
layouts = line.split(':')[1].strip().split(',')
if 'variant' in line:
variants = line.split(':')[1].strip().split(',')
result = []
for idx, layout in enumerate(layouts):
if len(variants) > idx and variants[idx]:
layout = '{}:{}'.format(layout, variants[idx])
result.append(layout)
return result if len(result) > 0 else ['n/a']
def current_layout(self, widget):
layouts = self.get_layouts()
return layouts[0]
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4