bazarr/bazarr.py

153 lines
5.0 KiB
Python
Raw Normal View History

# coding=utf-8
import os
2019-09-05 15:30:14 +00:00
import platform
import signal
2020-01-31 22:04:01 +00:00
import subprocess
import sys
import time
from bazarr.app.get_args import args
from bazarr.literals import EXIT_PYTHON_UPGRADE_NEEDED, EXIT_NORMAL, FILE_RESTART, FILE_STOP, ENV_RESTARTFILE, ENV_STOPFILE, EXIT_INTERRUPT
def exit_program(status_code):
print(f'Bazarr exited with status code {status_code}.')
raise SystemExit(status_code)
2019-09-05 15:30:14 +00:00
2019-09-05 15:30:14 +00:00
def check_python_version():
python_version = platform.python_version_tuple()
minimum_py3_tuple = (3, 8, 0)
2020-01-31 22:04:01 +00:00
minimum_py3_str = ".".join(str(i) for i in minimum_py3_tuple)
2020-04-15 02:20:23 +00:00
if int(python_version[0]) < minimum_py3_tuple[0]:
2020-01-31 22:04:01 +00:00
print("Python " + minimum_py3_str + " or greater required. "
"Current version is " + platform.python_version() + ". Please upgrade Python.")
exit_program(EXIT_PYTHON_UPGRADE_NEEDED)
2023-06-23 04:03:23 +00:00
elif int(python_version[0]) == 3 and int(python_version[1]) > 11:
print("Python version greater than 3.11.x is unsupported. Current version is " + platform.python_version() +
". Keep in mind that even if it works, you're on your own.")
elif (int(python_version[0]) == minimum_py3_tuple[0] and int(python_version[1]) < minimum_py3_tuple[1]) or \
2020-04-15 02:20:23 +00:00
(int(python_version[0]) != minimum_py3_tuple[0]):
print("Python " + minimum_py3_str + " or greater required. "
2020-01-31 22:04:01 +00:00
"Current version is " + platform.python_version() + ". Please upgrade Python.")
exit_program(EXIT_PYTHON_UPGRADE_NEEDED)
2019-09-05 15:30:14 +00:00
def get_python_path():
if sys.platform == "darwin":
# Do not run Python from within macOS framework bundle.
python_bundle_path = os.path.join(sys.base_exec_prefix, "Resources", "Python.app", "Contents", "MacOS", "Python")
if os.path.exists(python_bundle_path):
import tempfile
python_path = os.path.join(tempfile.mkdtemp(), "python")
os.symlink(python_bundle_path, python_path)
return python_path
return sys.executable
2019-09-05 15:30:14 +00:00
check_python_version()
2018-10-11 01:23:30 +00:00
dir_name = os.path.dirname(__file__)
def start_bazarr():
script = [get_python_path(), "-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, env=os.environ)
print(f"Bazarr starting child process with PID {ep.pid}...")
return ep
2024-04-20 14:50:32 +00:00
def terminate_child():
print(f"Terminating child process with PID {child_process.pid}")
child_process.terminate()
def get_stop_status_code(input_file):
2020-05-18 12:56:16 +00:00
try:
2024-04-20 14:50:32 +00:00
with open(input_file, 'r') as file:
# read status code from file, if it exists
line = file.readline()
try:
status_code = int(line)
except (ValueError, TypeError):
status_code = EXIT_NORMAL
file.close()
2024-04-20 14:50:32 +00:00
except Exception:
status_code = EXIT_NORMAL
return status_code
2020-04-15 02:20:23 +00:00
def check_status():
global child_process
if os.path.exists(stop_file):
status_code = get_stop_status_code(stop_file)
2020-04-15 02:20:23 +00:00
try:
2024-04-20 14:50:32 +00:00
print("Deleting stop file...")
os.remove(stop_file)
2024-04-20 14:50:32 +00:00
except Exception:
2020-04-15 02:20:23 +00:00
print('Unable to delete stop file.')
finally:
terminate_child()
exit_program(status_code)
2020-04-15 02:20:23 +00:00
if os.path.exists(restart_file):
2020-04-15 02:20:23 +00:00
try:
2024-04-20 14:50:32 +00:00
print("Deleting restart file...")
os.remove(restart_file)
2020-04-15 02:20:23 +00:00
except Exception:
print('Unable to delete restart file.')
finally:
terminate_child()
2024-04-20 14:50:32 +00:00
print("Bazarr is restarting...")
child_process = start_bazarr()
def interrupt_handler(signum, frame):
# catch and ignore keyboard interrupt Ctrl-C
# the child process Server object will catch SIGINT and perform an orderly shutdown
global interrupted
if not interrupted:
# ignore user hammering Ctrl-C; we heard you the first time!
interrupted = True
print('Handling keyboard interrupt...')
else:
2024-04-20 14:50:32 +00:00
print("Stop doing that! I heard you the first time!")
if __name__ == '__main__':
interrupted = False
signal.signal(signal.SIGINT, interrupt_handler)
restart_file = os.path.join(args.config_dir, FILE_RESTART)
stop_file = os.path.join(args.config_dir, FILE_STOP)
os.environ[ENV_STOPFILE] = stop_file
os.environ[ENV_RESTARTFILE] = restart_file
2020-01-31 22:04:01 +00:00
2020-04-15 02:20:23 +00:00
# Cleanup leftover files
try:
os.remove(restart_file)
2020-04-15 02:20:23 +00:00
except FileNotFoundError:
2018-10-11 01:23:30 +00:00
pass
2020-01-31 22:04:01 +00:00
2018-10-11 01:23:30 +00:00
try:
os.remove(stop_file)
2020-04-15 02:20:23 +00:00
except FileNotFoundError:
pass
2020-01-31 22:04:01 +00:00
2020-04-15 02:20:23 +00:00
# Initial start of main bazarr process
child_process = start_bazarr()
2020-01-31 22:04:01 +00:00
# Keep the script running forever until stop is requested through term, special files or keyboard interrupt
2020-04-15 02:20:23 +00:00
while True:
check_status()
try:
time.sleep(5)
except (KeyboardInterrupt, SystemExit, ChildProcessError):
# this code should never be reached, if signal handling is working properly
2024-04-20 14:50:32 +00:00
print('Bazarr exited main script file via keyboard interrupt.')
exit_program(EXIT_INTERRUPT)