Отправка ^ C в объекты подпроцесса Python в Windows

Ответ 1

Существует решение, используя оболочку (как описано в ссылке Vinay), которая запускается в новом окне консоли с помощью команды запуска Windows.

Код обертки:

#wrapper.py
import subprocess, time, signal, sys, os

def signal_handler(signal, frame):
  time.sleep(1)
  print 'Ctrl+C received in wrapper.py'

signal.signal(signal.SIGINT, signal_handler)
print "wrapper.py started"
subprocess.Popen("python demo.py")
time.sleep(3) #Replace with your IPC code here, which waits on a fire CTRL-C request
os.kill(signal.CTRL_C_EVENT, 0)

Код программы catch CTRL-C:

#demo.py

import signal, sys, time

def signal_handler(signal, frame):
  print 'Ctrl+C received in demo.py'
  time.sleep(1)
  sys.exit(0)

signal.signal(signal.SIGINT, signal_handler)
print 'demo.py started'
#signal.pause() # does not work under Windows
while(True):
  time.sleep(1)

Запустите оболочку, например, например:

PythonPrompt> import subprocess
PythonPrompt> subprocess.Popen("start python wrapper.py", shell=True)

Вам нужно добавить код IPC, который позволяет вам управлять оболочкой, запускающей команду os.kill(signal.CTRL_C_EVENT, 0). Я использовал сокеты для этой цели в своем приложении.

Объяснение:

Preinformation

• send_signal(CTRL_C_EVENT) не работает, потому что CTRL_C_EVENT предназначен только для os.kill. [REF1]
• os.kill(CTRL_C_EVENT) отправляет сигнал всем процессам, запущенным в текущем окне cmd [REF2]
• Popen(..., creationflags=CREATE_NEW_PROCESS_GROUP) не работает, потому что CTRL_C_EVENT игнорируется для групп процессов. [REF2] Это ошибка в документации python [REF3]

Реализованное решение

• Пусть ваша программа запускается в другом CMD-окне с запуском командной оболочки Windows.
• Добавьте оболочку запроса CTRL-C между вашим управляющим приложением и приложением, которое должно получить сигнал CTRL-C. Обертка будет работать в том же окне cmd, что и приложение, которое должно получить сигнал CTRL-C.
• Обертка выключится сама и программа, которая должна получить сигнал CTRL-C, отправив все процессы в CMD-окне CTRL_C_EVENT.
• Программа управления должна иметь возможность запросить обертку для отправки сигнала CTRL-C. Это может быть реализовано через средства IPC, например. розетки.

Полезные сообщения:

Мне пришлось удалить http перед ссылками, потому что я новый пользователь и мне не разрешено размещать более двух ссылок.

• http://social.msdn.microsoft.com/Forums/en-US/windowsgeneraldevelopmentissues/thread/dc9586ab-1ee8-41aa-a775-cf4828ac1239/#6589714f-12a7-447e-b214-27372f31ca11
• Можно ли отправить ctrl-C (SIGINT) в приложение в Windows?
• Отправка SIGINT подпроцесса python
• http://bugs.python.org/issue9524
• http://ss64.com/nt/start.html
• http://objectmix.com/python/387639-sending-cntrl-c.html#post1443948

Обновление: CTRL-C Wrapper на основе IPC

Здесь вы можете найти самозаписывающийся модуль python, предоставляющий обертку CTRL-C, включая IPC на основе сокетов. Синтаксис довольно похож на модуль подпроцесса.

Использование:

>>> import winctrlc
>>> p1 = winctrlc.Popen("python demo.py")
>>> p2 = winctrlc.Popen("python demo.py")
>>> p3 = winctrlc.Popen("python demo.py")
>>> p2.send_ctrl_c()
>>> p1.send_ctrl_c()
>>> p3.send_ctrl_c()

код

import socket
import subprocess
import time
import random
import signal, os, sys


