klippy: stats() method must check that mcu and toolhead classes exist

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 10 insertions, 5 deletions

diff --git a/klippy/klippy.py b/klippy/klippy.py
index 361075b8..978d6447 100644
--- a/klippy/klippy.py
+++ b/klippy/klippy.py
@@ -99,8 +99,10 @@ class Printer:
         out = []
         out.append(self.gcode.stats(eventtime))
         toolhead = self.objects.get('toolhead')
-        out.append(toolhead.stats(eventtime))
-        out.append(self.mcu.stats(eventtime))
+        if toolhead is not None:
+            out.append(toolhead.stats(eventtime))
+        if self.mcu is not None:
+            out.append(self.mcu.stats(eventtime))
         logging.info("Stats %.0f: %s" % (eventtime, ' '.join(out)))
         return eventtime + 1.
     def load_config(self):
@@ -186,9 +188,12 @@ class Printer:
         self.gcode.set_printer_ready(False)
         self.gcode.motor_heater_off()
     def disconnect(self):
-        if self.mcu is not None:
-            self.stats(time.time())
-            self.mcu.disconnect()
+        try:
+            if self.mcu is not None:
+                self.stats(time.time())
+                self.mcu.disconnect()
+        except:
+            logging.exception("Unhandled exception during disconnect")
     def request_restart(self):
         self.run_result = "restart"
         self.reactor.end()