diff options
| -rw-r--r-- | klippy/extras/tmc2130.py | 4 | ||||
| -rw-r--r-- | klippy/extras/tmc2208.py | 4 | ||||
| -rw-r--r-- | klippy/extras/tmc2660.py | 4 | ||||
| -rw-r--r-- | klippy/extras/tmc5160.py | 4 |
4 files changed, 12 insertions, 4 deletions
diff --git a/klippy/extras/tmc2130.py b/klippy/extras/tmc2130.py index 25da59e0..d5130cbf 100644 --- a/klippy/extras/tmc2130.py +++ b/klippy/extras/tmc2130.py @@ -354,7 +354,9 @@ class TMC2130: 'VALUE' not in params): raise gcode.error("Invalid command format") field = gcode.get_str('FIELD', params) - reg = self.fields.field_to_register[field] + reg = self.fields.field_to_register.get(field) + if reg is None: + raise gcode.error("Unknown field name '%s'" % field) value = gcode.get_int('VALUE', params) self.fields.set_field(field, value) print_time = self.printer.lookup_object('toolhead').get_last_move_time() diff --git a/klippy/extras/tmc2208.py b/klippy/extras/tmc2208.py index 4134f026..dcd7b890 100644 --- a/klippy/extras/tmc2208.py +++ b/klippy/extras/tmc2208.py @@ -427,7 +427,9 @@ class TMC2208: 'VALUE' not in params): raise gcode.error("Invalid command format") field = gcode.get_str('FIELD', params) - reg = self.fields.field_to_register[field] + reg = self.fields.field_to_register.get(field) + if reg is None: + raise gcode.error("Unknown field name '%s'" % field) value = gcode.get_int('VALUE', params) self.fields.set_field(field, value) self.printer.lookup_object('toolhead').wait_moves() diff --git a/klippy/extras/tmc2660.py b/klippy/extras/tmc2660.py index 721fbe43..7f0c95d1 100644 --- a/klippy/extras/tmc2660.py +++ b/klippy/extras/tmc2660.py @@ -294,7 +294,9 @@ class TMC2660: 'VALUE' not in params): raise gcode.error("Invalid command format") field = gcode.get_str('FIELD', params) - reg = self.fields.field_to_register[field] + reg = self.fields.field_to_register.get(field) + if reg is None: + raise gcode.error("Unknown field name '%s'" % field) value = gcode.get_int('VALUE', params) self.fields.set_field(field, value) pt = self.printer.lookup_object('toolhead').get_last_move_time() diff --git a/klippy/extras/tmc5160.py b/klippy/extras/tmc5160.py index b7f226b4..70736f2e 100644 --- a/klippy/extras/tmc5160.py +++ b/klippy/extras/tmc5160.py @@ -428,7 +428,9 @@ class TMC5160: 'VALUE' not in params): raise gcode.error("Invalid command format") field = gcode.get_str('FIELD', params) - reg = self.fields.field_to_register[field] + reg = self.fields.field_to_register.get(field) + if reg is None: + raise gcode.error("Unknown field name '%s'" % field) value = gcode.get_int('VALUE', params) self.fields.set_field(field, value) print_time = self.printer.lookup_object('toolhead').get_last_move_time() |