klippy: Rename lookup_module_objects() to lookup_objects()

Rename the method and support returning all known objects.

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

2 files changed, 11 insertions, 7 deletions

diff --git a/klippy/klippy.py b/klippy/klippy.py
index d3fedf10..d5dc8dae 100644
--- a/klippy/klippy.py
+++ b/klippy/klippy.py
@@ -164,11 +164,14 @@ class Printer:
         if default is ConfigWrapper.sentinel:
             raise self.config_error("Unknown config object '%s'" % (name,))
         return default
-    def lookup_module_objects(self, module_name):
-        prefix = module_name + ' '
-        objs = [self.objects[n] for n in self.objects if n.startswith(prefix)]
-        if module_name in self.objects:
-            return [self.objects[module_name]] + objs
+    def lookup_objects(self, module=None):
+        if module is None:
+            return list(self.objects.items())
+        prefix = module + ' '
+        objs = [(n, self.objects[n])
+                for n in self.objects if n.startswith(prefix)]
+        if module in self.objects:
+            return [(module, self.objects[module])] + objs
         return objs
     def set_rollover_info(self, name, info):
         logging.info(info)
@@ -280,7 +283,7 @@ class Printer:
             try:
                 self._stats(self.reactor.monotonic(), force_output=True)
                 if run_result == 'firmware_restart':
-                    for m in self.lookup_module_objects('mcu'):
+                    for n, m in self.lookup_objects(module='mcu'):
                         m.microcontroller_restart()
                 for cb in self.state_cb:
                     cb('disconnect')
diff --git a/klippy/toolhead.py b/klippy/toolhead.py
index abc0874a..71a2b7e6 100644
--- a/klippy/toolhead.py
+++ b/klippy/toolhead.py
@@ -196,7 +196,8 @@ class ToolHead:
     def __init__(self, config):
         self.printer = config.get_printer()
         self.reactor = self.printer.get_reactor()
-        self.all_mcus = self.printer.lookup_module_objects('mcu')
+        self.all_mcus = [
+            m for n, m in self.printer.lookup_objects(module='mcu')]
         self.mcu = self.all_mcus[0]
         self.move_queue = MoveQueue()
         self.commanded_pos = [0., 0., 0., 0.]