gtk template: break screen

Author: deltragon

safeeyes/glade/break_screen.glade

@@ -18,12 +18,13 @@
 ~ You should have received a copy of the GNU General Public License
 ~ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 -->
-<interface>
+<interface domain="safeyees">
   <requires lib="gtk" version="4.0"/>
-  <object class="GtkWindow" id="window_main">
+  <template parent="GtkWindow" class="BreakScreenWindow">
     <property name="icon-name">io.github.slgobinath.SafeEyes</property>
     <property name="decorated">0</property>
     <property name="deletable">0</property>
+    <signal name="close-request" handler="on_window_delete" swapped="no"/>
     <child>
       <object class="GtkGrid" id="grid1">
         <property name="row_homogeneous">1</property>
@@ -174,5 +175,5 @@
     <style>
       <class name="window_main"/>
     </style>
-  </object>
+  </template>
 </interface>

safeeyes/ui/break_screen.py

@@ -154,7 +154,7 @@ def __show_break_screen(self, message, image_path, widget, tray_actions):
         i = 0
 
         for monitor in monitors:
-            w = BreakScreenWindow(
+            window = BreakScreenWindow(
                 self.application,
                 message,
                 image_path,
@@ -167,8 +167,6 @@ def __show_break_screen(self, message, image_path, widget, tray_actions):
                 self.on_skip_clicked,
             )
 
-            window = w.window
-
             if self.context["is_wayland"]:
                 # Note: in theory, this could also be used on X11
                 # however, that already has its own implementation below
@@ -179,7 +177,7 @@ def __show_break_screen(self, message, image_path, widget, tray_actions):
 
             window.set_title("SafeEyes-" + str(i))
 
-            self.windows.append(w)
+            self.windows.append(window)
 
             if self.context["desktop"] == "kde":
                 # Fix flickering screen in KDE by setting opacity to 1
@@ -298,16 +296,26 @@ def __release_keyboard_x11(self):
     def __destroy_all_screens(self):
         """Close all the break screens."""
         for win in self.windows:
-            win.window.destroy()
+            win.destroy()
         del self.windows[:]
 
 
-class BreakScreenWindow:
+@Gtk.Template(filename=BREAK_SCREEN_GLADE)
+class BreakScreenWindow(Gtk.Window):
     """This class manages the UI for the break screen window.
 
     Each instance is a single window, covering a single monitor.
     """
 
+    __gtype_name__ = "BreakScreenWindow"
+
+    lbl_message = Gtk.Template.Child()
+    lbl_count = Gtk.Template.Child()
+    lbl_widget = Gtk.Template.Child()
+    img_break = Gtk.Template.Child()
+    box_buttons = Gtk.Template.Child()
+    toolbar = Gtk.Template.Child()
+
     def __init__(
         self,
         application,
@@ -321,21 +329,9 @@ def __init__(
         show_skip,
         on_skip,
     ):
-        self.on_close = on_close
+        super().__init__(application=application)
 
-        builder = Gtk.Builder()
-        builder.add_from_file(BREAK_SCREEN_GLADE)
-
-        self.window = builder.get_object("window_main")
-        self.window.set_application(application)
-        self.window.connect("close-request", self.on_window_delete)
-
-        lbl_message = builder.get_object("lbl_message")
-        self.lbl_count = builder.get_object("lbl_count")
-        lbl_widget = builder.get_object("lbl_widget")
-        img_break = builder.get_object("img_break")
-        box_buttons = builder.get_object("box_buttons")
-        toolbar = builder.get_object("toolbar")
+        self.on_close = on_close
 
         for tray_action in tray_actions:
             # TODO: apparently, this would be better served with an icon theme
@@ -350,7 +346,7 @@ def __init__(
                 tray_action,
             )
             toolbar_button.set_tooltip_text(_(tray_action.name))
-            toolbar.append(toolbar_button)
+            self.toolbar.append(toolbar_button)
             toolbar_button.show()
 
         # Add the buttons
@@ -360,21 +356,21 @@ def __init__(
             btn_postpone.get_style_context().add_class("btn_postpone")
             btn_postpone.connect("clicked", on_postpone)
             btn_postpone.set_visible(True)
-            box_buttons.append(btn_postpone)
+            self.box_buttons.append(btn_postpone)
 
         if show_skip:
             # Add the skip button
             btn_skip = Gtk.Button.new_with_label(_("Skip"))
             btn_skip.get_style_context().add_class("btn_skip")
             btn_skip.connect("clicked", on_skip)
             btn_skip.set_visible(True)
-            box_buttons.append(btn_skip)
+            self.box_buttons.append(btn_skip)
 
         # Set values
         if image_path:
-            img_break.set_from_file(image_path)
-        lbl_message.set_label(message)
-        lbl_widget.set_markup(widget)
+            self.img_break.set_from_file(image_path)
+        self.lbl_message.set_label(message)
+        self.lbl_widget.set_markup(widget)
 
     def set_count_down(self, count):
         self.lbl_count.set_text(count)
@@ -389,6 +385,7 @@ def __tray_action(self, button, tray_action: TrayAction):
             tray_action.reset()
         tray_action.action()
 
+    @Gtk.Template.Callback()
     def on_window_delete(self, *args):
         """Window close event handler."""
         logging.info("Closing the break screen")