89 lines
2.2 KiB
Python
89 lines
2.2 KiB
Python
from fabric.widgets.box import Box
|
|
from fabric.widgets.button import Button
|
|
from gi.repository import GLib
|
|
|
|
from snippets import MaterialIcon
|
|
|
|
|
|
class PowerMenu(Box):
|
|
def __init__(self, **kwargs):
|
|
super().__init__(
|
|
name="powermenu",
|
|
orientation="h",
|
|
spacing=10,
|
|
v_align="center",
|
|
h_align="center",
|
|
v_expand=True,
|
|
h_expand=True,
|
|
visible=True,
|
|
**kwargs,
|
|
)
|
|
|
|
self.launcher = kwargs["launcher"]
|
|
|
|
self.btn_lock = Button(
|
|
name="powermenu-button",
|
|
child=MaterialIcon("lock", 45),
|
|
on_clicked=self.lock,
|
|
)
|
|
|
|
self.btn_suspend = Button(
|
|
name="powermenu-button",
|
|
child=MaterialIcon("sleep", 45),
|
|
on_clicked=self.suspend,
|
|
)
|
|
|
|
self.btn_logout = Button(
|
|
name="powermenu-button",
|
|
child=MaterialIcon("logout", 45),
|
|
on_clicked=self.logout,
|
|
)
|
|
|
|
self.btn_reboot = Button(
|
|
name="powermenu-button",
|
|
child=MaterialIcon("restart_alt", 45),
|
|
on_clicked=self.reboot,
|
|
)
|
|
|
|
self.btn_shutdown = Button(
|
|
name="powermenu-button",
|
|
child=MaterialIcon("power_settings_new", 45),
|
|
on_clicked=self.poweroff,
|
|
)
|
|
|
|
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.launcher.close()
|
|
|
|
def lock(self, *_):
|
|
GLib.spawn_command_line_async("hyprlock --immediate")
|
|
self.close_menu()
|
|
|
|
def suspend(self, *_):
|
|
GLib.spawn_command_line_async("systemctl suspend")
|
|
self.close_menu()
|
|
|
|
def logout(self, *_):
|
|
GLib.spawn_command_line_async("hyprctl dispatch exit")
|
|
self.close_menu()
|
|
|
|
def reboot(self, *_):
|
|
GLib.spawn_command_line_async("systemctl reboot")
|
|
self.close_menu()
|
|
|
|
def poweroff(self, *_):
|
|
GLib.spawn_command_line_async("systemctl poweroff")
|
|
self.close_menu()
|