#!/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 """ # In case I want to put a video instead of an image: # https://gist.github.com/NBonaparte/89fb1b645c99470bc0f6. Check # `bin/wallpaper' for some sources. 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 and weather).""" # Clock cctx.set_operator(cairo.OPERATOR_SOURCE) wwidth, wheight = widget.get_parent().get_size() if ctx.clock: now = ctx.clock 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(re.sub(r"\d", "8", now)) 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 # 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. if ctx.weather: data = re.sub(r"%{F[#\d+-]+?}", "", ctx.weather) 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): """Clock may have changed. Update it. We are checking more often than once a minute because we want to update the clock swiftly after suspend. An alternative would be to listen to PrepareForSleep signal from org.freedesktop.login1, but this is more complex. """ now = datetime.datetime.now() new_clock = now.strftime("%H:%M") if new_clock != ctx.clock: ctx.clock = new_clock ctx.overlay.queue_draw() GLib.timeout_add(min(60 - now.second, 3) * 1000, on_clock_change, ctx) def on_weather_change(monitor, f1, f2, event, ctx): """Weather file has changed.""" if event not in ( Gio.FileMonitorEvent.CHANGES_DONE_HINT, Gio.FileMonitorEvent.RENAMED, ): return try: with open(ctx.weather_file) as wfile: ctx.weather = wfile.read() ctx.overlay.queue_draw() except Exception: pass 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.weather = "" ctx.clock = "" 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) gio_event_args = (None, None, None, Gio.FileMonitorEvent.CHANGES_DONE_HINT) 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(*gio_event_args, 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) GLib.timeout_add(1000, on_weather_change, *gio_event_args, ctx) GLib.timeout_add(1000, on_clock_change, ctx) ctx.window.show_all() # Main loop Gtk.main()