Триггерное действие при ручной остановке скрипта Python
Я искал довольно долго, но я не смог найти простого решения. У меня есть скрипт python, который работает бесконечно и сохраняет некоторые файлы при выполнении условия (собрано достаточно данных). Есть ли способ прервать выполнение скрипта и запустить функцию, которая сохранит собранные (но еще не сохраненные) данные? Каждый раз, когда мне нужно что-то сделать (скажем, закрыть компьютер), я должен вручную остановить (завершить) сценарий (в Pycharm), и я теряю часть данных, которые еще не сохранены.
Обновлено: благодаря @Alexander я смог решить эту проблему. Простой код, который мог бы лучше обрисовать решение:
import atexit
import time
@atexit.register
def on_close():
print('success') #save my data
while True:
print('a')
time.sleep(2)
Теперь при нажатии на кнопку «Стоп» выполняется функция «on_close», и я могу сохранить свои данные...
Рад, что вам понравилось, рассмотрите возможность проголосовать.
Ответы 1
Используйте модуль atexit. Это часть python std lib.
import atexit
@atexit.register
def on_close():
... do comething
atexit.register(func, *args, **kwargs)
Зарегистрируйте func как функцию, которая будет выполняться при завершении. Любые необязательные аргументы, которые должны быть переданы функции func, должны быть переданы в качестве аргументов функции register(). Одну и ту же функцию и аргументы можно зарегистрировать более одного раза.
Эта функция возвращает func, что позволяет использовать ее в качестве декоратора.
Большое спасибо, я думаю, это именно то, что я искал... Я сделал простой код, чтобы проверить, как он работает, и он отлично сработал: import atexit import time @atexit.register def on_close(): print ('успех') пока True: print('a') time.sleep(2)
@MatijaCerne Потрясающе рад, что смог помочь
Может быть, у вас есть идея, как заставить atexit работать даже тогда, когда, по-видимому, есть «нечистый» выход: «Процесс завершен с кодом выхода -1»? Я запускаю два цикла while одновременно, и по какой-то причине я не могу остановить выполнение, нажав кнопку «Стоп» только один раз - я должен нажать ее один раз, а затем еще раз, что, я думаю, убивает процесс...
Вам просто нужно убедиться, что вы поймали все возникающие исключения, а затем изящно завершили работу. Но вам лучше провести рефакторинг своего кода, чтобы он корректно завершал работу сам по себе. Плохая практика принудительно завершать процесс, чтобы остановить его. @MatijaCerne
Хорошо, понятно, спасибо.. Попробую поймать. Спасибо, что указали мне в этом направлении!
Еще одно решение ловить сигнал с кнопки стоп.