Python-Watchdog/python_watchdog.py at main · phantom101/Python-Watchdog · GitHub

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
# Python-Watchdog
# by Patrick Schmoll 2024
# For more information visit: https://github.com/phantom101/Python-Watchdog

import configparser
import subprocess
import time
import os
import threading

# Set the check interval how often watchdog should look for changes in the .ini files
check_interval = 5

# List for managing active watchdog threads
watchdog_threads = []

# Read the config file
def watchdog_read_config(config_file):
    config = configparser.ConfigParser()
    config.read(config_file)
    return config['settings']

# Function to start a script
def start_script(script_name):
    print(f"Starting {script_name}...")
    return subprocess.Popen(['python', script_name])

# Function to stop a script
def stop_script(process):
    if process:
        print("Stopping the script...")
        process.terminate()
        process.wait()
        print("Script stopped.")

# Function to update values in the ini files
def watchdog_update(config_file, option, value):
    config = configparser.ConfigParser()
    config.read(config_file)

    section = 'settings'

    if section not in config:
        print(f"The section '{section}' could not be found.")
        return
    if option not in config[section]:
        print(f"The option '{option}' could not be found.")
        return

    config[section][option] = str(value)

    with open(config_file, 'w') as configfile:
        config.write(configfile)

    print(f"The value of '{option}' in '{config_file}' has changed to '{value}'.")

# Function to manage the scripts (if started stop or restart etc.)
def manage_scripts(script_process_dict, config):
    for key in config.keys():
        if key.startswith('script'):
            script_name = config[key]
            start_key = f'start{key[-1]}'
            stop_key = f'stop{key[-1]}'
            restart_key = f'restart{key[-1]}'

            if config.getint(start_key) and (script_name not in script_process_dict or script_process_dict[script_name] is None):
                script_process_dict[script_name] = start_script(script_name)

            elif config.getint(stop_key) and script_name in script_process_dict:
                stop_script(script_process_dict[script_name])
                del script_process_dict[script_name]

            elif script_name in script_process_dict and config.getint(restart_key) == 1:
                if script_process_dict[script_name] is not None and script_process_dict[script_name].poll() is not None:
                    print(f"Script {script_name} has stopped, restarting...")
                    script_process_dict[script_name] = start_script(script_name)

# Run a watchdog
def run_watchdog(config_file='config.ini'):
    last_modified_time = os.path.getmtime(config_file)
    config = watchdog_read_config(config_file)
    script_process_dict = {}

    try:
        while True:
            manage_scripts(script_process_dict, config)

            current_modified_time = os.path.getmtime(config_file)
            if current_modified_time != last_modified_time:
                print("Configuration file has changed! Reloading...")
                last_modified_time = current_modified_time
                config = watchdog_read_config(config_file)

            time.sleep(check_interval)  # Ein Pause zwischen den Überprüfungen

    except KeyboardInterrupt:
        print("Watchdog stopped.")

# Function to call Run a watchdog inside a thread so we dont block the main script and also can call multiple watchdogs
def start_watchdog(config_file='config.ini'):
    watchdog_thread = threading.Thread(target=run_watchdog, args=(config_file,), daemon=True)
    watchdog_thread.start()
    watchdog_threads.append(watchdog_thread)