如何中断Ubuntu中断python程序

在Ubuntu中中断Python程序的方法有多种，主要包括使用键盘中断、使用信号模块、使用timeout装饰器、使用多线程或多进程管理、中断系统资源监控。其中，使用信号模块是一种常见且灵活的方法，可以捕捉和处理各种信号，并且适用于需要更精细控制的场景。

一、键盘中断

1.1 使用Ctrl+C中断

最简单的方式是在终端运行Python程序时，使用Ctrl+C组合键发送中断信号（SIGINT）来中断程序执行。这种方法适用于大多数情况。

try: while True: # 模拟长时间运行的任务 pass except KeyboardInterrupt: print("程序已中断")

1.2 捕获KeyboardInterrupt异常

可以在代码中捕获KeyboardInterrupt异常，以便在中断时执行一些清理操作。

try: while True: # 模拟长时间运行的任务 pass except KeyboardInterrupt: print("程序被手动中断") finally: print("清理资源")

二、使用信号模块

2.1 捕获信号

Python的signal模块可以捕获系统信号，并在接收到特定信号时执行自定义的处理函数。例如，可以捕获SIGINT信号并处理它。

import signal
import time
def handler(signum, frame):
    print("接收到信号:", signum)
    raise SystemExit("程序已中断")
signal.signal(signal.SIGINT, handler)
while True:
    time.sleep(1)
    print("运行中...")

2.2 自定义处理函数

可以为不同的信号定义不同的处理函数，并根据需要对信号进行处理。

import signal
import time
def sigterm_handler(signum, frame):
    print("接收到SIGTERM信号")
    raise SystemExit("程序已中断")
def sigint_handler(signum, frame):
    print("接收到SIGINT信号")
    raise SystemExit("程序已中断")
signal.signal(signal.SIGTERM, sigterm_handler)
signal.signal(signal.SIGINT, sigint_handler)
while True:
    time.sleep(1)
    print("运行中...")

三、使用`timeout`装饰器

3.1 设置超时

可以使用timeout装饰器为函数设置执行时间的限制，超时后自动中断程序执行。

import signal
class TimeoutError(Exception):
    pass
def timeout(seconds, error_message="超时"):
    def decorator(func):
        def _handle_timeout(signum, frame):
            raise TimeoutError(error_message)
        def wrapper(*args, kwargs):
            signal.signal(signal.SIGALRM, _handle_timeout)
            signal.alarm(seconds)
            try:
                result = func(*args, kwargs)
            finally:
                signal.alarm(0)
            return result
        return wrapper
    return decorator
@timeout(5)
def long_running_task():
    while True:
        print("运行中...")
        time.sleep(1)
try:
    long_running_task()
except TimeoutError as e:
    print(e)

四、使用多线程或多进程管理

4.1 多线程中断

在多线程环境中，可以使用threading模块管理线程，并通过设置标志位来中断线程。

import threading
import time
def worker(stop_event):
    while not stop_event.is_set():
        print("运行中...")
        time.sleep(1)
stop_event = threading.Event()
thread = threading.Thread(target=worker, args=(stop_event,))
thread.start()
time.sleep(5)
stop_event.set()
thread.join()
print("程序已中断")

4.2 多进程中断

在多进程环境中，可以使用multiprocessing模块管理进程，并通过发送终止信号来中断进程。

import multiprocessing
import time
def worker():
    while True:
        print("运行中...")
        time.sleep(1)
process = multiprocessing.Process(target=worker)
process.start()
time.sleep(5)
process.terminate()
process.join()
print("程序已中断")

五、中断系统资源监控

5.1 使用psutil模块

psutil模块可以用于监控系统资源，并通过条件判断中断Python程序。

import psutil
import time
def monitor_resources():
    while True:
        cpu_usage = psutil.cpu_percent(interval=1)
        print(f"CPU使用率: {cpu_usage}%")
        if cpu_usage > 80:
            raise SystemExit("CPU使用率过高，程序已中断")
try:
    monitor_resources()
except SystemExit as e:
    print(e)

5.2 自定义资源监控

可以自定义资源监控逻辑，并根据需要中断Python程序。

import psutil
import time
def custom_monitor():
    while True:
        memory_info = psutil.virtual_memory()
        print(f"内存使用率: {memory_info.percent}%")
        if memory_info.percent > 80:
            raise SystemExit("内存使用率过高，程序已中断")
        time.sleep(1)
try:
    custom_monitor()
except SystemExit as e:
    print(e)

六、总结

以上介绍了几种在Ubuntu中中断Python程序的方法，包括使用键盘中断、使用信号模块、使用timeout装饰器、使用多线程或多进程管理、中断系统资源监控。不同的方法适用于不同的应用场景，可以根据具体需求选择合适的中断方式。其中，使用信号模块是一种常见且灵活的方法，适用于需要更精细控制的场景。希望这些方法能帮助到你更好地管理和中断Python程序的执行。