From d161c3f1e425691bc9df37fdce3f64e3048052a6 Mon Sep 17 00:00:00 2001
From: alfrix <alfredomonclus@gmail.com>
Date: Tue, 29 Nov 2022 22:46:06 -0300
Subject: [PATCH] splash_screen: use initialized for klipper restart buttons

---
 panels/splash_screen.py | 2 +-
 screen.py               | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/panels/splash_screen.py b/panels/splash_screen.py
index e76fffd3..17d72ecf 100644
--- a/panels/splash_screen.py
+++ b/panels/splash_screen.py
@@ -79,7 +79,7 @@ class SplashScreenPanel(ScreenPanel):
                 logging.info(f"Associated power devices: {power_devices}")
                 self.add_power_button(self._screen.search_power_devices(power_devices))
 
-        if self._screen._ws and self._screen._ws.connected:
+        if self._screen.initialized:
             self.labels['actions'].add(self.labels['restart'])
             self.labels['actions'].add(self.labels['firmware_restart'])
         else:
diff --git a/screen.py b/screen.py
index ea06ba7a..4f9d0f14 100755
--- a/screen.py
+++ b/screen.py
@@ -85,6 +85,7 @@ class KlipperScreen(Gtk.Window):
     screensaver_timeout = None
     reinit_count = 0
     max_retries = 4
+    initialized = False
 
     def __init__(self, args, version):
         try:
@@ -617,11 +618,13 @@ class KlipperScreen(Gtk.Window):
         self.files = None
         self.printer.reset()
         self.printer = None
+        self.initialized = False
         self.connect_printer(self.connecting_to_printer)
 
     def state_disconnected(self):
         logging.debug("### Going to disconnected")
         self.close_screensaver()
+        self.initialized = False
         self.printer_initializing(_("Klipper has disconnected"), remove=True)
 
     def state_error(self):
@@ -862,6 +865,7 @@ class KlipperScreen(Gtk.Window):
         self.files.refresh_files()
 
         logging.info("Printer initialized")
+        self.initialized = True
         self.reinit_count = 0
 
     def base_panel_show_all(self):