如何在不同Python文件中终止线程？

本文旨在解决在不同Python文件中启动和终止线程时遇到的`AttributeError`问题。通过分析`__name__`变量在模块导入时的行为，提供了两种解决方案：一是移除`if __name__ == "__main__":`的判断，直接在模块级别创建进程；二是将进程创建代码封装在函数中，并在主程序中调用该函数，从而实现跨文件控制线程的启动和终止。

在Python多进程编程中，经常需要在不同的文件中启动和终止进程。如果直接在导入的模块中使用multiprocessing.Process创建进程，并尝试在主程序中终止它，可能会遇到AttributeError: module 'xxx' has no attribute 'yyy'的错误。这是因为Python在导入模块时，__name__变量的值会发生变化，导致某些代码块不会执行。

以下将介绍两种解决此问题的方法。

方法一：移除if __name__ == "__main__":判断

最简单的解决方法是移除script1.py中的if __name__ == "__main__":判断。这样，无论script1.py是被直接执行还是被导入，都会执行创建进程的代码。

立即学习“Python免费学习笔记（深入）”；

script1.py

import multiprocessing
import time

def worker():
    while True:
        print(f"Working........")
        time.sleep(5)

demo = multiprocessing.Process(target=worker, args=())
demo.start()
#demo.terminate()

script2.py

import time
import script1

if __name__ == "__main__":
    print("script2......")
    time.sleep(2)
    script1.demo.terminate()

在这个例子中，script1.py在被script2.py导入时，会直接创建并启动demo进程。然后在script2.py中，可以通过script1.demo.terminate()来终止这个进程。

注意事项： 这种方法简单直接，但可能会导致script1.py在被其他模块导入时，意外地启动进程。因此，需要谨慎使用。

稿定在线PS

PS软件网页版

99

查看详情

方法二：将进程创建代码封装在函数中

更安全和灵活的方法是将进程创建的代码封装在一个函数中，然后在需要启动进程的地方调用这个函数。

script1.py

import multiprocessing
import time

def worker():
    while True:
        print(f"Working........")
        time.sleep(5)

def run_process():
    demo = multiprocessing.Process(target=worker, args=())
    demo.start()
    return demo

if __name__ == "__main__":
    demo = run_process()
    #demo.terminate()

script2.py

import time
import script1

if __name__ == "__main__":
    demo = script1.run_process()

    time.sleep(2)
    demo.terminate()

在这个例子中，script1.py定义了一个run_process()函数，用于创建并启动进程。在script2.py中，我们导入script1.py，然后调用script1.run_process()来启动进程，并将返回的demo对象保存起来，以便后续终止进程。

优点：

• 更安全：只有在显式调用run_process()函数时才会创建进程。
• 更灵活：可以在不同的地方多次调用run_process()函数，创建多个进程。

总结

通过以上两种方法，可以解决在不同Python文件中启动和终止线程时遇到的AttributeError问题。选择哪种方法取决于具体的应用场景和需求。如果只是简单地需要在主程序中终止一个在另一个文件中启动的进程，并且可以接受在导入模块时启动进程的风险，那么第一种方法可能更简单。如果需要更安全和灵活地控制进程的创建和终止，那么第二种方法更合适。

在实际应用中，还需要注意进程间的通信和同步问题，以确保程序的正确性和稳定性。例如，可以使用multiprocessing.Queue或multiprocessing.Pipe来实现进程间的数据传递，使用multiprocessing.Lock或multiprocessing.Semaphore来实现进程间的同步。

以上就是如何在不同Python文件中终止线程？的详细内容，更多请关注php中文网其它相关文章！