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

shell_command: add "log_complete" argument to run methods 

This brings consistency to the "run" methods.  It also removes potential confusion between the previous "quiet" argument and the "verbose" argument in "run()".

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Arksine 2021-03-06 19:25:14 -05:00
parent 44f1147f71
commit 5d095125d3
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moonraker/plugins/proc_stats.py
View File

@ -103,7 +103,7 @@ class ProcStats:
async def _check_throttled_state(self):
try:
resp = await self.vcgencmd.run_with_response(
timeout=.5, quiet=True)
timeout=.5, log_complete=False)
ts = int(resp.strip().split("=")[-1], 16)
except Exception:
return {'bits': 0, 'flags': ["?"]}

13
moonraker/plugins/shell_command.py
View File

@ -102,7 +102,7 @@ class ShellCommand:
def get_return_code(self):
return self.return_code
async def run(self, timeout=2., verbose=True):
async def run(self, timeout=2., verbose=True, log_complete=True):
self.return_code = self.proc = None
self.cancelled = False
if not timeout:
@ -124,9 +124,10 @@ class ShellCommand:
await self.proc.cancel()
else:
complete = not self.cancelled
return self._check_proc_success(complete)
return self._check_proc_success(complete, log_complete)
async def run_with_response(self, timeout=2., retries=1, quiet=False):
async def run_with_response(self, timeout=2., retries=1,
log_complete=True):
self.return_code = self.proc = None
self.cancelled = False
while retries > 0:
@ -143,7 +144,7 @@ class ShellCommand:
complete = not self.cancelled
if self.log_stderr and stderr:
logging.info(f"{self.command[0]}: {stderr.decode()}")
if self._check_proc_success(complete, quiet):
if self._check_proc_success(complete, log_complete):
return stdout.decode().rstrip("\n")
elif stdout:
logging.debug(
@ -175,7 +176,7 @@ class ShellCommand:
return False
return True
def _check_proc_success(self, complete, quiet=False):
def _check_proc_success(self, complete, log_complete):
self.return_code = self.proc.returncode
success = self.return_code == 0 and complete
if success:
@ -187,7 +188,7 @@ class ShellCommand:
else:
msg = f"Command ({self.name}) exited with return code" \
f" {self.return_code}"
if not quiet:
if log_complete:
logging.info(msg)
return success