80 lines
No EOL
3.8 KiB
Python
80 lines
No EOL
3.8 KiB
Python
from fastapi import APIRouter, HTTPException
|
||
from typing import List, Dict
|
||
from models.subnet import SubnetScanResult
|
||
from models.wifi import WifiScanResult
|
||
from models.request import NoteRequest
|
||
from services.subnet_service import SubnetService
|
||
from services.wifi_service import WifiService
|
||
|
||
router = APIRouter()
|
||
|
||
@router.get("/scan", response_model=SubnetScanResult)
|
||
def scan_subnet(subnet: str = "192.168.1.0/24", ports: List[int] = None):
|
||
"""Сканирует локальную подсеть и возвращает данные об устройствах"""
|
||
try:
|
||
subnet_service = SubnetService()
|
||
result = subnet_service.scan_subnet(subnet, ports)
|
||
return result
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"Ошибка сканирования: {str(e)}")
|
||
|
||
@router.get("/wifi", response_model=WifiScanResult)
|
||
def scan_wifi(timeout: int = 30):
|
||
"""Сканирует Wi-Fi точки доступа и возвращает данные о них"""
|
||
try:
|
||
wifi_service = WifiService()
|
||
result = wifi_service.scan_wifi(timeout=timeout)
|
||
return result
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"Ошибка сканирования Wi-Fi: {str(e)}")
|
||
|
||
@router.post("/devices/{mac}/note")
|
||
def add_note(mac: str, request: NoteRequest):
|
||
"""Добавляет заметку для устройства по MAC-адресу"""
|
||
try:
|
||
subnet_service = SubnetService()
|
||
success, error_message = subnet_service.add_note(mac, request.note)
|
||
if not success:
|
||
if "не найдено" in error_message:
|
||
raise HTTPException(status_code=404, detail=error_message)
|
||
elif "уже существует" in error_message:
|
||
raise HTTPException(status_code=400, detail=error_message)
|
||
return {"message": f"Заметка '{request.note}' добавлена для устройства {mac}"}
|
||
except HTTPException as e:
|
||
raise e
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"Ошибка добавления заметки: {str(e)}")
|
||
|
||
@router.put("/devices/{mac}/note")
|
||
def update_note(mac: str, request: NoteRequest):
|
||
"""Обновляет заметку для устройства по MAC-адресу"""
|
||
try:
|
||
subnet_service = SubnetService()
|
||
success, error_message = subnet_service.update_note(mac, request.note)
|
||
if not success:
|
||
if "не найдено" in error_message:
|
||
raise HTTPException(status_code=404, detail=error_message)
|
||
elif "не существует" in error_message:
|
||
raise HTTPException(status_code=400, detail=error_message)
|
||
return {"message": f"Заметка '{request.note}' обновлена для устройства {mac}"}
|
||
except HTTPException as e:
|
||
raise e
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"Ошибка обновления заметки: {str(e)}")
|
||
|
||
@router.delete("/devices/{mac}/note")
|
||
def delete_note(mac: str):
|
||
"""Удаляет заметку для устройства по MAC-адресу"""
|
||
try:
|
||
subnet_service = SubnetService()
|
||
success, error_message = subnet_service.delete_note(mac)
|
||
if not success:
|
||
if "не найдено" in error_message:
|
||
raise HTTPException(status_code=404, detail=error_message)
|
||
elif "не существует" in error_message:
|
||
raise HTTPException(status_code=400, detail=error_message)
|
||
return {"message": f"Заметка удалена для устройства {mac}"}
|
||
except HTTPException as e:
|
||
raise e
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}") |