3

TL;DR：如果你有一个程序应该运行一段不确定的时间，当用户决定是时候了，你如何编写代码来阻止它？（没有 KeyboardInterrupt 或杀死任务）

--

我最近发布了这个问题：如何使我的代码可停止？（不是杀死/中断） 答案确实解决了我的问题，但从终止/中断的角度来看，这并不是我真正想要的。（虽然，我的问题没有说清楚）所以，我重新措辞。

我为示例目的创建了一个通用脚本。所以我有这个类，它从通用 API 收集数据并将数据写入 csv。代码是通过python main.py在终端窗口上键入来启动的。

import time,csv

import GenericAPI

class GenericDataCollector:
    def __init__(self):
        self.generic_api = GenericAPI()
        self.loop_control = True

    def collect_data(self):
        while self.loop_control: #Can this var be changed from outside of the class? (Maybe one solution)
            data = self.generic_api.fetch_data() #Returns a JSON with some data
            self.write_on_csv(data)
            time.sleep(1)

    def write_on_csv(self, data):
        with open('file.csv','wt') as f:
            writer = csv.writer(f)
            writer.writerow(data)

def run():
    obj = GenericDataCollector()
    obj.collect_data()

if __name__ == "__main__":
    run()

该脚本应该永远运行，或者直到我命令它停止。我知道我可以直接KeyboardInterrupt (Ctrl+C)或突然终止任务。那不是我要找的。我想要一种“软”的方式来告诉脚本该停止了，不仅因为中断是不可预测的，而且它也是一种严厉的停止方式。

Ctrl+C如果该脚本在 docker 容器上运行（例如），除非您碰巧在 docker 内的终端/bash 中，否则您将无法运行。

或者另一种情况：如果该脚本是为客户制作的，我认为告诉客户不好，只需使用Ctrl+C/kill 任务停止它。绝对违反直觉，特别是如果它是一个非技术人员。

我正在寻找编写另一个脚本的方法（假设这是一个可能的解决方案），该脚本将更改为False属性obj.loop_control，一旦完成就完成循环。可以通过在（不同的）终端上键入来运行的东西python stop_script.py。

它不一定需要这样。其他解决方案也是可以接受的，只要它不涉及 KeyboardInterrupt 或 Killing 任务。如果我可以在类中使用一个方法，那就太好了，只要我可以从另一个终端/脚本调用它。

有没有办法做到这一点？

如果你有一个程序应该运行一段不确定的时间，当用户决定时间到时你如何编写代码来阻止它？

4

2 回答 2

4

一般来说，有两种主要方法可以做到这一点（据我所知）。第一个是让您的脚本检查一些可以从外部修改的条件（例如某些文件/套接字的存在或内容）。或者正如@Green Cloak Guy 所说，使用管道是一种进程间通信形式。

第二个是使用内置机制进行进程间通信，称为信号，存在于运行 python 的每个操作系统中。当用户按下 Ctrl+C 时，终端会向前台进程发送特定信号。但是您可以通过编程方式（即从另一个脚本）发送相同（或另一个）信号。

阅读您的另一个问题的答案，我会说解决这个问题所缺少的是一种向您已经运行的进程发送适当信号的方法。本质上，这可以通过使用os.kill()函数来完成。请注意，虽然该函数被称为“kill”，但它可以发送任何信号（不仅仅是SIGKILL）。

为了使其工作，您需要拥有正在运行的进程的进程 ID。了解这一点的常用方法是让您的脚本在启动到存储在公共位置的文件时保存其进程 ID。要获取当前进程 ID，您可以使用该os.getpid()函数。

总而言之，我想说实现您想要的步骤是：

  1. 修改您当前的脚本以将其进程 ID（可通过使用获取os.getpid()）存储到公共位置的文件中，例如/tmp/myscript.pid. 请注意，如果您希望您的脚本是可移植的，您将需要以一种适用于非 unix（如 Windows）等操作系统的方式来解决这个问题。
  2. 选择一个信号（通常是SIGINTorSIGSTOP或SIGTERM）并修改您的脚本以注册一个自定义处理程序，使用signal.signal()该处理程序来解决脚本的正常终止。
  3. 创建另一个（请注意，它可能是带有一些命令行参数的相同脚本）脚本，该脚本从已知文件（aka /tmp/myscript.pid）中读取进程 ID，并使用os.kill().

请注意，使用信号而不是外部方式（文件、管道等）来实现此目的的一个优点是用户仍然可以按 Ctrl+C（如果您选择了SIGINT），这将产生与“停止脚本”相同的行为' 将。

于 2019-07-11T20:05:25.360 回答
3

您真正需要的是任何将信号从一个程序发送到另一个独立程序的方法。一种方法是使用进程间管道。Python为此提供了一个模块（诚然，它似乎需要一个符合POSIX的shell，但大多数主要操作系统都应该提供它）。

你需要做的是事先在你的运行程序（比如说main.py）和你的停止程序（比如说）之间就文件路径达成一致stop.sh。然后你可以让主程序运行，直到有人向那个管道输入一些东西：

import pipes
...
t = pipes.Template()
# create a pipe in the first place
t.open("/tmp/pipefile", "w")
# create a lasting pipe to read from that
pipefile = t.open("/tmp/pipefile", "r")
...

现在，在您的程序中，将您的循环条件更改为“只要此文件没有输入 - 除非有人向其中写入内容，否则.read()将返回一个空字符串：

while not pipefile.read():
    # do stuff

要停止它，您可以放置​​另一个文件或脚本或将写入该文件的内容。使用 shell 脚本最容易做到这一点：

#!/usr/bin/env sh
echo STOP >> /tmp/pipefile

其中，如果您要对其进行容器化，则可以将其放入/usr/bin并命名stop，至少授予它0111权限，然后告诉您的用户“停止程序，只需执行docker exec containername stop”。

（使用>>代替>很重要，因为我们只想附加到管道，而不是覆盖它）。

我的 python 控制台上的概念证明：

>>> import pipes
>>> t = pipes.Template()
>>> t.open("/tmp/file1", "w")
<_io.TextIOWrapper name='/tmp/file1' mode='w' encoding='UTF-8'>
>>> pipefile = t.open("/tmp/file1", "r")
>>> i = 0
>>> while not pipefile.read():
...     i += 1
...

此时我转到另一个终端选项卡并执行

$ echo "Stop" >> /tmp/file1

然后我回到我的 python 选项卡，while循环不再执行，所以我可以检查i我离开时发生了什么。

>>> print(i)
1704312
于 2019-07-11T20:03:35.637 回答