Merge remote-tracking branch 'origin/main'

src/backend/.env.txt

@@ -0,0 +1,11 @@
+# 硅基流动API配置
+SILICONFLOW_API_KEY=sk-mhzuedasunrgdrxfkcxmxgaypgjnxgodvvmrzzdbqrwtkqej
+SILICONFLOW_API_URL=https://api.siliconflow.ai/v1
+
+# 交换机登录凭证
+SWITCH_USERNAME=admin
+SWITCH_PASSWORD=your_switch_password
+SWITCH_TIMEOUT=10
+
+# 应用设置
+DEBUG=True

src/backend/.gitignore

@@ -1,7 +1,10 @@
+# Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
 *$py.class
 *.so
+
+# Python packaging
 .Python
 build/
 develop-eggs/
@@ -23,6 +26,8 @@ MANIFEST
 # IDE
 .idea/
 .vscode/
+*.swp
+*.swo
 
 # Environment
 .env
@@ -35,3 +40,10 @@ venv.bak/
 
 # Logs
 *.log
+*.log.*
+logs/
+
+# Secrets
+*.secret
+*.key
+*.pem

src/backend/Dockerfile

@@ -2,14 +2,21 @@ FROM python:3.13-slim
 
 WORKDIR /app
 
+# 1. 先复制依赖文件并安装
 COPY ./requirements.txt /app/requirements.txt
-
 RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
 
+# 2. 复制项目代码（排除 .env 和缓存文件）
 COPY . /app
 
-ENV PYTHONPATH=/app
-ENV PORT=8000
-ENV HOST=0.0.0.0
+# 3. 环境变量配置
+ENV PYTHONPATH=/app \
+    PORT=8000 \
+    HOST=0.0.0.0
 
-CMD ["uvicorn", "run:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
+# 4. 安全设置
+RUN find /app -name "*.pyc" -delete && \
+    find /app -name "__pycache__" -exec rm -rf {} +
+
+# 5. 启动命令（修正路径）
+CMD ["uvicorn", "src.backend.app:app", "--host", "0.0.0.0", "--port", "8000"]

src/backend/app/__init__.py

@@ -1,8 +1,10 @@
 from fastapi import FastAPI
-from src.backend.app.api.endpoints import router as api_router
+from src.backend.app.api.endpoints import router
 from src.backend.app.utils.logger import setup_logging
 from src.backend.config import settings
 
+app = FastAPI()
+app.include_router(router,prefix="/api")
 
 def create_app() -> FastAPI:
     # 设置日志
@@ -18,6 +20,6 @@ def create_app() -> FastAPI:
     )
 
     # 添加API路由
-    app.include_router(api_router, prefix=settings.API_PREFIX)
+    app.include_router(router, prefix=settings.API_PREFIX)
 
     return app

src/backend/app/api/__init__.py

@@ -1,4 +1,7 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, FastAPI
 from .endpoints import router
 
-__all__ = ["router"]
+app=FastAPI()
+app.include_router(router)
+
+#__all__ = ["app","router"]

src/backend/app/api/endpoints.py

@@ -2,11 +2,35 @@ from fastapi import APIRouter, Depends, HTTPException
 from typing import Any
 from pydantic import BaseModel
 
-from src.backend.app.services.ai_services import AIService
-from src.backend.app.api.network_config import SwitchConfigurator
-from src.backend.config import settings
+from ...app.services.ai_services import AIService
+from ...app.api.network_config import SwitchConfigurator
+from ...config import settings
+from ..services.network_scanner import NetworkScanner
 
-router = APIRouter()
+router = APIRouter(prefix="/api", tags=["API"])
+scanner = NetworkScanner()
+
+@router.get("/test")
+async def test_endpoint():
+    return {"message": "Hello World"}
+
+@router.get("/scan_network", summary="扫描网络中的交换机")
+async def scan_network(subnet: str = "192.168.1.0/24"):
+    try:
+        devices = scanner.scan_subnet(subnet)
+        return {
+            "success": True,
+            "devices": devices,
+            "count": len(devices)
+        }
+    except Exception as e:
+        raise HTTPException(500, f"扫描失败: {str(e)}")
+
+@router.get("/list_devices", summary="列出已发现的交换机")
+async def list_devices():
+    return {
+        "devices": scanner.load_cached_devices()
+    }
 
 class CommandRequest(BaseModel):
     command: str

src/backend/app/services/network_scanner.py

@@ -0,0 +1,49 @@
+import nmap
+import json
+from pathlib import Path
+from typing import List, Dict
+from ..utils.logger import logger
+
+
+class NetworkScanner:
+    def __init__(self, cache_path: str = "switch_devices.json"):
+        self.cache_path = Path(cache_path)
+        self.nm = nmap.PortScanner()
+
+    def scan_subnet(self, subnet: str = "192.168.1.0/24") -> List[Dict]:
+        """扫描指定子网的交换机设备"""
+        logger.info(f"Scanning subnet: {subnet}")
+
+        # 扫描开放22(SSH)或23(Telnet)端口的设备
+        self.nm.scan(
+            hosts=subnet,
+            arguments="-p 22,23 --open -T4"
+        )
+
+        devices = []
+        for host in self.nm.all_hosts():
+            if self.nm[host].state() == "up":
+                device = {
+                    "ip": host,
+                    "ports": list(self.nm[host]["tcp"].keys()),
+                    "mac": self.nm[host].get("addresses", {}).get("mac", "unknown")
+                }
+                devices.append(device)
+                logger.debug(f"Found device: {device}")
+
+        self._save_to_cache(devices)
+        return devices
+
+    def _save_to_cache(self, devices: List[Dict]):
+        """保存扫描结果到本地文件"""
+        with open(self.cache_path, "w") as f:
+            json.dump(devices, f, indent=2)
+        logger.info(f"Saved {len(devices)} devices to cache")
+
+    def load_cached_devices(self) -> List[Dict]:
+        """从缓存加载设备列表"""
+        if not self.cache_path.exists():
+            return []
+
+        with open(self.cache_path) as f:
+            return json.load(f)

src/backend/config.py

@@ -1,14 +1,16 @@
+from pydantic_settings import BaseSettings
+from dotenv import load_dotenv
+
+load_dotenv()
+
 import os
-from pydantic import BaseSettings
-
-
 class Settings(BaseSettings):
     APP_NAME: str = "AI Network Configurator"
     DEBUG: bool = True
     API_PREFIX: str = "/api"
 
     # 硅基流动API配置
-    SILICONFLOW_API_KEY: str = os.getenv("SILICONFLOW_API_KEY", "")
+    SILICONFLOW_API_KEY: str = os.getenv("SILICON_API_KEY", "")
     SILICONFLOW_API_URL: str = os.getenv("SILICONFLOW_API_URL", "https://api.siliconflow.ai/v1")
 
     # 交换机配置

src/backend/requirements.txt

@@ -5,3 +5,4 @@ requests==2.28.2
 paramiko==3.1.0
 pydantic==1.10.7
 loguru==0.7.0
+python-nmap==0.7.1