Files
Modus/modules/widget.py
T

901 lines
28 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.circularprogressbar import CircularProgressBar
from widgets.wayland import WaylandWindow as Window
from fabric.utils import invoke_repeater
import requests
import urllib.parse
import datetime
from gi.repository import GLib
from concurrent.futures import ThreadPoolExecutor
import time
import config.data as data
from config.data import load_config
import subprocess
import calendar
executor = ThreadPoolExecutor(max_workers=4)
config = load_config()
def margin():
return (
data.DOCK_ICON_SIZE + 10
if not data.DOCK_ALWAYS_OCCLUDED and data.DOCK_ENABLED
else 0
)
def get_location():
"""Fetch location from config file or IP API asynchronously."""
for attempt in range(5):
try:
response = requests.get("https://ipinfo.io/json", timeout=3)
if response.status_code == 200:
return response.json().get("city", "").replace(" ", "")
except requests.RequestException as e:
print(f"Error getting location: {e}")
if attempt < 4:
time.sleep(10)
else:
return ""
def get_location_async(callback):
"""Fetch location asynchronously to prevent UI freeze."""
executor.submit(lambda: GLib.idle_add(callback, get_location()))
def get_weather(callback):
"""Fetch weather data asynchronously and update UI."""
def fetch_weather():
location = get_location()
if not location:
return GLib.idle_add(callback, None)
encoded_location = urllib.parse.quote(location)
url = f"https://wttr.in/{encoded_location}?format=j1"
urlemoji = f"https://wttr.in/{encoded_location}?format=%c"
for attempt in range(5):
try:
response = requests.get(urlemoji, timeout=3)
responseinfo = requests.get(url, timeout=3).json()
if response.status_code == 200:
temp_unit = "C"
temp = (
responseinfo["current_condition"][0][f"temp_{temp_unit}"] + "°"
)
condition = responseinfo["current_condition"][0]["weatherDesc"][0][
"value"
]
maxtemp = responseinfo["weather"][0]["maxtempC"] + "°"
mintemp = responseinfo["weather"][0]["mintempC"] + "°"
location = responseinfo["nearest_area"][0]["areaName"][0]["value"]
emoji = response.text.strip()
GLib.idle_add(
callback,
[
emoji,
temp,
condition,
location,
maxtemp,
mintemp,
],
)
return
except requests.RequestException as e:
print(f"Error fetching weather (attempt {attempt + 1}): {e}")
if attempt < 4:
time.sleep(10)
else:
GLib.idle_add(callback, None)
return
executor.submit(fetch_weather)
def update_weather(widget):
def fetch_and_update():
get_weather(lambda weather_info: update_widget(widget, weather_info))
return True
GLib.timeout_add_seconds(600, fetch_and_update)
fetch_and_update()
def update_widget(widget, weather_info):
if weather_info:
widget.weatherinfo = weather_info
widget.update_labels(weather_info)
class Sysinfo(Box):
@staticmethod
def bake_progress_bar(name: str = "progress-bar", size: int = 45, **kwargs):
return CircularProgressBar(
name=name,
start_angle=180,
end_angle=540,
min_value=0,
max_value=100,
size=size,
**kwargs,
)
@staticmethod
def bake_progress_icon(**kwargs):
return Label(**kwargs).build().add_style_class("progress-icon").unwrap()
def __init__(self, **kwargs):
super().__init__(
layer="bottom",
title="sysinfo",
name="sysinfo",
visible=False,
all_visible=False,
**kwargs,
)
self.cpu_progress = self.bake_progress_bar()
self.ram_progress = self.bake_progress_bar()
self.bat_circular = self.bake_progress_bar().build().set_value(42).unwrap()
self.progress_container = Box(
name="progress-bar-container",
spacing=12,
children=[
Box(
children=[
Overlay(
child=self.cpu_progress,
tooltip_text="",
overlays=[
self.bake_progress_icon(
label="",
name="progress-icon-cpu",
)
],
),
],
),
Box(
children=[
Overlay(
child=self.ram_progress,
tooltip_text="",
overlays=[
self.bake_progress_icon(
name="progress-icon-ram",
label="󰘚",
)
],
)
]
),
Box(
children=[
Overlay(
child=self.bat_circular,
tooltip_text="",
overlays=[
self.bake_progress_icon(
label="󱊣",
name="progress-icon-bat",
)
],
),
],
),
],
)
self.update_status()
invoke_repeater(1000, self.update_status)
self.add(
Box(
name="progress-bar-container-main",
orientation="v",
spacing=24,
children=[self.progress_container],
),
)
self.show_all()
def update_status(self):
"""Update system info asynchronously to prevent UI lag."""
def update():
cpu = psutil.cpu_percent()
ram = psutil.virtual_memory().percent
battery = (
psutil.sensors_battery().percent if psutil.sensors_battery() else 80
)
GLib.idle_add(self.cpu_progress.set_value, cpu)
GLib.idle_add(self.ram_progress.set_value, ram)
GLib.idle_add(self.bat_circular.set_value, battery)
GLib.idle_add(self.cpu_progress.set_tooltip_text, f"{str(round(cpu))}%")
GLib.idle_add(self.ram_progress.set_tooltip_text, f"{str(round(ram))}%")
GLib.idle_add(self.bat_circular.set_tooltip_text, f"{str(round(battery))}%")
executor.submit(update)
return True
def fetch_quote(callback):
"""Fetch quotes asynchronously."""
def fetch():
quotes_type = "stoic"
url = (
"https://stoic-quotes.com/api/quote"
if quotes_type == "stoic"
else "https://zenquotes.io/api/random"
)
for attempt in range(5):
try:
response = requests.get(url, timeout=3)
response.raise_for_status()
respdata = response.json()
quote = (
f"{respdata[0]['q']} - {respdata[0]['a']}"
if quotes_type == "zen"
else f"{respdata['text']} - {respdata['author']}"
)
break
except requests.RequestException as e:
print(f"Error fetching quote: {e}")
if attempt < 4:
time.sleep(10)
else:
try:
result = subprocess.run(
["hyprctl", "splash"],
capture_output=True,
text=True,
check=True,
)
quote = result.stdout.strip() + " - Team Hyprland"
except subprocess.CalledProcessError as e:
print(f"Error fetching quote from hyprctl: {e}")
quote = "I learn from the mistakes of people who take my advice - Trix"
GLib.idle_add(callback, quote)
executor.submit(fetch)
def fetch_quote_async(callback):
GLib.idle_add(lambda: fetch_quote(callback))
class QuoteWidget(Label):
def __init__(self, **kwargs):
super().__init__(
name="quote",
label="",
anchor="center",
h_align="center",
v_align="center",
h_expand=True,
justification="center",
v_expand=True,
visible=False,
)
fetch_quote_async(self.update_label)
def update_label(self, quote):
"""Update quote asynchronously."""
max_width = 150 # Set the maximum width for the quote
if len(quote) > max_width:
words = quote.split()
line1, line2 = "", ""
for word in words:
if len(line1) + len(word) + 1 <= max_width:
line1 += word + " "
else:
line2 += word + " "
quote = line1.strip() + "\n" + line2.strip()
self.set_label(quote)
self.set_visible(True)
class ActivationMainText(Label):
def __init__(self, **kwargs):
super().__init__(
name="activation-main-text",
label="",
anchor="bottom right",
justification="left",
v_align="start",
h_align="start",
h_expand=True,
v_expand=True,
visible=False,
)
self.set_label("Activate Linux")
class ActivationSubText(Label):
def __init__(self, **kwargs):
super().__init__(
name="activation-sub-text",
label="",
anchor="bottom right",
justification="left",
v_align="start",
h_align="start",
h_expand=True,
v_expand=True,
visible=False,
)
self.set_label("Go to Settings to activate Linux")
def create_widgets(config):
widgets = []
if config.get("widgets_displaytype_visible", True):
if config.get("widgets_clock_visible", True):
widgets.append(
DateTime(formatters=["%A, %d %B"], interval=10000, name="date")
)
if config.get("widgets_date_visible", True):
widgets.append(DateTime(formatters=["%I:%M %p"], name="clock"))
if config.get("widgets_quote_visible", True):
widgets.append(QuoteWidget())
if config.get("widgets_weatherwid_visible", True):
widgets.append(WeatherContainer())
else:
widgets.append(DateTime(formatters=["%I:%M %p"], name="clock"))
widgets.append(DateTime(formatters=["%A. %d %B"], interval=10000, name="date"))
return widgets
class Weather(Box):
def __init__(self, parent, **kwargs):
super().__init__(
name="weather-widget",
h_expand=True,
v_expand=True,
justification="right",
orientation="v",
all_visible=False,
**kwargs,
)
self.parent = parent
self.city = Label(
name="city", label="New York", justification="right", h_align="start"
)
self.temperature = Label(name="temperature", label="36", h_align="start")
self.condition_em = Label(name="condition-emoji", label="☁️", h_align="start")
self.condition = Label(name="condition", label="Cloudy", h_align="start")
self.feels_like = Label(name="feels-like", label="H:28 L:20", h_align="start")
self.add(self.city)
self.add(self.temperature)
self.add(self.condition_em)
self.add(self.condition)
self.add(self.feels_like)
update_weather(self)
def update_labels(self, weather_info):
if not self.weatherinfo:
return
# Unpack weather info into variables for better readability
emoji, temp, condition, location, maxtemp, mintemp = self.weatherinfo
maxmin = f"H:{maxtemp} L:{mintemp}"
# Store references to deeply nested children to avoid repeated lookups
self.city.set_label(location)
self.feels_like.set_label(maxmin)
self.condition.set_label(condition)
self.temperature.set_label(temp)
self.condition_em.set_label(emoji)
self.parent.set_visible(True)
class WeatherContainer(Box):
def __init__(self, **kwargs):
super().__init__(
orientation="v",
name="box-widget-2",
v_expand=True,
v_align="center",
size=(170, 170),
visible=True,
h_align="center",
children=[
Weather(self),
],
**kwargs,
)
# self.set_style("background-color: rgba(0, 0, 0, 0.5);")
# self.add(Weather())
class Date(Box):
def __init__(self, **kwargs):
super().__init__(
name="date-widget",
h_expand=True,
v_expand=True,
justification="center",
h_align="center",
v_align="start",
orientation="v",
**kwargs,
)
self.top = Box(orientation="h", name="date-top", h_expand=True)
self.dateone = DateTime(formatters=["%a"], interval=10000, name="day")
self.datetwo = DateTime(formatters=["%b"], interval=10000, name="month")
self.datethree = DateTime(formatters=["%-d"], interval=10000, name="date")
self.top.add(self.dateone)
self.top.add(self.datetwo)
self.add(self.top)
self.add(self.datethree)
class DateContainer(Box):
def __init__(self, **kwargs):
super().__init__(
orientation="v",
name="box-widget",
v_expand=True,
size=(170, 170),
v_align="center",
h_align="center",
children=[Date()],
**kwargs,
)
class Calendar(Box):
def __init__(self, **kwargs):
super().__init__(
name="calendar-widget",
h_expand=True,
v_expand=True,
orientation="v",
**kwargs,
)
# Create current date for calendar
now = datetime.datetime.now()
self.current_month = now.month
self.current_year = now.year
self.current_day = now.day
# Month and year header
self.month_label = Label(
name="calendar-month",
label=f"{calendar.month_name[self.current_month]}",
h_align="start",
justification="left",
)
# Day abbreviations
days_header = Box(
name="calendar-days-header",
orientation="h",
h_expand=True,
spacing=2
)
day_names = ["S", "M", "T", "W", "T", "F", "S"]
for i, day_name in enumerate(day_names):
# Saturday (6) and Sunday (0) get weekend styling
is_weekend = i == 0 or i == 6
day_label = Label(
name=(
"calendar-day-header-weekend"
if is_weekend
else "calendar-day-header"
),
label=day_name,
h_align="center",
h_expand=True,
)
days_header.add(day_label)
# Calendar grid
self.calendar_grid = Box(
name="calendar-grid",
orientation="v",
spacing=1
)
self.update_calendar()
self.add(self.month_label)
self.add(days_header)
self.add(self.calendar_grid)
# Update calendar daily
invoke_repeater(86400000, self.update_calendar_if_needed) # 24 hours
def update_calendar_if_needed(self):
"""Check if date changed and update calendar if needed."""
now = datetime.datetime.now()
if (
now.month != self.current_month
or now.year != self.current_year
or now.day != self.current_day
):
self.current_month = now.month
self.current_year = now.year
self.current_day = now.day
self.update_calendar()
return True
def update_calendar(self):
"""Update the calendar grid with current month."""
# Clear existing calendar
for child in self.calendar_grid.get_children():
self.calendar_grid.remove(child)
# Update month label
self.month_label.set_label(f"{calendar.month_name[self.current_month]}")
# Get calendar for current month
cal = calendar.monthcalendar(self.current_year, self.current_month)
for week in cal:
week_box = Box(orientation="h", spacing=2, h_expand=True)
for day_index, day in enumerate(week):
if day == 0:
# Empty day
day_label = Label(
name="calendar-day-empty",
label="",
h_align="center",
h_expand=True,
)
else:
# Regular day
is_today = (
day == self.current_day
and self.current_month == datetime.datetime.now().month
and self.current_year == datetime.datetime.now().year
)
# Saturday (6) and Sunday (0) get weekend styling
is_weekend = day_index == 0 or day_index == 6
if is_today:
name = "calendar-day-today"
elif is_weekend:
name = "calendar-day-weekend"
else:
name = "calendar-day"
day_label = Label(
name=name, label=str(day), h_align="center", h_expand=True
)
week_box.add(day_label)
self.calendar_grid.add(week_box)
class CalendarContainer(Box):
def __init__(self, **kwargs):
super().__init__(
orientation="v",
name="calendar-box-widget",
v_expand=True,
size=(170, 170),
v_align="center",
h_align="center",
children=[Calendar()],
**kwargs,
)
class RamInfo(Box):
@staticmethod
def bake_progress_bar(name: str = "progress-bar", size: int = 80, **kwargs):
return CircularProgressBar(
name=name,
start_angle=270,
end_angle=630,
min_value=0,
max_value=100,
size=size,
**kwargs,
)
@staticmethod
def bake_progress_label(text, **kwargs):
return Label(label=text, **kwargs)
def __init__(self, **kwargs):
super().__init__(
layer="bottom",
title="sysinfo",
name="info-box-widget",
visible=True,
size=(170, 170),
h_expand=True,
v_expand=True,
all_visible=True,
**kwargs,
)
self.ram_progress = self.bake_progress_bar(name="progress")
self.ram_label = Label(
label="15%\nRam", justification="center", name="progress-label"
)
self.ram_info_label = Label(
label="ram info", name="info", justification="center", h_align="center"
)
self.progress_container = Box(
name="progress-bar-container",
h_expand=True,
v_expand=True,
orientation="v",
spacing=12,
h_align="center",
v_align="center",
children=[
Box(
children=[
Overlay(
child=self.ram_progress,
tooltip_text="",
overlays=self.ram_label,
),
],
),
Box(
h_align="center",
justification="centre",
orientation="v",
children=[
self.ram_info_label,
],
),
],
)
self.add(self.progress_container)
invoke_repeater(1000, self.update)
def update(self):
mem = psutil.virtual_memory()
self.ram_label.set_label(f" {round(mem.percent):<2} %\nRAM")
used_gb = mem.used / (1024**3)
free_gb = mem.available / (1024**3)
self.ram_info_label.set_markup(
f"<span foreground='#8E8E8E'>Used</span> {round(used_gb, 1)}GB\n<span foreground='#8E8E8E'>Free</span> {round(free_gb, 1)}GB"
)
GLib.idle_add(self.ram_progress.set_value, mem.percent)
# executor.submit(update)
return True
class CpuInfo(Box):
@staticmethod
def bake_progress_bar(name: str = "progress-bar", size: int = 80, **kwargs):
return CircularProgressBar(
name=name,
start_angle=270,
end_angle=630,
min_value=0,
max_value=100,
size=size,
**kwargs,
)
@staticmethod
def bake_progress_label(text, **kwargs):
return Label(label=text, **kwargs)
def __init__(self, **kwargs):
super().__init__(
layer="bottom",
title="sysinfo",
name="info-box-widget",
visible=True,
size=(170, 170),
h_expand=True,
v_expand=True,
all_visible=True,
**kwargs,
)
self.cpu_progress = self.bake_progress_bar(name="progress")
self.cpu_label = Label(
label="15%\nCPU", justification="center", name="progress-label"
)
self.cpu_info_label = Label(
label="CPU Info", name="info", justification="center", h_align="center"
)
self.progress_container = Box(
name="progress-bar-container",
h_expand=True,
v_expand=True,
orientation="v",
spacing=12,
h_align="center",
v_align="center",
children=[
Box(
children=[
Overlay(
child=self.cpu_progress,
tooltip_text="",
overlays=self.cpu_label,
),
],
),
Box(
h_align="center",
justification="centre",
orientation="v",
children=[
self.cpu_info_label,
],
),
],
)
self.add(self.progress_container)
invoke_repeater(1000, self.update)
def get_cpu_temp(self):
temps = psutil.sensors_temperatures()
if not temps:
return None
for name, entries in temps.items():
if (
"coretemp" in name.lower()
or "k10temp" in name.lower()
or "cpu" in name.lower()
):
for entry in entries:
# Match common CPU temp labels
if (
"package id 0" in (entry.label or "").lower()
or "core 0" in (entry.label or "").lower()
or (entry.label is None or entry.label.strip() == "")
):
return round(entry.current, 1)
return None
def update(self):
cpu = psutil.cpu_percent()
self.cpu_label.set_label(f" {round(cpu):<2} %\nCPU")
self.cpu_info_label.set_markup(
f"<span foreground='#8E8E8E'>Temp</span> {self.get_cpu_temp()}°C"
)
GLib.idle_add(self.cpu_progress.set_value, cpu)
# executor.submit(update)
return True
# FIX: GTK ERRORS
class Deskwidgets(Window):
config = load_config()
def __init__(self, **kwargs):
top_left = Window(
anchor="top left",
orientation="h",
layer="bottom",
child=Box(
name="desktop-widgets-container",
children=[
DateContainer(),
WeatherContainer(),
CalendarContainer(),
],
),
)
bottom_left = Window(
anchor="bottom right",
orientation="h",
layer="bottom",
child=Box(
name="desktop-widgets-container",
children=[
CpuInfo(),
RamInfo(),
],
),
)
container = Box(
orientation="v",
children=[
top_left,
bottom_left,
],
)
super().__init__(
name="desktop",
layer="bottom",
title="desktop-widgets",
orientation="v",
exclusivity="none",
child=container,
**kwargs,
)
# else:
#
# class Deskwidgets(Window):
# def __init__(self, **kwargs):
# config = load_config()
# super().__init__(name="desktop", **kwargs)
# desktop_widget = Window(
# layer="bottom",
# anchor="bottom left",
# v_align="start",
# h_align="start",
# h_expand=True,
# v_expand=True,
# justification="left",
# exclusivity="none",
# child=Box(
# orientation="v",
# children=create_widgets(config),
# ),
# all_visible=True,
# )
# if config.get("widgets_activation_visible", True):
# activationnag = Window(
# name="activation",
# anchor="bottom right",
# layer="top",
# justification="left",
# v_align="start",
# h_align="start",
# h_expand=True,
# v_expand=True,
# child=Box(
# orientation="v",
# children=[
# ActivationMainText(),
# ActivationSubText(),
# ],
# ),
# all_visible=True,
# )
# else:
# activationnag = None
#
# else:
#
# class Deskwidgets(Window):
# def __init__(self, **kwargs):
# super().__init__(name="desktop", **kwargs)
# self.set_visible(False)