大家好,今天来学习用Flask API创建Python后台任务。

在Python中,有若干解决方案可以实现后台任务,比如Celery或Redis Queue,都是实现任务队列系统的好方法。但使用这二者都比较麻烦。设想这样一个场景,我们用Flask建立一个API,实现从一个终端调用后台任务,并用另一个终端停止后台任务。

使用Flask建立了一个简单的API,有两个主要的方法,一个用于启动后台任务,另一个用于停止。

为了管理任务的生命周期,我们使用Event Objects,这是一种简单的线程间通信机制。

如下所示是所需导入的库、线程事件声明和后台任务方法:

from time import sleep
from flask import Flask
from flask_cors import CORS
import threading
thread_event = threading.Event()
def backgroundTask():
    while thread_event.is_set():
        print('Background task running!')
        sleep(5)

这里的关键是is_set()方法,它将返回内部线程事件标志的值:truefalse

首先,使用set()方法把标志设置为true,它将启动一个线程,并连续运行backgroundTask方法。

@app.route("/start", methods=["POST"])
def startBackgroundTask():
    try:
        thread_event.set()
        thread = threading.Thread(target=backgroundTask)
        thread.start()
        return "Background task started!"
    except Exception as error:
        return str(error)

如果要停止任务,调用clear()方法将标志设置为false,以停止正在运行的线程。

@app.route("/stop", methods=["POST"])
def stopBackgroundTask():
    try:
        thread_event.clear()
        return "Background task stopped!"
    except Exception as error:
        return str(error)

以上就是后台任务的代码解决方案,这比用任务队列实现后台任务要简单得多。可以使用上述代码来建立具有一个或多个后台任务的API,大家可以自行实践。

原文地址:https://blog.csdn.net/csdn1561168266/article/details/130159771

相关文章: