Перезапуск файла Python без утраты аргументов и окружения

Перезапуск с добавленным флагом и сохранением исходных аргументов

Пример показывает передачу аргументов в следующий запуск. Первый процесс добавляет флаг --resume, а второй процесс завершает цепочку после обработки этого флага.

import os
import sys

if '--resume' in sys.argv:
    print('Вторая стадия, аргументы:', sys.argv[1:])
    sys.exit(0)

script_path = os.path.abspath(__file__)
arguments = [sys.executable, script_path] + sys.argv[1:] + ['--resume']
print('Первая стадия, аргументы:', sys.argv[1:])
sys.stdout.flush()
os.execv(sys.executable, arguments)

Результат для команды python app.py mode=fast:

Первая стадия, аргументы: mode=fast
Вторая стадия, аргументы: mode=fast --resume

Использование абсолютного пути к файлу делает команду независимой от смены рабочего каталога. Список sys.argv[1:] сохраняет пользовательские аргументы на протяжении всего перезапуска.

Ограниченная серия перезапусков с задержкой между попытками

Здесь переменная окружения хранит номер текущего запуска. Процесс выполняет код, делает паузу и заменяет себя новой копией. После третьего запуска цепочка останавливается.

import os
import sys
import time

max_runs = 3
run_number = int(os.environ.get('RESTART_RUN', '0')) + 1
os.environ['RESTART_RUN'] = str(run_number)
print('Номер запуска:', run_number)

if run_number >= max_runs:
    print('Все запуски выполнены')
else:
    time.sleep(1)
    script_path = os.path.abspath(__file__)
    os.execv(sys.executable, [sys.executable, script_path] + sys.argv[1:])

Вывод:

Номер запуска: 1
Номер запуска: 2
Номер запуска: 3
Все запуски выполнены

Задержка перед заменой процесса предотвращает быстрое повторение кода при нестабильном состоянии системы.

Автоматический перезапуск при изменении конфигурационного файла

Внешний наблюдатель следит за временем изменения settings.ini. Если файл изменился, текущий child останавливается и запускается новая копия worker.py. Такой механизм востребован для файловых настроек без ручного вмешательства оператора.

import os
import subprocess
import sys
import time

worker_path = 'worker.py'
config_path = 'settings.ini'

def get_mtime():
    return os.path.getmtime(config_path)

previous_mtime = get_mtime()
child = subprocess.Popen([sys.executable, worker_path])

try:
    while True:
        time.sleep(1)
        current_mtime = get_mtime()
        if current_mtime == previous_mtime:
            continue
        print('settings.ini изменён')
        child.terminate()
        child.wait()
        child = subprocess.Popen([sys.executable, worker_path])
        previous_mtime = current_mtime
except KeyboardInterrupt:
    child.terminate()
    child.wait()

После изменения settings.ini в консоли появляется сообщение:

settings.ini изменён

Старая версия worker.py останавливается, а новая запускается с прежними параметрами окружения.

Запуск worker.py в отдельной директории с собственным окружением

Этот сценарий удобен, когда перезапускаемый файл находится в другом проекте или требует особого каталога. Параметр cwd задаёт рабочий каталог, env расширяет переменные окружения, а stdout и stderr направляются в общий лог.

import os
import subprocess
import sys

base_dir = '/home/user/project'
env = os.environ.copy()
env['APP_CONFIG'] = 'production'
log = open('restart.log', 'w')
worker = subprocess.Popen(
    [sys.executable, 'worker.py'],
    cwd=base_dir,
    env=env,
    stdout=log,
    stderr=log
)
print('worker запущен с PID', worker.pid)

Вывод:

worker запущен с PID 12345

Все сообщения worker.py попадают в restart.log, а процесс не зависит от текущего каталога родительской программы.

Повторное выполнение модуля через importlib.reload

В редких случаях вместо перезапуска файла нужно заново выполнить код уже импортированного модуля. Функция importlib.reload запускает модуль ещё раз в текущем процессе, не закрывая открытые соединения и не теряя переменные окружения.

import importlib
import my_module

print('Значение до перезагрузки:', my_module.VALUE)
my_module.VALUE = 100
my_module = importlib.reload(my_module)
print('Значение после перезагрузки:', my_module.VALUE)

Если файл my_module.py содержит строку VALUE = 1, вывод будет:

Значение до перезагрузки: 1
Значение после перезагрузки: 1

Такой подход подходит для модулей с настройками, но не заменяет полный перезапуск процесса Python.