LongArticleTaskServer/routes/blueprint.py

from quart import Blueprint, jsonify, request
from applications.ab_test import GetCoverService

from applications.tasks import TaskScheduler

server_blueprint = Blueprint("api", __name__, url_prefix="/api")


def server_routes(pools, log_service):

    @server_blueprint.route("/get_cover", methods=["POST"])
    async def get_cover():
        params = await request.get_json()
        task = GetCoverService(pools, params)
        return jsonify(await task.deal())

    @server_blueprint.route("/run_task", methods=["POST"])
    async def run_task():
        data = await request.get_json()
        print("ss", data)
        task_scheduler = TaskScheduler(data, log_service, pools)
        response = await task_scheduler.deal()
        print(response)
        return jsonify(response)

    @server_blueprint.route("/finish_task", methods=["POST"])
    async def finish_task():
        data = await request.get_json()
        return jsonify({"message": "hello world"})

    return server_blueprint