在扭曲的程序中中断raw_input

时间:2013-12-02 18:56:47

标签: python twisted interrupt keyboard-events keyboardinterrupt

我将参考此explanation和此workaround

所以我正在做的是:

def interrupted(signum, stackframe):
    log.warning('interrupted > Got signal: %s', signum)
    menu.quitMenu = True # to stop my code

signal.signal(signal.SIGINT, interrupted) # Handle KeyboardInterrupt

问题在于,虽然菜单被通知它必须停止,并且很快就会这样做,但它现在无法执行,因为它被卡在raw_input中:

def askUser(self):
    current_date   = datetime.now().isoformat(' ')
    choice = raw_input('%s > ' % current_date)
    return choice

因此,由于twisted正在删除默认的中断处理程序,因此raw_input不会停止。我仍然需要在enter之后按^C才能停止。

如何强制raw_input停止,不用安装默认的中断处理程序,这是扭曲上下文中的问题来源(因为扭曲本身不会被打断)

我猜这个问题与raw_input无关:任何无限时间(或超过预设限制)的函数都应该以某种方式中断。

这是否有可接受的扭曲模式?

修改

这是完整的测试代码:

from datetime import datetime

class Menu:

    def __init__(self):
        self.quitMenu = False

    def showMenu(self):
        print '''

A) Do A
B) Do B

'''

    def askUser(self):
        current_date   = datetime.now().isoformat(' ')
        choice = raw_input('%s > Please select option > ' % current_date)
        print
        return choice

    def stopMe(self):
        self.quitMenu = True

    def alive(self):
        return self.quitMenu == False

    def doMenuOnce(self):
        self.showMenu()
        choice = self.askUser()
        if not self.alive() : # Maybe somebody has tried to stop the menu while in askUser
            return
        if   choice == 'A' : print 'A selected'
        elif choice == 'B' : print 'B selected'
        else               : print 'ERR: choice %s not supported' % (choice)

    def forever(self):
        while self.alive():
            self.doMenuOnce()

from twisted.internet import reactor, threads
import signal

class MenuTwisted:

    def __init__(self, menu):
        self.menu = menu
        signal.signal(signal.SIGINT, self.interrupted) # Handle KeyboardInterrupt

    def interrupted(self, signum, stackframe):
        print 'Interrupted!'
        self.menu.stopMe()

    def doMenuOnce(self):
        threads.deferToThread(self.menu.doMenuOnce).addCallback(self.forever)

    def forever(self, res=None):
        if self.menu.alive() :
            reactor.callLater(0, self.doMenuOnce)
        else : 
            reactor.callFromThread(reactor.stop)

    def run(self):
        self.forever()
        reactor.run()

我可以用两种不同的方式运行。

正常方式:

menu = Menu()
menu.forever()

^C会立即停止该程序:

A) Do A
B) Do B


2013-12-03 11:00:26.288846 > Please select option > ^CTraceback (most recent call last):
  File "twisted_keyboard_interrupt.py", line 72, in <module>
    menu.forever()
  File "twisted_keyboard_interrupt.py", line 43, in forever
    self.doMenuOnce()
  File "twisted_keyboard_interrupt.py", line 34, in doMenuOnce
    choice = self.askUser()
  File "twisted_keyboard_interrupt.py", line 22, in askUser
    choice = raw_input('%s > Please select option > ' % current_date)
KeyboardInterrupt

正如所料。

扭曲的方式:

menu = Menu()
menutw = MenuTwisted(menu)
menutw.run()

^C将产生:

A) Do A
B) Do B


2013-12-03 11:04:18.678219 > Please select option > ^CInterrupted!

askUser实际上并未中断:我仍然需要enterraw_input才能完成。

1 个答案:

答案 0 :(得分:3)

处理此问题的正确方法是handle console input asynchronously,而不是试图使阻塞输入函数可中断。换句话说,raw_input从根本上解决了你正在攻击的问题。

但是,如果你真的只是想了解这里发生了什么,那就是在调用reactor.callFromThread(reactor.stop)之后,你必须以某种方式提示raw_input退出;它不会正常。但是,由于您在一个线程中运行它,它实际上根本不可中断,因为Python中的only the main thread is interruptable。所以我认为你想要的可能实际上是不可能的。我相信也许关闭sys.stdin可能会从raw_input下面拉出来,即使它在一个线程中,但似乎底层库正在做一些比简单地从FD读取更聪明的东西,所以关闭它没有用。