mirror of
https://github.com/morpheus65535/bazarr
synced 2025-03-03 10:06:24 +00:00
Simplify daemon mechanism.
This commit is contained in:
parent
e4460b622f
commit
2c6a4583d0
2 changed files with 46 additions and 137 deletions
164
bazarr.py
164
bazarr.py
|
@ -2,28 +2,25 @@
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import platform
|
import platform
|
||||||
import signal
|
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
import time
|
import time
|
||||||
|
import atexit
|
||||||
|
|
||||||
from bazarr.get_args import args
|
from bazarr.get_args import args
|
||||||
from libs.six import PY3
|
|
||||||
|
|
||||||
|
|
||||||
def check_python_version():
|
def check_python_version():
|
||||||
python_version = platform.python_version_tuple()
|
python_version = platform.python_version_tuple()
|
||||||
minimum_py2_tuple = (2, 7, 13)
|
|
||||||
minimum_py3_tuple = (3, 7, 0)
|
minimum_py3_tuple = (3, 7, 0)
|
||||||
minimum_py2_str = ".".join(str(i) for i in minimum_py2_tuple)
|
|
||||||
minimum_py3_str = ".".join(str(i) for i in minimum_py3_tuple)
|
minimum_py3_str = ".".join(str(i) for i in minimum_py3_tuple)
|
||||||
|
|
||||||
if int(python_version[0]) < 3:
|
if int(python_version[0]) < minimum_py3_tuple[0]:
|
||||||
print("Python " + minimum_py3_str + " or greater required. "
|
print("Python " + minimum_py3_str + " or greater required. "
|
||||||
"Current version is " + platform.python_version() + ". Please upgrade Python.")
|
"Current version is " + platform.python_version() + ". Please upgrade Python.")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
elif (int(python_version[0]) == minimum_py3_tuple[0] and int(python_version[1]) < minimum_py3_tuple[1]) or \
|
elif (int(python_version[0]) == minimum_py3_tuple[0] and int(python_version[1]) < minimum_py3_tuple[1]) or \
|
||||||
(int(python_version[0]) != minimum_py3_tuple[0] and int(python_version[0]) != minimum_py2_tuple[0]):
|
(int(python_version[0]) != minimum_py3_tuple[0]):
|
||||||
print("Python " + minimum_py3_str + " or greater required. "
|
print("Python " + minimum_py3_str + " or greater required. "
|
||||||
"Current version is " + platform.python_version() + ". Please upgrade Python.")
|
"Current version is " + platform.python_version() + ". Please upgrade Python.")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
@ -34,138 +31,55 @@ check_python_version()
|
||||||
dir_name = os.path.dirname(__file__)
|
dir_name = os.path.dirname(__file__)
|
||||||
|
|
||||||
|
|
||||||
class ProcessRegistry:
|
def start_bazarr():
|
||||||
|
|
||||||
def register(self, process):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def unregister(self, process):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class DaemonStatus(ProcessRegistry):
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.__should_stop = False
|
|
||||||
self.__processes = set()
|
|
||||||
|
|
||||||
def register(self, process):
|
|
||||||
self.__processes.add(process)
|
|
||||||
|
|
||||||
def unregister(self, process):
|
|
||||||
self.__processes.remove(process)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def __send_signal(processes, signal_no, live_processes=None):
|
|
||||||
"""
|
|
||||||
Sends to every single of the specified processes the given signal and (if live_processes is not None) append to
|
|
||||||
it processes which are still alive.
|
|
||||||
"""
|
|
||||||
for ep in processes:
|
|
||||||
if ep.poll() is None:
|
|
||||||
if live_processes is not None:
|
|
||||||
live_processes.append(ep)
|
|
||||||
try:
|
|
||||||
ep.send_signal(signal_no)
|
|
||||||
except Exception as e:
|
|
||||||
print('Failed sending signal %s to process %s because of an unexpected error: %s' % (
|
|
||||||
signal_no, ep.pid, e))
|
|
||||||
return live_processes
|
|
||||||
|
|
||||||
def stop(self):
|
|
||||||
"""
|
|
||||||
Flags this instance as should stop and terminates as smoothly as possible children processes.
|
|
||||||
"""
|
|
||||||
self.__should_stop = True
|
|
||||||
DaemonStatus.__send_signal(self.__processes, signal.SIGINT, list())
|
|
||||||
|
|
||||||
def force_stop(self):
|
|
||||||
self.__should_stop = True
|
|
||||||
DaemonStatus.__send_signal(self.__processes, signal.SIGTERM)
|
|
||||||
|
|
||||||
def should_stop(self):
|
|
||||||
return self.__should_stop
|
|
||||||
|
|
||||||
|
|
||||||
def start_bazarr(process_registry=ProcessRegistry()):
|
|
||||||
script = [sys.executable, "-u", os.path.normcase(os.path.join(dir_name, 'bazarr', 'main.py'))] + sys.argv[1:]
|
script = [sys.executable, "-u", os.path.normcase(os.path.join(dir_name, 'bazarr', 'main.py'))] + sys.argv[1:]
|
||||||
|
ep = subprocess.Popen(script, stdout=None, stderr=None, stdin=subprocess.DEVNULL)
|
||||||
|
atexit.register(lambda: ep.kill())
|
||||||
|
|
||||||
print("Bazarr starting...")
|
|
||||||
if PY3:
|
def check_status():
|
||||||
ep = subprocess.Popen(script, stdout=None, stderr=None, stdin=subprocess.DEVNULL)
|
if os.path.exists(stopfile):
|
||||||
else:
|
try:
|
||||||
ep = subprocess.Popen(script, stdout=None, stderr=None, stdin=None)
|
os.remove(stopfile)
|
||||||
process_registry.register(ep)
|
except Exception:
|
||||||
try:
|
print('Unable to delete stop file.')
|
||||||
ep.wait()
|
finally:
|
||||||
process_registry.unregister(ep)
|
print('Bazarr exited.')
|
||||||
except KeyboardInterrupt:
|
sys.exit(0)
|
||||||
pass
|
|
||||||
|
if os.path.exists(restartfile):
|
||||||
|
try:
|
||||||
|
os.remove(restartfile)
|
||||||
|
except Exception:
|
||||||
|
print('Unable to delete restart file.')
|
||||||
|
else:
|
||||||
|
print("Bazarr is restarting...")
|
||||||
|
start_bazarr()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
restartfile = os.path.normcase(os.path.join(args.config_dir, 'bazarr.restart'))
|
restartfile = os.path.join(args.config_dir, 'bazarr.restart')
|
||||||
stopfile = os.path.normcase(os.path.join(args.config_dir, 'bazarr.stop'))
|
stopfile = os.path.join(args.config_dir, 'bazarr.stop')
|
||||||
|
|
||||||
|
# Cleanup leftover files
|
||||||
try:
|
try:
|
||||||
os.remove(restartfile)
|
os.remove(restartfile)
|
||||||
except Exception:
|
except FileNotFoundError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
try:
|
try:
|
||||||
os.remove(stopfile)
|
os.remove(stopfile)
|
||||||
except Exception:
|
except FileNotFoundError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
# Initial start of main bazarr process
|
||||||
def daemon(bazarr_runner=lambda: start_bazarr()):
|
print("Bazarr starting...")
|
||||||
if os.path.exists(stopfile):
|
start_bazarr()
|
||||||
try:
|
|
||||||
os.remove(stopfile)
|
|
||||||
except Exception:
|
|
||||||
print('Unable to delete stop file.')
|
|
||||||
else:
|
|
||||||
print('Bazarr exited.')
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
if os.path.exists(restartfile):
|
|
||||||
try:
|
|
||||||
os.remove(restartfile)
|
|
||||||
except Exception:
|
|
||||||
print('Unable to delete restart file.')
|
|
||||||
else:
|
|
||||||
bazarr_runner()
|
|
||||||
|
|
||||||
|
|
||||||
bazarr_runner = lambda: start_bazarr()
|
|
||||||
|
|
||||||
should_stop = lambda: False
|
|
||||||
|
|
||||||
if PY3:
|
|
||||||
daemonStatus = DaemonStatus()
|
|
||||||
|
|
||||||
def force_shutdown():
|
|
||||||
# force the killing of children processes
|
|
||||||
daemonStatus.force_stop()
|
|
||||||
# if a new SIGTERM signal is caught the standard behaviour should be followed
|
|
||||||
signal.signal(signal.SIGTERM, signal.SIG_DFL)
|
|
||||||
# emulate a Ctrl C command on itself (bypasses the signal thing but, then, emulates the "Ctrl+C break")
|
|
||||||
os.kill(os.getpid(), signal.SIGINT)
|
|
||||||
|
|
||||||
def shutdown():
|
|
||||||
# indicates that everything should stop
|
|
||||||
daemonStatus.stop()
|
|
||||||
# if a new sigterm signal is caught it should force the shutdown of children processes
|
|
||||||
signal.signal(signal.SIGTERM, lambda signal_no, frame: force_shutdown())
|
|
||||||
|
|
||||||
signal.signal(signal.SIGTERM, lambda signal_no, frame: shutdown())
|
|
||||||
|
|
||||||
should_stop = lambda: daemonStatus.should_stop()
|
|
||||||
bazarr_runner = lambda: start_bazarr(daemonStatus)
|
|
||||||
|
|
||||||
bazarr_runner()
|
|
||||||
|
|
||||||
# Keep the script running forever until stop is requested through term or keyboard interrupt
|
# Keep the script running forever until stop is requested through term or keyboard interrupt
|
||||||
while not should_stop():
|
while True:
|
||||||
daemon(bazarr_runner)
|
check_status()
|
||||||
time.sleep(1)
|
try:
|
||||||
|
time.sleep(1)
|
||||||
|
except (KeyboardInterrupt, SystemExit):
|
||||||
|
pass
|
||||||
|
|
|
@ -403,15 +403,12 @@
|
||||||
});
|
});
|
||||||
|
|
||||||
$('#shutdown').on('click', function(){
|
$('#shutdown').on('click', function(){
|
||||||
$.ajax({
|
document.open();
|
||||||
url: "{{base_url}}shutdown",
|
document.write('Bazarr has shutdown.');
|
||||||
async: false
|
document.close();
|
||||||
|
$.ajax({
|
||||||
|
url: "{{base_url}}shutdown"
|
||||||
})
|
})
|
||||||
.always(function(){
|
|
||||||
document.open();
|
|
||||||
document.write('Bazarr has shutdown.');
|
|
||||||
document.close();
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|
||||||
$('#logout').on('click', function(){
|
$('#logout').on('click', function(){
|
||||||
|
@ -422,11 +419,9 @@
|
||||||
$('#loader_text').text("Bazarr is restarting, please wait...");
|
$('#loader_text').text("Bazarr is restarting, please wait...");
|
||||||
$.ajax({
|
$.ajax({
|
||||||
url: "{{base_url}}restart",
|
url: "{{base_url}}restart",
|
||||||
async: true,
|
async: true
|
||||||
error: (function () {
|
|
||||||
setTimeout(function () { setInterval(ping, 2000); }, 8000);
|
|
||||||
})
|
|
||||||
});
|
});
|
||||||
|
setTimeout(function () { setInterval(ping, 2000); }, 8000);
|
||||||
});
|
});
|
||||||
|
|
||||||
% from config import settings
|
% from config import settings
|
||||||
|
|
Loading…
Reference in a new issue