From 74d527776e9d37f285e9e51f6552ea27cd611a6e Mon Sep 17 00:00:00 2001 From: Arksine <9563098+Arksine@users.noreply.github.com> Date: Thu, 1 Apr 2021 06:33:14 -0400 Subject: [PATCH] bed_mesh: lookup the toolhead object in "handle_connect" Signed-off-by: Eric Callahan --- klippy/extras/bed_mesh.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/klippy/extras/bed_mesh.py b/klippy/extras/bed_mesh.py index e78ebaeb..8151fe97 100644 --- a/klippy/extras/bed_mesh.py +++ b/klippy/extras/bed_mesh.py @@ -59,6 +59,8 @@ class BedMesh: self.printer = config.get_printer() self.printer.register_event_handler("klippy:ready", self.handle_ready) + self.printer.register_event_handler("klippy:connect", + self.handle_connect) self.last_position = [0., 0., 0., 0.] self.bmc = BedMeshCalibrate(config, self) self.z_mesh = None @@ -90,8 +92,9 @@ class BedMesh: # Register transform gcode_move = self.printer.load_object(config, 'gcode_move') gcode_move.set_move_transform(self) - def handle_ready(self): + def handle_connect(self): self.toolhead = self.printer.lookup_object('toolhead') + def handle_ready(self): self.bmc.print_generated_points(logging.info) self.pmgr.initialize() def set_mesh(self, mesh):