summaryrefslogtreecommitdiffstats
path: root/modules/keyboard-layout.py
blob: 1ed3cee665b5de13a9880c4e5adb4d75f33d3793 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
from subprocess import check_output
from time import time

"""
Module for showing current keyboard layout.

Requires:
    - xkblayout-state
    or
    - setxkbmap

@author shadowprince
@license Eclipse Public License
"""

CACHE_TIMEOUT = 10  # refresh time to update indicator

# colors of layouts, check your command's output to match keys
LANG_COLORS = {
    'fr': '#268BD2',  # solarized blue
    'ru': '#F75252',  # red
    'ua': '#FCE94F',  # yellow
    'us': '#729FCF',  # light blue
}


def xbklayout():
    """
    check using xkblayout-state (preferred method)
    """
    return check_output(
        ["xkblayout-state", "print", "%s"]
    ).decode('utf-8')


def setxkbmap():
    """
    check using setxkbmap >= 1.3.0

    Please read issue 33 for more information :
        https://github.com/ultrabug/py3status/pull/33
    """
    q = check_output(['setxkbmap', '-query']).decode('utf-8')
    return q.replace(' ', '').split(':')[-1]


class Py3status:
    def __init__(self):
        """
        find the best implementation to get the keyboard's layout
        """
        try:
            xbklayout()
        except:
            self.command = setxkbmap
        else:
            self.command = xbklayout

    def keyboard_layout(self, i3status_output_json, i3status_config):
        response = {
            'full_text': '',
            'name': 'keyboard-layout',
            'cached_until': time() + CACHE_TIMEOUT
        }

        lang = self.command().strip()
        lang_color = LANG_COLORS.get(lang)

        response['full_text'] = lang or '??'
        if lang_color:
            response['color'] = lang_color

        return (0, response)