freeleaps-service-hub/apps/notification/webapi/routes/__init__.py

20 lines
860 B
Python

from fastapi import APIRouter
from .send_notification import router as sn_router
from .online_platform_notification import router as ws_router
from .template_massege import router as template_router
from .email_sender import router as email_sender_router
api_router = APIRouter(prefix="/notification")
api_router.include_router(sn_router, tags=["notification"])
api_router.include_router(template_router, prefix="/templates", tags=["templates"])
api_router.include_router(email_sender_router, tags=["email_sender"])
# WebSocket路由
websocket_router = APIRouter()
websocket_router.include_router(ws_router, prefix="/downstream", tags=["downstream"])
# 将WebSocket路由也注册到主路由中
api_router.include_router(websocket_router)
websocket_router = APIRouter()
websocket_router.include_router(ws_router, prefix="/downstream", tags=["downstream"])