From 857e7ed5f1a725e1342359b6c071586a3d495702 Mon Sep 17 00:00:00 2001 From: Kevin O'Connor Date: Tue, 8 Jan 2019 11:09:55 -0500 Subject: klippy: Convert printer_state("connect") to an event handler Convert all users of the printer_state("connect") handler to register a "klippy:connect" event handler instead. Signed-off-by: Kevin O'Connor --- klippy/extras/bed_mesh.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'klippy/extras/bed_mesh.py') diff --git a/klippy/extras/bed_mesh.py b/klippy/extras/bed_mesh.py index 1625bbcc..dc6eba33 100644 --- a/klippy/extras/bed_mesh.py +++ b/klippy/extras/bed_mesh.py @@ -53,6 +53,8 @@ class BedMesh: FADE_DISABLE = 0x7FFFFFFF def __init__(self, config): self.printer = config.get_printer() + self.printer.register_event_handler("klippy:connect", + self.handle_connect) self.last_position = [0., 0., 0., 0.] self.calibrate = BedMeshCalibrate(config, self) self.z_mesh = None @@ -74,10 +76,9 @@ class BedMesh: 'BED_MESH_CLEAR', self.cmd_BED_MESH_CLEAR, desc=self.cmd_BED_MESH_CLEAR_help) self.gcode.set_move_transform(self) - def printer_state(self, state): - if state == 'connect': - self.toolhead = self.printer.lookup_object('toolhead') - self.calibrate.load_default_profile() + def handle_connect(self): + self.toolhead = self.printer.lookup_object('toolhead') + self.calibrate.load_default_profile() def set_mesh(self, mesh): if mesh is not None: if self.base_fade_target is None: -- cgit v1.2.3-70-g09d2