#!/usr/bin/python3 """Saver module for xsecurelock. It displays a background image, clock and weather. Configuration is done through environment variables: - XSECURELOCK_SAVER_IMAGE: path to the background image to use - XSECURELOCK_SAVER_WEATHER: path to weather text - XSECURELOCK_SAVER_FONT: font family to use to display clock and weather - XSECURELOCK_SAVER_CLOCK_FONT_SIZE: font size to use to display clock - XSECURELOCK_SAVER_WEATHER_FONT_SIZE: font size to use to display weather """ import os import types import datetime import re import gi gi.require_version("Gtk", "3.0") from gi.repository import Gtk, Gdk, GdkX11, GLib, GdkPixbuf, Gio import cairo def on_win_realize(widget, ctx): """On realization, embed into XSCREENSAVER_WINDOW and remember parent position.""" parent_wid = int(os.getenv("XSCREENSAVER_WINDOW", 0)) if not parent_wid: return parent = GdkX11.X11Window.foreign_new_for_display(widget.get_display(), parent_wid) x, y, w, h = parent.get_geometry() ctx.position = x, y window = widget.get_window() window.resize(w, h) window.reparent(parent, 0, 0) def on_win_draw(widget, cctx, ctx): """Draw background image.""" cctx.set_operator(cairo.OPERATOR_SOURCE) if not ctx.background: cctx.set_source_rgba(0, 0, 0, 1) cctx.paint() return x, y = ctx.position wwidth, wheight = widget.get_size() scale = widget.get_scale_factor() bg = ctx.background.new_subpixbuf( x * scale, y * scale, wwidth * scale, wheight * scale ) cctx.save() cctx.scale(1 / scale, 1 / scale) Gdk.cairo_set_source_pixbuf(cctx, bg, 0, 0) cctx.paint() cctx.restore() def on_overlay_draw(widget, cctx, ctx): """Draw overlay.""" # Clock cctx.set_operator(cairo.OPERATOR_SOURCE) wwidth, wheight = widget.get_parent().get_size() now = datetime.datetime.now().strftime("%H:%M") cctx.select_font_face( ctx.font_family, cairo.FONT_SLANT_NORMAL, cairo.FONT_WEIGHT_BOLD ) cctx.set_font_size(ctx.clock_font_size) _, _, twidth, theight, _, _ = cctx.text_extents("00:00") text_position = wwidth // 2 - twidth // 2, wheight // 3 - theight // 2 cctx.move_to(*text_position) cctx.set_source_rgba(1, 1, 1, 0.8) cctx.show_text(now) cctx.move_to(*text_position) cctx.set_source_rgb(0, 0, 0) cctx.set_line_width(2) cctx.text_path(now) cctx.stroke() # Weather try: with open(ctx.weather_file) as wfile: data = wfile.read() except Exception: return # We can have polybar markups in it. We assume %{Tx} means to use # Font Awesome 6 and we ignore font color change. The parsing is # quite basic. data = re.sub(r'%{F[#\d+-]+?}', '', data) data = re.split(r'(%{T[1-9-]})', data) font = ctx.font_family cctx.move_to(20, wheight - 20) for chunk in data: if chunk == "%{T-}": font = ctx.font_family continue elif chunk.startswith("%{T"): font = "Font Awesome 6 Pro" continue elif not chunk: continue cctx.select_font_face(font, cairo.FONT_SLANT_NORMAL, cairo.FONT_WEIGHT_NORMAL) cctx.set_font_size(ctx.weather_font_size) cur_position = cctx.get_current_point() cctx.set_source_rgb(0, 0, 0) cctx.set_line_width(1) cctx.text_path(chunk) cctx.stroke() cctx.move_to(*cur_position) cctx.set_source_rgba(1, 1, 1, 0.8) cctx.show_text(chunk) def on_background_change(monitor, f1, f2, event, ctx): """Update background when changed.""" if event not in ( Gio.FileMonitorEvent.CHANGES_DONE_HINT, Gio.FileMonitorEvent.RENAMED, ): return try: new_background = GdkPixbuf.Pixbuf.new_from_file(ctx.background_image) except Exception: return ctx.background = new_background ctx.window.queue_draw() def on_clock_change(ctx): ctx.overlay.queue_draw() now = datetime.datetime.now() GLib.timeout_add((60 - now.second) * 1000, on_clock_change, ctx) def on_weather_change(monitor, f1, f2, event, ctx): if event not in ( Gio.FileMonitorEvent.CHANGES_DONE_HINT, Gio.FileMonitorEvent.RENAMED, ): return ctx.overlay.queue_draw() if __name__ == "__main__": ctx = types.SimpleNamespace() ctx.background_image = os.getenv("XSECURELOCK_SAVER_IMAGE", None) ctx.clock_font_size = int(os.getenv("XSECURELOCK_SAVER_CLOCK_FONT_SIZE", 120)) ctx.weather_font_size = int(os.getenv("XSECURELOCK_SAVER_CLOCK_FONT_SIZE", 40)) ctx.weather_file = os.getenv("XSECURELOCK_SAVER_WEATHER", None) ctx.font_family = os.getenv("XSECURELOCK_SAVER_FONT", "Iosevka Aile") ctx.background = None ctx.position = [0, 0] ctx.window = Gtk.Window() ctx.window.set_app_paintable(True) ctx.window.set_visual(ctx.window.get_screen().get_rgba_visual()) ctx.window.connect("realize", on_win_realize, ctx) ctx.window.connect("draw", on_win_draw, ctx) ctx.window.connect("delete-event", Gtk.main_quit) ctx.overlay = Gtk.DrawingArea() ctx.overlay.connect("draw", on_overlay_draw, ctx) ctx.window.add(ctx.overlay) on_clock_change(ctx) if ctx.background_image: gfile = Gio.File.new_for_path(ctx.background_image) monitor = gfile.monitor_file(Gio.FileMonitorFlags.WATCH_MOVES, None) monitor.connect("changed", on_background_change, ctx) on_background_change(None, None, None, Gio.FileMonitorEvent.CHANGES_DONE_HINT, ctx) if ctx.weather_file: gfile = Gio.File.new_for_path(ctx.weather_file) monitor = gfile.monitor_file(Gio.FileMonitorFlags.WATCH_MOVES, None) monitor.connect("changed", on_weather_change, ctx) ctx.window.show_all() # Main loop Gtk.main()