初始化

2026-04-02 23:40:04 +08:00
commit 381ea7085d
107 changed files with 11858 additions and 0 deletions
@@ -0,0 +1,77 @@
+"""数据导入 API"""
+from fastapi import APIRouter, UploadFile, File, Depends, Response
+from sqlalchemy.orm import Session
+from app.core.database import get_db
+from app.services.import_service import ImportService
+import shutil
+import io
+from openpyxl import Workbook
+
+router = APIRouter(prefix="/api/import", tags=["数据导入"])
+
+
+@router.get("/template")
+def download_template():
+    """下载导入数据模板"""
+    wb = Workbook()
+    ws = wb.active
+    ws.title = "ONU设备导入模板"
+
+    # 表头：序号|区域|学校名称|楼宇|场所类型|房间号|MAC地址|备注
+    headers = ["mac_address", "region", "school_name", "building", "place_type", "room_number", "notes"]
+    ws.append(headers)
+
+    # 示例数据
+    example_data = [
+        ["AA:BB:CC:DD:EE:01", "区域1", "学校1", "1号楼", "教室", "101", ""],
+        ["AA:BB:CC:DD:EE:02", "区域1", "学校1", "1号楼", "办公室", "102", ""],
+    ]
+    for row in example_data:
+        ws.append(row)
+
+    # 设置列宽
+    ws.column_dimensions['A'].width = 20  # mac_address
+    ws.column_dimensions['B'].width = 12  # region
+    ws.column_dimensions['C'].width = 18  # school_name
+    ws.column_dimensions['D'].width = 12  # building
+    ws.column_dimensions['E'].width = 12  # place_type
+    ws.column_dimensions['F'].width = 12  # room_number
+    ws.column_dimensions['G'].width = 20  # notes
+
+    # 保存到内存
+    output = io.BytesIO()
+    wb.save(output)
+    output.seek(0)
+
+    return Response(
+        content=output.getvalue(),
+        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+        headers={"Content-Disposition": "attachment; filename=onu_import_template.xlsx"}
+    )
+
+
+@router.post("/upload")
+async def upload_excel(
+    file: UploadFile = File(...),
+    olt_id: int = None,
+    db: Session = Depends(get_db)
+):
+    """上传并导入 Excel 文件（仅导入 MAC 信息，不关联 OLT）"""
+    file_path = f"/tmp/{file.filename}"
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+
+    service = ImportService(db)
+    records = service.parse_excel(file_path)
+    validation = service.validate_data(records)
+
+    success_count = 0
+    if validation['valid']:
+        result = service.import_devices(validation['valid'], olt_id)
+        success_count = result['success']
+
+    return {
+        "success": success_count,
+        "failed": validation['invalid']
+    }
+