You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
bazarr/bazarr.py

172 lines
5.0 KiB

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:
6 years ago
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)