134 lines
3.9 KiB
Python
134 lines
3.9 KiB
Python
import psutil
|
|
from fabric.widgets.box import Box
|
|
from fabric.widgets.label import Label
|
|
from fabric.widgets.overlay import Overlay
|
|
from fabric.widgets.datetime import DateTime
|
|
from fabric.widgets.centerbox import CenterBox
|
|
from bar.modules.player import Player
|
|
from bar.modules.vinyl import VinylButton
|
|
from fabric.widgets.wayland import WaylandWindow as Window
|
|
from fabric.system_tray.widgets import SystemTray
|
|
from fabric.river.widgets import (
|
|
RiverWorkspaces,
|
|
RiverWorkspaceButton,
|
|
RiverActiveWindow,
|
|
get_river_connection,
|
|
)
|
|
from fabric.utils import (
|
|
invoke_repeater,
|
|
)
|
|
from fabric.widgets.circularprogressbar import CircularProgressBar
|
|
|
|
from bar.config import VINYL
|
|
|
|
|
|
class StatusBar(Window):
|
|
def __init__(
|
|
self,
|
|
display: int,
|
|
tray: SystemTray | None = None,
|
|
monitor: int = 1,
|
|
river_service=None,
|
|
):
|
|
super().__init__(
|
|
name="bar",
|
|
layer="top",
|
|
anchor="left top right",
|
|
margin="0px 0px -2px 0px",
|
|
exclusivity="auto",
|
|
visible=False,
|
|
all_visible=False,
|
|
monitor=monitor,
|
|
)
|
|
if river_service:
|
|
self.river = river_service
|
|
else:
|
|
self.river = get_river_connection()
|
|
|
|
self.workspaces = RiverWorkspaces(
|
|
display,
|
|
name="workspaces",
|
|
spacing=4,
|
|
buttons_factory=lambda ws_id: RiverWorkspaceButton(id=ws_id, label=None),
|
|
river_service=self.river,
|
|
)
|
|
self.date_time = DateTime(name="date-time", formatters="%d %b - %H:%M")
|
|
self.system_tray = tray
|
|
|
|
self.active_window = RiverActiveWindow(
|
|
name="active-window",
|
|
max_length=50,
|
|
style="color: #ffffff; font-size: 14px; font-weight: bold;",
|
|
)
|
|
|
|
self.ram_progress_bar = CircularProgressBar(
|
|
name="ram-progress-bar", pie=True, size=24
|
|
)
|
|
self.cpu_progress_bar = CircularProgressBar(
|
|
name="cpu-progress-bar", pie=True, size=24
|
|
)
|
|
|
|
self.progress_label = Label(
|
|
"", style="margin: 0px 6px 0px 0px; font-size: 12px"
|
|
)
|
|
self.progress_bars_overlay = Overlay(
|
|
child=self.ram_progress_bar,
|
|
overlays=[self.cpu_progress_bar, self.progress_label],
|
|
)
|
|
self.player = Player()
|
|
self.vinyl = None
|
|
if VINYL["enabled"]:
|
|
self.vinyl = VinylButton()
|
|
|
|
self.status_container = Box(
|
|
name="widgets-container",
|
|
spacing=4,
|
|
orientation="h",
|
|
children=self.progress_bars_overlay,
|
|
)
|
|
|
|
end_container_children = []
|
|
|
|
if self.vinyl:
|
|
end_container_children.append(self.vinyl)
|
|
|
|
if self.system_tray:
|
|
end_container_children.append(self.system_tray)
|
|
|
|
end_container_children.append(self.date_time)
|
|
end_container_children.append(self.status_container)
|
|
|
|
self.children = CenterBox(
|
|
name="bar-inner",
|
|
start_children=Box(
|
|
name="start-container",
|
|
spacing=6,
|
|
orientation="h",
|
|
children=[
|
|
Label(name="nixos-label", markup=""),
|
|
self.workspaces,
|
|
],
|
|
),
|
|
center_children=Box(
|
|
name="center-container",
|
|
spacing=4,
|
|
orientation="h",
|
|
children=[self.active_window],
|
|
),
|
|
end_children=Box(
|
|
name="end-container",
|
|
spacing=4,
|
|
orientation="h",
|
|
children=end_container_children,
|
|
),
|
|
)
|
|
|
|
invoke_repeater(1000, self.update_progress_bars)
|
|
|
|
self.show_all()
|
|
|
|
def update_progress_bars(self):
|
|
self.ram_progress_bar.value = psutil.virtual_memory().percent / 100
|
|
self.cpu_progress_bar.value = psutil.cpu_percent() / 100
|
|
return True
|