class Popen:
  _port = random.randint(10000, 50000)
  _connection = ''

  def _start_ctrl_c_wrapper(self, cmd):
    cmd_str = "start \"\" python winctrlc.py "+"\""+cmd+"\""+" "+str(self._port)
    subprocess.Popen(cmd_str, shell=True)

  def _create_connection(self):
    self._connection = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    self._connection.connect(('localhost', self._port))

  def send_ctrl_c(self):
    self._connection.send(Wrapper.TERMINATION_REQ)
    self._connection.close()

  def __init__(self, cmd):
    self._start_ctrl_c_wrapper(cmd)
    self._create_connection()


class Wrapper:
  TERMINATION_REQ = "Terminate with CTRL-C"

  def _create_connection(self, port):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(('localhost', port))
    s.listen(1)
    conn, addr = s.accept()
    return conn

  def _wait_on_ctrl_c_request(self, conn):
    while True:
      data = conn.recv(1024)
      if data == self.TERMINATION_REQ:
        ctrl_c_received = True
        break
      else:
        ctrl_c_received = False
    return ctrl_c_received

  def _cleanup_and_fire_ctrl_c(self, conn):
    conn.close()
    os.kill(signal.CTRL_C_EVENT, 0)

  def _signal_handler(self, signal, frame):
    time.sleep(1)
    sys.exit(0)

  def __init__(self, cmd, port):
    signal.signal(signal.SIGINT, self._signal_handler)
    subprocess.Popen(cmd)
    conn = self._create_connection(port)
    ctrl_c_req_received = self._wait_on_ctrl_c_request(conn)
    if ctrl_c_req_received:
      self._cleanup_and_fire_ctrl_c(conn)
    else:
      sys.exit(0)


if __name__ == "__main__":
  command_string = sys.argv[1]
  port_no = int(sys.argv[2])
  Wrapper(command_string, port_no)

Ответ 2

Попробуйте вызвать GenerateConsoleCtrlEvent с помощью ctypes. Когда вы создаете новую группу процессов, идентификатор группы процессов должен быть таким же, как и pid. Итак, что-то вроде

import ctypes

ctypes.windll.kernel32.GenerateConsoleCtrlEvent(0, proc.pid) # 0 => Ctrl-C

должен работать.

Обновление: Вы правы, я пропустил эту часть деталей. Здесь сообщение, которое предлагает возможное решение, хотя оно немного клочковое. Более подробная информация находится в этом ответе.

Ответ 3

Я пробовал это, но по какой-то причине ctrl + break работает, а ctrl + c - нет. Поэтому использование os.kill(signal.CTRL_C_EVENT, 0) завершается с ошибкой, но работает os.kill(signal.CTRL_C_EVENT, 1). Мне сказали, что это связано с тем, что владелец процесса создания является единственным, кто может передать ctrl c? Это имеет смысл?

Чтобы уточнить, при запуске fio вручную в командном окне он работает как ожидалось. Использование прерываний CTRL + BREAK без сохранения журнала, как ожидается, и CTRL + C завершает запись в файл также, как ожидалось. Проблема заключается в сигнале для CTRL_C_EVENT.

Он почти кажется ошибкой в ​​Python, но скорее может быть ошибкой в ​​Windows. Кроме того, еще одна вещь, у меня была версия cygwin и отправка ctrl + c в python, там тоже работало, но опять же мы на самом деле не запускаем родные окна.

Пример:

import subprocess, time, signal, sys, os
command = '"C:\\Program Files\\fio\\fio.exe" --rw=randrw --bs=1M --numjobs=8 --iodepth=64 --direct=1 ' \
    '--sync=0 --ioengine=windowsaio --name=test --loops=10000 ' \
    '--size=99901800 --rwmixwrite=100 --do_verify=0 --filename=I\\:\\test ' \
    '--thread --output=C:\\output.txt'
def signal_handler(signal, frame):
  time.sleep(1)
  print 'Ctrl+C received in wrapper.py'

signal.signal(signal.SIGINT, signal_handler)
print 'command Starting'
subprocess.Popen(command)
print 'command started'
time.sleep(15) 
print 'Timeout Completed'
os.kill(signal.CTRL_C_EVENT, 0)