使用Python轻松创建Flask API后台任务

大家好，今天来学习用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()方法，它将返回内部线程事件标志的值：true或false。

首先，使用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，大家可以自行实践。