mirror of
https://github.com/Jerryplusy/AI-powered-switches.git
synced 2025-07-04 21:29:18 +00:00
114514
This commit is contained in:
parent
60359b54ee
commit
6f74a80036
@ -1,11 +1,8 @@
|
|||||||
from fastapi import FastAPI
|
from fastapi import FastAPI, responses
|
||||||
from src.backend.app.api.endpoints import router
|
from src.backend.app.api.endpoints import router
|
||||||
from src.backend.app.utils.logger import setup_logging
|
from src.backend.app.utils.logger import setup_logging
|
||||||
from src.backend.config import settings
|
from src.backend.config import settings
|
||||||
|
|
||||||
app = FastAPI()
|
|
||||||
app.include_router(router,prefix="/api")
|
|
||||||
|
|
||||||
def create_app() -> FastAPI:
|
def create_app() -> FastAPI:
|
||||||
# 设置日志
|
# 设置日志
|
||||||
setup_logging()
|
setup_logging()
|
||||||
@ -19,7 +16,20 @@ def create_app() -> FastAPI:
|
|||||||
openapi_url=f"{settings.API_PREFIX}/openapi.json"
|
openapi_url=f"{settings.API_PREFIX}/openapi.json"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# 添加根路径重定向
|
||||||
|
@app.get("/", include_in_schema=False)
|
||||||
|
async def root():
|
||||||
|
return responses.RedirectResponse(url=f"{settings.API_PREFIX}/docs")
|
||||||
|
|
||||||
|
# 添加favicon处理
|
||||||
|
@app.get("/favicon.ico", include_in_schema=False)
|
||||||
|
async def favicon():
|
||||||
|
return responses.Response(status_code=204)
|
||||||
|
|
||||||
# 添加API路由
|
# 添加API路由
|
||||||
app.include_router(router, prefix=settings.API_PREFIX)
|
app.include_router(router, prefix=settings.API_PREFIX)
|
||||||
|
|
||||||
return app
|
|
||||||
|
|
||||||
|
return app
|
||||||
|
app = create_app()
|
@ -1,4 +1,4 @@
|
|||||||
from fastapi import (APIRouter, HTTPException)
|
from fastapi import (APIRouter, HTTPException, Response)
|
||||||
from typing import List
|
from typing import List
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
@ -10,6 +10,27 @@ from ..services.network_scanner import NetworkScanner
|
|||||||
router = APIRouter(prefix="/api", tags=["API"])
|
router = APIRouter(prefix="/api", tags=["API"])
|
||||||
scanner = NetworkScanner()
|
scanner = NetworkScanner()
|
||||||
|
|
||||||
|
# 添加根路径路由
|
||||||
|
@router.get("/", include_in_schema=False)
|
||||||
|
async def root():
|
||||||
|
return {
|
||||||
|
"message": "欢迎使用AI交换机配置系统",
|
||||||
|
"docs": f"{settings.API_PREFIX}/docs",
|
||||||
|
"redoc": f"{settings.API_PREFIX}/redoc",
|
||||||
|
"endpoints": [
|
||||||
|
"/parse_command",
|
||||||
|
"/apply_config",
|
||||||
|
"/scan_network",
|
||||||
|
"/list_devices",
|
||||||
|
"/batch_apply_config"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
# 添加favicon处理
|
||||||
|
@router.get("/favicon.ico", include_in_schema=False)
|
||||||
|
async def favicon():
|
||||||
|
return Response(status_code=204)
|
||||||
|
|
||||||
class BatchConfigRequest(BaseModel):
|
class BatchConfigRequest(BaseModel):
|
||||||
config: dict
|
config: dict
|
||||||
switch_ips: List[str] # 支持多个IP
|
switch_ips: List[str] # 支持多个IP
|
||||||
@ -20,11 +41,13 @@ async def batch_apply_config(request: BatchConfigRequest):
|
|||||||
for ip in request.switch_ips:
|
for ip in request.switch_ips:
|
||||||
try:
|
try:
|
||||||
configurator = SwitchConfigurator()
|
configurator = SwitchConfigurator()
|
||||||
results[ip] = await configurator.safe_apply(ip, request.config)
|
results[ip] = await configurator.apply_config(ip, request.config)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
results[ip] = str(e)
|
results[ip] = str(e)
|
||||||
return {"results": results}
|
return {"results": results}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/test")
|
@router.get("/test")
|
||||||
async def test_endpoint():
|
async def test_endpoint():
|
||||||
return {"message": "Hello World"}
|
return {"message": "Hello World"}
|
||||||
|
@ -1,11 +1,8 @@
|
|||||||
import uvicorn
|
import uvicorn
|
||||||
from src.backend.app import create_app
|
|
||||||
|
|
||||||
app = create_app()
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
uvicorn.run(
|
uvicorn.run(
|
||||||
"src.backend.app:app",
|
"src.backend.app:app", # 使用字符串引用方式
|
||||||
host="0.0.0.0",
|
host="0.0.0.0",
|
||||||
port=8000,
|
port=8000,
|
||||||
log_level="info",
|
log_level="info",
|
||||||
|
Loading…
x
Reference in New Issue
Block a user