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

shell_command: add run_with_response() method 

This method collects the entire response and returns it with the call.  Suitable for commands that produce little output.

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Arksine 2020-11-18 09:10:40 -05:00
parent 706d86dd21
commit 116c1e9f29
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
moonraker/plugins/shell_command.py
View File

@ -93,6 +93,20 @@ class ShellCommand:
self.io_loop.remove_handler(fd) self.io_loop.remove_handler(fd)
return complete return complete
async def run_with_response(self, timeout=2.):
result = []
def cb(data):
data = data.strip()
if data:
result.append(data.decode())
prev_cb = self.output_cb
self.output_cb = cb
await self.run(timeout)
self.output_cb = prev_cb
return "\n".join(result)
class ShellCommandFactory: class ShellCommandFactory:
def build_shell_command(self, cmd, callback): def build_shell_command(self, cmd, callback):
return ShellCommand(cmd, callback) return ShellCommand(cmd, callback)