module just started

nwg_panel/modules/keyboard_layout.py

@@ -0,0 +1,144 @@
+#!/usr/bin/env python3
+
+import os
+import subprocess
+
+import gi
+from gi.repository import GLib
+
+from nwg_panel.tools import check_key, update_image, eprint, hyprctl
+
+gi.require_version('Gtk', '3.0')
+gi.require_version('Gdk', '3.0')
+
+from gi.repository import Gtk, Gdk
+
+
+class KeyboardLayout(Gtk.EventBox):
+    def __init__(self, settings, icons_path, executor_name):
+        self.name = executor_name
+        self.settings = settings
+        self.icons_path = icons_path
+        Gtk.EventBox.__init__(self)
+        self.box = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=0)
+        self.add(self.box)
+        self.image = Gtk.Image()
+        self.label = Gtk.Label.new("")
+        self.icon_path = None
+
+        self.kb_layout = self.get_kb_layout()
+
+        check_key(settings, "root-css-name", "root-executor")
+        check_key(settings, "css-name", "")
+        check_key(settings, "icon-placement", "left")
+        check_key(settings, "icon-size", 16)
+        check_key(settings, "tooltip-text", "")
+        check_key(settings, "angle", 0.0)
+
+        self.label.set_angle(settings["angle"])
+
+        if settings["angle"] != 0.0:
+            self.box.set_orientation(Gtk.Orientation.VERTICAL)
+
+        update_image(self.image, "input-keyboard", self.settings["icon-size"], self.icons_path)
+
+        self.set_property("name", settings["root-css-name"])
+
+        if settings["css-name"]:
+            self.label.set_property("name", settings["css-name"])
+        else:
+            self.label.set_property("name", "executor-label")
+
+        if settings["tooltip-text"]:
+            self.set_tooltip_text(settings["tooltip-text"])
+
+        self.connect('button-press-event', self.on_button_press)
+
+        self.connect('enter-notify-event', self.on_enter_notify_event)
+        self.connect('leave-notify-event', self.on_leave_notify_event)
+
+        self.build_box()
+        self.refresh()
+
+    def get_kb_layout(self):
+        option = hyprctl("j/getoption input:kb_layout")
+        if option and "str" in option:
+            return option["str"]
+        return ""
+
+    def update_widget(self, output):
+        # parse output
+        label = new_path = None
+        if output:
+            output = [o.strip() for o in output]
+            if len(output) == 1:
+                if os.path.splitext(output[0])[1] in ('.svg', '.png'):
+                    new_path = output[0]
+                else:
+                    label = output[0]
+            elif len(output) == 2:
+                new_path, label = output
+
+        # update widget contents
+        if new_path and new_path != self.icon_path:
+            try:
+                update_image(self.image,
+                             new_path,
+                             self.settings["icon-size"],
+                             self.icons_path,
+                             fallback=False)
+                self.icon_path = new_path
+            except:
+                print("Failed setting image from {}".format(new_path))
+                new_path = None
+
+        if label:
+            self.label.set_markup(label)
+
+        # update widget visibility
+        if new_path:
+            if not self.image.get_visible():
+                self.image.show()
+        else:
+            if self.image.get_visible():
+                self.image.hide()
+
+        if label:
+            if not self.label.get_visible():
+                self.label.show()
+        else:
+            if self.label.get_visible():
+                self.label.hide()
+
+        return False
+
+    def refresh(self):
+        if "script" in self.settings and self.settings["script"]:
+            try:
+                output = subprocess.check_output(self.settings["script"].split()).decode("utf-8").splitlines()
+                GLib.idle_add(self.update_widget, output)
+            except Exception as e:
+                eprint(e)
+
+    def build_box(self):
+        if self.settings["icon-placement"] == "left":
+            self.box.pack_start(self.image, False, False, 2)
+        self.box.pack_start(self.label, False, False, 2)
+        if self.settings["icon-placement"] != "left":
+            self.box.pack_start(self.image, False, False, 2)
+
+    def on_enter_notify_event(self, widget, event):
+        widget.set_state_flags(Gtk.StateFlags.DROP_ACTIVE, clear=False)
+        widget.set_state_flags(Gtk.StateFlags.SELECTED, clear=False)
+
+    def on_leave_notify_event(self, widget, event):
+        widget.unset_state_flags(Gtk.StateFlags.DROP_ACTIVE)
+        widget.unset_state_flags(Gtk.StateFlags.SELECTED)
+
+    def on_button_press(self, widget, event):
+        if event.button == 1 and self.settings["on-left-click"]:
+            self.launch(self.settings["on-left-click"])
+        elif event.button == 2 and self.settings["on-middle-click"]:
+            self.launch(self.settings["on-middle-click"])
+        elif event.button == 3 and self.settings["on-right-click"]:
+            self.launch(self.settings["on-right-click"])