再删掉一些东西

This commit is contained in:
Jerry 2025-08-12 00:31:15 +08:00
parent d77a0dacad
commit 59c8604cda

View File

@ -2,17 +2,12 @@ import socket
from fastapi import (APIRouter, HTTPException, Response) from fastapi import (APIRouter, HTTPException, Response)
from typing import List from typing import List
from pydantic import BaseModel from pydantic import BaseModel
from fastapi.responses import HTMLResponse
import psutil import psutil
import ipaddress import ipaddress
from ...app.services.ai_services import AIService from ...app.services.ai_services import AIService
from ...app.api.network_config import SwitchConfigurator from ...app.api.network_config import SwitchConfigurator
from ...config import settings from ...config import settings
from ..services.network_scanner import NetworkScanner from ..services.network_scanner import NetworkScanner
from ..services.network_visualizer import NetworkVisualizer
from ..services.config_validator import ConfigValidator
from ..services.report_generator import ReportGenerator
@ -208,38 +203,3 @@ async def get_network_adapters():
except Exception as e: except Exception as e:
return {"error": f"获取网络适配器信息失败: {str(e)}"} return {"error": f"获取网络适配器信息失败: {str(e)}"}
visualizer = NetworkVisualizer()
report_gen = ReportGenerator()
@router.get("/topology/visualize", response_class=HTMLResponse)
async def visualize_topology():
"""获取网络拓扑可视化图"""
try:
devices = await list_devices()
visualizer.update_topology(devices["devices"])
image_data = visualizer.generate_topology_image()
return f"""
<html>
<head><title>Network Topology</title></head>
<body>
<h1>Network Topology</h1>
<img src="data:image/png;base64,{image_data}" alt="Network Topology">
</body>
</html>
"""
except Exception as e:
raise HTTPException(500, detail=str(e))
@router.post("/config/validate")
async def validate_config(config: dict):
"""验证配置有效性"""
is_valid, errors = ConfigValidator.validate_full_config(config)
return {
"valid": is_valid,
"errors": errors,
"has_security_risks": len(ConfigValidator.check_security_risks(config.get("commands", []))) > 0
}