add icon

2025-05-04 20:38:00 +02:00
parent ce030a8734
commit 2541edd0f4
6 changed files with 847 additions and 124 deletions
--- a/bar/modules/power.py
+++ b/bar/modules/power.py
@@ -0,0 +1,112 @@
+from fabric.widgets.box import Box
+from fabric.widgets.label import Label
+from fabric.widgets.button import Button
+from fabric.utils.helpers import exec_shell_command_async
+import modules.icons as icons
+
+
+class PowerMenu(Box):
+    def __init__(self, **kwargs):
+        super().__init__(
+            name="power-menu",
+            orientation="h",
+            spacing=4,
+            v_align="center",
+            h_align="center",
+            visible=True,
+            **kwargs,
+        )
+
+        self.notch = kwargs["notch"]
+
+        self.btn_lock = Button(
+            name="power-menu-button",
+            child=Label(name="button-label", markup=icons.lock),
+            on_clicked=self.lock,
+            h_expand=False,
+            v_expand=False,
+            h_align="center",
+            v_align="center",
+        )
+
+        self.btn_suspend = Button(
+            name="power-menu-button",
+            child=Label(name="button-label", markup=icons.suspend),
+            on_clicked=self.suspend,
+            h_expand=False,
+            v_expand=False,
+            h_align="center",
+            v_align="center",
+        )
+
+        self.btn_logout = Button(
+            name="power-menu-button",
+            child=Label(name="button-label", markup=icons.logout),
+            on_clicked=self.logout,
+            h_expand=False,
+            v_expand=False,
+            h_align="center",
+            v_align="center",
+        )
+
+        self.btn_reboot = Button(
+            name="power-menu-button",
+            child=Label(name="button-label", markup=icons.reboot),
+            on_clicked=self.reboot,
+            h_expand=False,
+            v_expand=False,
+            h_align="center",
+            v_align="center",
+        )
+
+        self.btn_shutdown = Button(
+            name="power-menu-button",
+            child=Label(name="button-label", markup=icons.shutdown),
+            on_clicked=self.poweroff,
+            h_expand=False,
+            v_expand=False,
+            h_align="center",
+            v_align="center",
+        )
+
+        self.buttons = [
+            self.btn_lock,
+            self.btn_suspend,
+            self.btn_logout,
+            self.btn_reboot,
+            self.btn_shutdown,
+        ]
+
+        for button in self.buttons:
+            self.add(button)
+
+        self.show_all()
+
+    def close_menu(self):
+        self.notch.close_notch()
+
+    # Métodos de acción
+    def lock(self, *args):
+        print("Locking screen...")
+        exec_shell_command_async("loginctl lock-session")
+        self.close_menu()
+
+    def suspend(self, *args):
+        print("Suspending system...")
+        exec_shell_command_async("systemctl suspend")
+        self.close_menu()
+
+    def logout(self, *args):
+        print("Logging out...")
+        exec_shell_command_async("hyprctl dispatch exit")
+        self.close_menu()
+
+    def reboot(self, *args):
+        print("Rebooting system...")
+        exec_shell_command_async("systemctl reboot")
+        self.close_menu()
+
+    def poweroff(self, *args):
+        print("Powering off...")
+        exec_shell_command_async("systemctl poweroff")
+        self.close_menu()