3diy
/
klipper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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>
This commit is contained in:
Kevin O'Connor 2018-09-02 12:51:37 -04:00
parent 50196c7141
commit 18b04ffe68
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
klippy/klippy.py
View File

@ -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')

3
klippy/toolhead.py
View File

@ -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.]