从 python 控制台外部发送变量或从外部停止脚本

Question

我编写了一个 Python 脚本，该脚本执行优化并运行数天以获得解决方案（由于目标函数成本高昂）。在所有的工作中，只要在某个时候停止计算就足够了，因为解决方案对我来说已经足够好了（但对于优化算法来说却不是）。

问题是，我总是可以中止按 Ctrl+C。但是没有机会很好地输出当前的最佳参数，绘制数据，保存它等等。在下一次计算目标函数之后以受控方式停止脚本会很棒。所以我的想法是质疑一些变量（如果 user_stop=True）并以编程方式停止优化。但是如何设置这样的变量呢？ python控制台在执行过程中被阻塞。

我考虑过设置文本文件的内容并在每次迭代中读取它，但对于脚本的其他用户来说，这不仅糟糕而且难以解释。从理论上讲，我也可以要求用户输入，但脚本不会自动运行（它应该直到有人决定停止）。

对我的问题有什么想法吗？

基本上就是这样 - 在某个点停止循环但执行打印：

a = 0
while True:
    a = a + 1
print(a)

Answer 1

如果您不经常轮询“变量”（比如最多每 20 秒一次），那么测试文件的开销可以忽略不计。类似的东西

import os
QUITFILE = "/home/myscript/quit_now.txt"
# and for convenience, delete any old QUITFILE that may exist at init time

...  # days later 
if os.path.isfile( QUITFILE) 
    # tidy up, delete QUITFILE, and exit

然后只需echo please > home/myscript/quit_now.txt 告诉您的程序退出。

Answer 2

也许您可以使用 do-while 循环。将目标保存在变量中 在循环之外并开始循环计算 while

Answer 3

对于 Windows，我会使用 msvcrt.getch()

例如，此脚本将循环直到按下某个键，然后，如果是q，则提示用户退出：（注意if 语句使用'short circuiting' 仅评估@987654328 @ - 阻塞 - 当我们知道某个键被按下时。）

import msvcrt, time

while True: #This is your optimization loop
    if msvcrt.kbhit() and msvcrt.getch() == 'q':
        retval = raw_input('Quit? (Y/N) >')
        if retval.lower() == 'y':
            print 'Quitting'
            break #Or set a flag...
    else:
        time.sleep(1)
        print('Processing...')

如果您将此if 块放置在优化循环中将经常运行的某个点，它将允许您在方便的点进行操作，或者至少设置一个您可以在最后检查的标志每次优化运行。

如果你不能把它放在经常被检查的地方，那么你可以看看处理 Ctrl-C 引发的KeyboardInterrupt

如果您在 Linux 上运行，或者需要跨平台功能，请查看this 获取按键的答案。