baseimage-tutorial-framework/lib/tfw/components/supervisor.py

37 lines
1.2 KiB
Python
Raw Normal View History

from os import remove
from contextlib import suppress
2018-02-13 16:51:19 +00:00
import xmlrpc.client
from xmlrpc.client import Fault as SupervisorFault
class SupervisorBase:
def __init__(self, supervisor_uri):
self.supervisor = xmlrpc.client.ServerProxy(supervisor_uri).supervisor
2018-02-13 16:51:19 +00:00
class ProcessManager(SupervisorBase):
def stop_process(self, process_name):
2018-02-13 16:51:19 +00:00
with suppress(SupervisorFault):
self.supervisor.stopProcess(process_name)
2018-02-13 16:51:19 +00:00
def start_process(self, process_name):
self.supervisor.startProcess(process_name)
2018-02-13 16:51:19 +00:00
def restart_process(self, process_name):
self.stop_process(process_name)
self.start_process(process_name)
class LogManager(SupervisorBase):
def read_stdout(self, process_name, tail=0):
return self.supervisor.readProcessStdoutLog(process_name, -tail, 0)
def read_stderr(self, process_name, tail=0):
return self.supervisor.readProcessStderrLog(process_name, -tail, 0)
def clear_logs(self, process_name):
for logfile in ('stdout_logfile', 'stderr_logfile'):
with suppress(FileNotFoundError):
2018-05-28 14:36:59 +00:00
remove(self.supervisor.getProcessInfo(process_name)[logfile])
2018-03-08 15:05:44 +00:00
self.supervisor.clearProcessLogs(process_name)