i3pystatus/i3pystatus/keyboard_locks.py

59 lines
1.4 KiB
Python

import subprocess
from i3pystatus import IntervalModule
class Keyboard_locks(IntervalModule):
"""
Shows the status of CAPS LOCK, NUM LOCK and SCROLL LOCK
Available formatters:
* `{caps}` — the current status of CAPS LOCK
* `{num}` — the current status of NUM LOCK
* `{scroll}` — the current status of SCROLL LOCK
"""
interval = 1
settings = (
"format",
"caps_on",
"caps_off",
"num_on",
"num_off",
"scroll_on",
"scroll_off",
"color"
)
format = "{caps} {num} {scroll}"
caps_on = "CAP"
caps_off = "___"
num_on = "NUM"
num_off = "___"
scroll_on = "SCR"
scroll_off = "___"
color = "#FFFFFF"
fdict = {}
def get_status(self):
xset = str(subprocess.check_output(["xset", "q"]))
cap = xset.split("Caps Lock:")[1][0:8]
num = xset.split("Num Lock:")[1][0:8]
scr = xset.split("Scroll Lock:")[1][0:8]
return("on" in cap, "on" in num, "on" in scr)
def run(self):
(cap, num, scr) = self.get_status()
self.fdict["caps"] = self.caps_on if cap else self.caps_off
self.fdict["num"] = self.num_on if num else self.num_off
self.fdict["scroll"] = self.scroll_on if scr else self.scroll_off
output_format = self.format
self.output = {
"full_text": output_format.format(**self.fdict),
"color": self.color,
}