import subprocess as sp import threading import time import os import signal import platform import logging import sys import getopt log = logging.getLogger() log.setLevel(logging.DEBUG) out_hdlr = logging.StreamHandler(sys.stdout) out_hdlr.setLevel(logging.INFO) log.addHandler(out_hdlr) arguments = [] try: opts, args = getopt.getopt(sys.argv[1:],"h:",["no-update", "config="]) except getopt.GetoptError: print 'bazarr.py -h --no-update --config ' sys.exit(2) for opt, arg in opts: arguments.append(opt) if arg != '': arguments.append(arg) def start_bazarr(): script = [sys.executable, os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr/main.py'))] + globals()['arguments'] pidfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.pid')) ep = sp.Popen(script, stdout=sp.PIPE, stderr=sp.STDOUT) try: file = open(pidfile,'w') except: logging.error("Error trying to write pid file.") else: file.write(str(ep.pid)) file.close() logging.info("Bazarr starting with process id: " + str(ep.pid) + "...") for line in iter(ep.stdout.readline, ''): sys.stdout.write(line) def shutdown_bazarr(restarting): pidfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.pid')) if os.path.exists(pidfile): try: file = open(pidfile,'r') except: logging.error("Error trying to read pid file.") else: pid = file.read() file.close() try: os.remove(pidfile) except: logging.error("Unable to delete pid file.") else: logging.info('Bazarr stopping...') if restarting is False: stopfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.stop')) file = open(stopfile, 'w') file.write('') file.close() os.kill(int(pid), signal.SIGINT) else: logging.warn("pid file doesn't exist. You must start Bazarr first.") def restart_bazarr(): restartfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.restart')) file = open(restartfile, 'w') file.write('') file.close() # GetExitCodeProcess uses a special exit code to indicate that the process is # still running. _STILL_ACTIVE = 259 def is_pid_running(pid): return (_is_pid_running_on_windows(pid) if platform.system() == "Windows" else _is_pid_running_on_unix(pid)) def _is_pid_running_on_unix(pid): try: os.kill(pid, 0) except OSError: return False return True def _is_pid_running_on_windows(pid): import ctypes.wintypes kernel32 = ctypes.windll.kernel32 handle = kernel32.OpenProcess(1, 0, pid) if handle == 0: return False # If the process exited recently, a pid may still exist for the handle. # So, check if we can get the exit code. exit_code = ctypes.wintypes.DWORD() is_running = ( kernel32.GetExitCodeProcess(handle, ctypes.byref(exit_code)) == 0) kernel32.CloseHandle(handle) # See if we couldn't get the exit code or the exit code indicates that the # process is still running. return is_running or exit_code.value == _STILL_ACTIVE if __name__ == '__main__': pidfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.pid')) restartfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.restart')) stopfile = os.path.normcase(os.path.join(os.path.dirname(__file__), 'bazarr.stop')) try: os.remove(pidfile) os.remove(restartfile) os.remove(stopfile) except: pass def daemon(): threading.Timer(1.0, daemon).start() if os.path.exists(stopfile): try: os.remove(stopfile) except: logging.error('Unable to delete stop file.') else: logging.info('Bazarr exited.') os._exit(0) if os.path.exists(restartfile): try: os.remove(restartfile) except: logging.error('Unable to delete restart file.') else: shutdown_bazarr(True) start_bazarr() if os.path.exists(pidfile): try: file = open(pidfile, 'r') except: logging.error("Error trying to read pid file.") else: pid = int(file.read()) file.close() if is_pid_running(pid) is False: logging.warn('Bazarr unexpectedly exited. Starting it back.') start_bazarr() daemon() start_bazarr() # Keep the script running forever. while True: time.sleep(0.001)