summaryrefslogtreecommitdiffhomepage
path: root/bazarr.py
blob: f5fc76bb8d9ddc0fea4f502645aaaf8ef58c51b8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
# coding=utf-8

import os
import platform
import signal
import subprocess
import sys
import time
import atexit

from bazarr.app.get_args import args


def check_python_version():
    python_version = platform.python_version_tuple()
    minimum_py3_tuple = (3, 8, 0)
    minimum_py3_str = ".".join(str(i) for i in minimum_py3_tuple)

    if int(python_version[0]) < minimum_py3_tuple[0]:
        print("Python " + minimum_py3_str + " or greater required. "
              "Current version is " + platform.python_version() + ". Please upgrade Python.")
        sys.exit(1)
    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 \
            (int(python_version[0]) != minimum_py3_tuple[0]):
        print("Python " + minimum_py3_str + " or greater required. "
              "Current version is " + platform.python_version() + ". Please upgrade Python.")
        sys.exit(1)


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


check_python_version()

dir_name = os.path.dirname(__file__)


def end_child_process(ep):
    try:
        if os.name != 'nt':
            try:
                ep.send_signal(signal.SIGINT)
            except ProcessLookupError:
                pass
        else:
            import win32api
            import win32con
            try:
                win32api.GenerateConsoleCtrlEvent(win32con.CTRL_C_EVENT, ep.pid)
            except KeyboardInterrupt:
                pass
    except:
        ep.terminate()


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)
    atexit.register(end_child_process, ep=ep)
    signal.signal(signal.SIGTERM, lambda signal_no, frame: end_child_process(ep))


def check_status():
    if os.path.exists(stopfile):
        try:
            os.remove(stopfile)
        except Exception:
            print('Unable to delete stop file.')
        finally:
            print('Bazarr exited.')
            sys.exit(0)

    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__':
    restartfile = os.path.join(args.config_dir, 'bazarr.restart')
    stopfile = os.path.join(args.config_dir, 'bazarr.stop')

    # Cleanup leftover files
    try:
        os.remove(restartfile)
    except FileNotFoundError:
        pass

    try:
        os.remove(stopfile)
    except FileNotFoundError:
        pass

    # Initial start of main bazarr process
    print("Bazarr starting...")
    start_bazarr()

    # Keep the script running forever until stop is requested through term or keyboard interrupt
    while True:
        check_status()
        try:
            if sys.platform.startswith('win'):
                time.sleep(5)
            else:
                os.wait()
                time.sleep(1)
        except (KeyboardInterrupt, SystemExit, ChildProcessError):
            print('Bazarr exited.')
            sys.exit(0)