Разрешение Ctrl-C прерывать C-расширение Python

Я бы переделал расширения C, чтобы они не работали в течение длительного периода.

Итак, разделите их на более элементарные шаги (каждый выполняется в течение короткого периода времени, например, от 10 до 50 миллисекунд), и пусть эти более элементарные шаги будут вызываться кодом Python.

стиль прохождения продолжения может быть уместным для понимания как стиль программирования ...

Однако Ctrl-C, похоже, не имеет никакого эффекта.

Ctrl-C в оболочке отправляет SIGINT группе процессов переднего плана. python при получении сигнала устанавливает флаг в коде C. Если ваше расширение C работает в основном потоке, обработчик сигналов Python не будет запускаться (и, следовательно, вы не увидите KeyboardInterrupt исключение на Ctrl-C), если вы не вызовете _ 6_, который проверяет флаг (это означает: он не должен замедлять вас) и запускает обработчики сигналов Python, если это необходимо или если ваше моделирование позволяет коду Python выполняться (например, если симуляция использует обратные вызовы Python). Вот пример кода модуля расширения для CPython, созданного с использованием pybind11, предложенного @Matt:

PYBIND11_MODULE(example, m)
{
    m.def("long running_func", []()
    {
        for (;;) {
            if (PyErr_CheckSignals() != 0)
                throw py::error_already_set();
            // Long running iteration
        }
    });
}

Если расширение выполняется в фоновом потоке, достаточно выпустить GIL (чтобы позволить коду Python запускаться в основном потоке, который позволяет запускать обработчики сигналов). PyErr_CheckSignals() всегда возвращает 0 в фоновом потоке.

Связано: Cython, Python и KeybordInterrupt включены

Python имеет обработчик сигналов, установленный на SIGINT, который просто устанавливает флаг, который проверяется основным циклом интерпретатора. Чтобы этот обработчик работал правильно, интерпретатор Python должен запускать код Python.

Вам доступны несколько вариантов:

1. Используйте _2 _ / _ 3_, чтобы освободить GIL вокруг вашего кода расширения C. Вы не можете использовать какие-либо функции Python, если не удерживаете GIL, но код Python (и другой код C) может выполняться одновременно с вашим потоком C (настоящая многопоточность). Отдельный поток Python может выполняться вместе с расширением C и перехватывать сигналы Ctrl + C.
2. Настройте свой собственный SIGINT обработчик и вызовите исходный обработчик сигнала (Python). Затем ваш обработчик SIGINT может сделать все, что ему нужно, чтобы отменить код расширения C и вернуть управление интерпретатору Python.

Существует альтернативный способ решения этой проблемы, если вы не хотите, чтобы ваше расширение C (или ctypes DLL) было привязано к Python, например, в случае, когда вы хотите создать библиотеку C с привязками на нескольких языках, вы должны разрешить Расширение C для работы в течение длительного времени, и вы можете изменить расширение C:

Включите заголовок сигнала в расширение C.

#include <signal.h>

Создайте typedef обработчика сигнала в расширении C.

typedef void (*sighandler_t)(int);

Добавьте обработчики сигналов в расширение C, которые будут выполнять действия, необходимые для прерывания любого продолжительного кода (установить флаг остановки и т. Д.), И сохранить существующие обработчики сигналов Python.

sighandler_t old_sig_int_handler = signal(SIGINT, your_sig_handler);
sighandler_t old_sig_term_handler = signal(SIGTERM, your_sig_handler);

Восстанавливайте существующие обработчики сигналов всякий раз, когда возвращается расширение C. Этот шаг обеспечивает повторное применение обработчиков сигналов Python.

signal(SIGINT, old_sig_int_handler);
signal(SIGTERM, old_sig_term_handler);

Если длительный код прерывается (флаг и т. Д.), Верните управление Python с кодом возврата, указывающим номер сигнала.

return SIGINT;

В Python отправьте сигнал, полученный в расширении C.

import os
import signal

status = c_extension.run()

if status in [signal.SIGINT, signal.SIGTERM]:
    os.kill(os.getpid(), status)

Python выполнит ожидаемое действие, например вызовет KeyboardInterrupt для SIGINT.