fan: Allow heater_fan to work with heater_bed

Fix order of init error preventing heater_fan from being used with
heater_bed.

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 5 insertions, 6 deletions

diff --git a/klippy/extruder.py b/klippy/extruder.py
index 45ed3bfc..44315f9b 100644
--- a/klippy/extruder.py
+++ b/klippy/extruder.py
@@ -252,11 +252,10 @@ def get_printer_extruders(printer):
     return out
 
 def get_printer_heater(printer, name):
-    if name == 'heater_bed':
-        return printer.objects.get(name)
+    if name == 'heater_bed' and name in printer.objects:
+        return printer.objects[name]
     if name == 'extruder':
         name = 'extruder0'
-    extruder = printer.objects.get(name)
-    if extruder is None:
-        return None
-    return extruder.get_heater()
+    if name.startswith('extruder') and name in printer.objects:
+        return printer.objects[name].get_heater()
+    raise printer.config_error("Unknown heater '%s'" % (name,))