diff options
Diffstat (limited to 'klippy/extras/homing_heaters.py')
| -rw-r--r-- | klippy/extras/homing_heaters.py | 37 |
1 files changed, 22 insertions, 15 deletions
diff --git a/klippy/extras/homing_heaters.py b/klippy/extras/homing_heaters.py index 4bb9ba04..e9b37f99 100644 --- a/klippy/extras/homing_heaters.py +++ b/klippy/extras/homing_heaters.py @@ -6,18 +6,20 @@ import logging + class HomingHeaters: def __init__(self, config): self.printer = config.get_printer() - self.printer.register_event_handler("klippy:connect", - self.handle_connect) - self.printer.register_event_handler("homing:homing_move_begin", - self.handle_homing_move_begin) - self.printer.register_event_handler("homing:homing_move_end", - self.handle_homing_move_end) + self.printer.register_event_handler("klippy:connect", self.handle_connect) + self.printer.register_event_handler( + "homing:homing_move_begin", self.handle_homing_move_begin + ) + self.printer.register_event_handler( + "homing:homing_move_end", self.handle_homing_move_end + ) self.disable_heaters = config.getlist("heaters", None) self.flaky_steppers = config.getlist("steppers", None) - self.pheaters = self.printer.load_object(config, 'heaters') + self.pheaters = self.printer.load_object(config, "heaters") self.target_save = {} def handle_connect(self): @@ -29,30 +31,34 @@ class HomingHeaters: if not all(x in all_heaters for x in self.disable_heaters): raise self.printer.config_error( "One or more of these heaters are unknown: %s" - % (self.disable_heaters,)) + % (self.disable_heaters,) + ) # steppers valid? - kin = self.printer.lookup_object('toolhead').get_kinematics() + kin = self.printer.lookup_object("toolhead").get_kinematics() all_steppers = [s.get_name() for s in kin.get_steppers()] if self.flaky_steppers is None: return if not all(x in all_steppers for x in self.flaky_steppers): raise self.printer.config_error( - "One or more of these steppers are unknown: %s" - % (self.flaky_steppers,)) + "One or more of these steppers are unknown: %s" % (self.flaky_steppers,) + ) + def check_eligible(self, endstops): if self.flaky_steppers is None: return True - steppers_being_homed = [s.get_name() - for es in endstops - for s in es.get_steppers()] + steppers_being_homed = [ + s.get_name() for es in endstops for s in es.get_steppers() + ] return any(x in self.flaky_steppers for x in steppers_being_homed) + def handle_homing_move_begin(self, hmove): if not self.check_eligible(hmove.get_mcu_endstops()): return for heater_name in self.disable_heaters: heater = self.pheaters.lookup_heater(heater_name) self.target_save[heater_name] = heater.get_temp(0)[1] - heater.set_temp(0.) + heater.set_temp(0.0) + def handle_homing_move_end(self, hmove): if not self.check_eligible(hmove.get_mcu_endstops()): return @@ -60,5 +66,6 @@ class HomingHeaters: heater = self.pheaters.lookup_heater(heater_name) heater.set_temp(self.target_save[heater_name]) + def load_config(config): return HomingHeaters(config) |