diff options
Diffstat (limited to 'klippy/extras/bed_screws.py')
| -rw-r--r-- | klippy/extras/bed_screws.py | 19 |
1 files changed, 14 insertions, 5 deletions
diff --git a/klippy/extras/bed_screws.py b/klippy/extras/bed_screws.py index 4a6acb83..4749d4e1 100644 --- a/klippy/extras/bed_screws.py +++ b/klippy/extras/bed_screws.py @@ -7,9 +7,7 @@ class BedScrews: def __init__(self, config): self.printer = config.get_printer() - self.state = None - self.current_screw = 0 - self.accepted_screws = 0 + self.reset() self.number_of_screws = 0 # Read config screws = [] @@ -39,6 +37,10 @@ class BedScrews: self.gcode.register_command("BED_SCREWS_ADJUST", self.cmd_BED_SCREWS_ADJUST, desc=self.cmd_BED_SCREWS_ADJUST_help) + def reset(self): + self.state = None + self.current_screw = 0 + self.accepted_screws = 0 def move(self, coord, speed): self.printer.lookup_object('toolhead').manual_move(coord, speed) def move_to_screw(self, state, screw): @@ -64,6 +66,13 @@ class BedScrews: self.gcode.register_command('ACCEPT', None) self.gcode.register_command('ADJUSTED', None) self.gcode.register_command('ABORT', None) + def get_status(self, eventtime): + return { + 'is_active': self.state is not None, + 'state': self.state, + 'current_screw': self.current_screw, + 'accepted_screws': self.accepted_screws + } cmd_BED_SCREWS_ADJUST_help = "Tool to help adjust bed leveling screws" def cmd_BED_SCREWS_ADJUST(self, gcmd): if self.state is not None: @@ -92,7 +101,7 @@ class BedScrews: self.move_to_screw('fine', 0) return # Done - self.state = None + self.reset() self.move((None, None, self.horizontal_move_z), self.lift_speed) gcmd.respond_info("Bed screws tool completed successfully") cmd_ADJUSTED_help = "Accept bed screw position after notable adjustment" @@ -103,7 +112,7 @@ class BedScrews: cmd_ABORT_help = "Abort bed screws tool" def cmd_ABORT(self, gcmd): self.unregister_commands() - self.state = None + self.reset() def load_config(config): return BedScrews(config) |