1 files changed, 86 insertions, 0 deletions
diff --git a/py3status/modules/keyboard_layout.py b/py3status/modules/keyboard_layout.py
new file mode 100644
index 0000000..ffc3907
--- /dev/null
+++ b/py3status/modules/keyboard_layout.py
@@ -0,0 +1,86 @@
+# -*- coding: utf-8 -*-
+"""
+Module for showing current keyboard layout.
+
+Requires:
+    - xkblayout-state
+    or
+    - setxkbmap
+
+@author shadowprince
+@license Eclipse Public License
+"""
+
+from subprocess import check_output
+from time import time
+
+# 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:
+
+    # available configuration parameters
+    cache_timeout = 10
+    color = ''
+
+    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, i3s_output_list, i3s_config):
+        response = {
+            'cached_until': time() + self.cache_timeout,
+            'full_text': ''
+        }
+
+        lang = self.command().strip()
+        lang_color = self.color if self.color else LANG_COLORS.get(lang)
+
+        response['full_text'] = lang or '??'
+        if lang_color:
+            response['color'] = lang_color
+
+        return response
+
+if __name__ == "__main__":
+    """
+    Test this module by calling it directly.
+    """
+    from time import sleep
+    x = Py3status()
+    while True:
+        print(x.keyboard_layout([], {}))
+        sleep(1)