i3pystatus/i3pystatus/shell.py

39 lines
999 B
Python

from i3pystatus import IntervalModule
from i3pystatus.core.command import run_through_shell
import logging
# logger = logging.getLogger(__name__)
class Shell(IntervalModule):
"""
Shows output of shell command
"""
color = "#FFFFFF"
error_color = "#FF0000"
settings = (
("command", "command to be executed"),
("color", "standard color"),
("error_color", "color to use when non zero exit code is returned")
)
required = ("command",)
def run(self):
retvalue, out, stderr = run_through_shell(self.command, enable_shell=True)
if retvalue != 0:
self.logger.error(stderr if stderr else "Unknown error")
if out:
out.replace("\n", " ").strip()
elif stderr:
out = stderr
self.output = {
"full_text": out if out else "Command `%s` returned %d" % (self.command, retvalue),
"color": self.color if retvalue == 0 else self.error_color
}