Refactor: Clean main.py

app/api/routes/devices.py

@@ -0,0 +1,64 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy import select
+from app.core.state import state_manager, discovery_service
+from app.core.database import async_session
+from app.models.device import GroupModel, GroupCreateSchema
+from app.api.deps import verify_token
+from app.drivers.wiz import WizDriver
+
+# Создаем роутер с защитой
+router = APIRouter(dependencies=[Depends(verify_token)])
+wiz = WizDriver()
+
+
+@router.get("")
+async def get_all_devices():
+    return state_manager.devices
+
+
+@router.get("/groups")
+async def get_groups():
+    return state_manager.groups
+
+
+@router.get("/scenes")
+async def get_scenes():
+    return wiz.SCENES
+
+
+@router.post("/groups")
+async def create_group(data: GroupCreateSchema):
+    async with async_session() as session:
+        existing = await session.get(GroupModel, data.id)
+        if existing:
+            raise HTTPException(status_code=400, detail="Группа с таким ID уже есть")
+
+        new_group = GroupModel(id=data.id, name=data.name, device_ids=data.macs)
+        session.add(new_group)
+        await session.commit()
+        state_manager.groups[data.id] = new_group
+    return {"status": "created", "group": data.name}
+
+
+@router.delete("/groups/{group_id}")
+async def delete_group(group_id: str):
+    async with async_session() as session:
+        result = await session.execute(
+            select(GroupModel).where(GroupModel.id == group_id)
+        )
+        group = result.scalar_one_or_none()
+        if not group:
+            raise HTTPException(status_code=404, detail="Группа не найдена")
+
+        await session.delete(group)
+        await session.commit()
+        state_manager.groups.pop(group_id, None)
+    return {"status": "deleted", "id": group_id}
+
+
+@router.post("/rescan")
+async def rescan_network():
+    found_devices = await discovery_service.scan_network()
+    for dev_data in found_devices:
+        state_manager.update_device(dev_data)
+    return {"status": "ok", "found": len(state_manager.devices)}