45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from typing import Any
|
|
|
|
from fastapi import APIRouter, BackgroundTasks, Request
|
|
|
|
from app import schemas
|
|
from app.chain.webhook import WebhookChain
|
|
from app.core.config import settings
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
def start_webhook_chain(body: Any, form: Any, args: Any):
|
|
"""
|
|
启动链式任务
|
|
"""
|
|
WebhookChain().message(body=body, form=form, args=args)
|
|
|
|
|
|
@router.post("/", summary="Webhook消息响应", response_model=schemas.Response)
|
|
async def webhook_message(background_tasks: BackgroundTasks,
|
|
token: str, request: Request) -> Any:
|
|
"""
|
|
Webhook响应
|
|
"""
|
|
if token != settings.API_TOKEN:
|
|
return schemas.Response(success=False, message="token认证不通过")
|
|
body = await request.body()
|
|
form = await request.form()
|
|
args = request.query_params
|
|
background_tasks.add_task(start_webhook_chain, body, form, args)
|
|
return schemas.Response(success=True)
|
|
|
|
|
|
@router.get("/", summary="Webhook消息响应", response_model=schemas.Response)
|
|
async def webhook_message(background_tasks: BackgroundTasks,
|
|
token: str, request: Request) -> Any:
|
|
"""
|
|
Webhook响应
|
|
"""
|
|
if token != settings.API_TOKEN:
|
|
return schemas.Response(success=False, message="token认证不通过")
|
|
args = request.query_params
|
|
background_tasks.add_task(start_webhook_chain, None, None, args)
|
|
return schemas.Response(success=True)
|