diff --git a/klippy/extras/adc_temperature.py b/klippy/extras/adc_temperature.py index 086f2858..10825901 100644 --- a/klippy/extras/adc_temperature.py +++ b/klippy/extras/adc_temperature.py @@ -191,7 +191,7 @@ def load_config(config): for sensor_type, params in [("AD595", AD595), ("PT100 INA826", PT100)]: func = (lambda config, params=params: PrinterADCtoTemperature(config, LinearVoltage(config, params))) - pheater.add_sensor(sensor_type, func) + pheater.add_sensor_factory(sensor_type, func) def load_config_prefix(config): if config.get("resistance1", None) is None: @@ -199,4 +199,4 @@ def load_config_prefix(config): else: custom_sensor = CustomLinearResistance(config) pheater = config.get_printer().lookup_object("heater") - pheater.add_sensor(custom_sensor.name, custom_sensor.create) + pheater.add_sensor_factory(custom_sensor.name, custom_sensor.create) diff --git a/klippy/extras/spi_temperature.py b/klippy/extras/spi_temperature.py index 6dbb0b94..ff4b10e4 100644 --- a/klippy/extras/spi_temperature.py +++ b/klippy/extras/spi_temperature.py @@ -342,4 +342,4 @@ def load_config(config): # Register sensors pheater = config.get_printer().lookup_object("heater") for name, klass in Sensors.items(): - pheater.add_sensor(name, klass) + pheater.add_sensor_factory(name, klass) diff --git a/klippy/extras/thermistor.py b/klippy/extras/thermistor.py index 3dff1520..eee14b1c 100644 --- a/klippy/extras/thermistor.py +++ b/klippy/extras/thermistor.py @@ -117,9 +117,9 @@ def load_config(config): pheater = config.get_printer().lookup_object("heater") for sensor_type, params in Sensors.items(): func = (lambda config, params=params: PrinterThermistor(config, params)) - pheater.add_sensor(sensor_type, func) + pheater.add_sensor_factory(sensor_type, func) def load_config_prefix(config): thermistor = CustomThermistor(config) pheater = config.get_printer().lookup_object("heater") - pheater.add_sensor(thermistor.name, thermistor.create) + pheater.add_sensor_factory(thermistor.name, thermistor.create) diff --git a/klippy/heater.py b/klippy/heater.py index 7fb89563..7165582d 100644 --- a/klippy/heater.py +++ b/klippy/heater.py @@ -228,15 +228,15 @@ class ControlPID: class PrinterHeaters: def __init__(self, config): self.printer = config.get_printer() - self.sensors = {} + self.sensor_factories = {} self.heaters = {} self.heaters_gcode_id = {} # Register TURN_OFF_HEATERS command gcode = self.printer.lookup_object('gcode') gcode.register_command("TURN_OFF_HEATERS", self.cmd_TURN_OFF_HEATERS, desc=self.cmd_TURN_OFF_HEATERS_help) - def add_sensor(self, sensor_type, sensor_factory): - self.sensors[sensor_type] = sensor_factory + def add_sensor_factory(self, sensor_type, sensor_factory): + self.sensor_factories[sensor_type] = sensor_factory def setup_heater(self, config, gcode_id): heater_name = config.get_name().split()[-1] if heater_name == 'extruder': @@ -261,10 +261,10 @@ class PrinterHeaters: self.printer.try_load_module(config, "adc_temperature") self.printer.try_load_module(config, "spi_temperature") sensor_type = config.get('sensor_type') - if sensor_type not in self.sensors: - raise self.printer.config_error("Unknown temperature sensor '%s'" % ( - sensor_type,)) - return self.sensors[sensor_type](config) + if sensor_type not in self.sensor_factories: + raise self.printer.config_error( + "Unknown temperature sensor '%s'" % (sensor_type,)) + return self.sensor_factories[sensor_type](config) def get_all_heaters(self): return self.heaters.values() cmd_TURN_OFF_HEATERS_help = "Turn off all heaters"