49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from typing import Any
|
|
|
|
from fastapi import APIRouter, BackgroundTasks, Request, Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app import schemas
|
|
from app.chain.webhook import WebhookChain
|
|
from app.core.config import settings
|
|
from app.db import get_db
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
def start_webhook_chain(db: Session, body: Any, form: Any, args: Any):
|
|
"""
|
|
启动链式任务
|
|
"""
|
|
WebhookChain(db).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,
|
|
db: Session = Depends(get_db),) -> 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, db, 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,
|
|
db: Session = Depends(get_db)) -> 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, db, None, None, args)
|
|
return schemas.Response(success=True)
|