30 lines
852 B
Python
30 lines
852 B
Python
|
"""Wrapper Class to handle reboots"""
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
class Wrapper:
|
||
|
|
def __init__(self, args, stdout):
|
||
|
|
"""
|
||
|
|
Wraps another program and helps with restarting should the program request.
|
||
|
|
:param args: Arguments used to run program
|
||
|
|
:param stdout: The stdout stream to send output to
|
||
|
|
"""
|
||
|
|
self._args = args
|
||
|
|
self.process = None
|
||
|
|
self._stdout = stdout
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
"""
|
||
|
|
Start the wrapper.
|
||
|
|
:return: None
|
||
|
|
"""
|
||
|
|
self.process = subprocess.Popen(self._args, stdout=self._stdout)
|
||
|
|
while True:
|
||
|
|
while self.process.poll() is None:
|
||
|
|
pass
|
||
|
|
if self.process.returncode == 26:
|
||
|
|
self.process = subprocess.Popen(self._args, stdout=self._stdout)
|
||
|
|
else:
|
||
|
|
sys.exit()
|