bazarr/bazarr.py

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 <config_directory>'
    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'))
    if os.path.exists(pidfile):
        logging.error("Bazarr already running, please stop it first.")
    else:
        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)