From 18b04ffe6815987284130337e5eb4f46f3ba1c69 Mon Sep 17 00:00:00 2001 From: Kevin O'Connor Date: Sun, 2 Sep 2018 12:51:37 -0400 Subject: klippy: Rename lookup_module_objects() to lookup_objects() Rename the method and support returning all known objects. Signed-off-by: Kevin O'Connor --- klippy/klippy.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'klippy/klippy.py') 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') -- cgit v1.2.3-70-g09d2