From 92a4263e4f973fcc92e2106b65d843d5c28636fa Mon Sep 17 00:00:00 2001 From: Timofey Titovets Date: Tue, 2 Dec 2025 22:00:36 +0100 Subject: [PATCH] stepper_enable: refactor to mux commands Signed-off-by: Timofey Titovets --- klippy/extras/stepper_enable.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/klippy/extras/stepper_enable.py b/klippy/extras/stepper_enable.py index 926e95922..ba2968bcb 100644 --- a/klippy/extras/stepper_enable.py +++ b/klippy/extras/stepper_enable.py @@ -81,11 +81,12 @@ class PrinterStepperEnable: gcode = self.printer.lookup_object('gcode') gcode.register_command("M18", self.cmd_M18) gcode.register_command("M84", self.cmd_M18) - gcode.register_command("SET_STEPPER_ENABLE", - self.cmd_SET_STEPPER_ENABLE, - desc=self.cmd_SET_STEPPER_ENABLE_help) def register_stepper(self, config, mcu_stepper): name = mcu_stepper.get_name() + gcode = self.printer.lookup_object('gcode') + gcode.register_mux_command('SET_STEPPER_ENABLE', "STEPPER", name, + self.cmd_SET_STEPPER_ENABLE, + desc=self.cmd_SET_STEPPER_ENABLE_help) enable = setup_enable_pin(self.printer, config.get('enable_pin', None)) self.enable_lines[name] = EnableTracking(mcu_stepper, enable) def set_motors_enable(self, stepper_names, enable): @@ -128,11 +129,7 @@ class PrinterStepperEnable: self.motor_off() cmd_SET_STEPPER_ENABLE_help = "Enable/disable individual stepper by name" def cmd_SET_STEPPER_ENABLE(self, gcmd): - stepper_name = gcmd.get('STEPPER', None) - if stepper_name not in self.enable_lines: - gcmd.respond_info('SET_STEPPER_ENABLE: Invalid stepper "%s"' - % (stepper_name,)) - return + stepper_name = gcmd.get('STEPPER') stepper_enable = gcmd.get_int('ENABLE', 1) self.set_motors_enable([stepper_name], stepper_enable) if stepper_enable: