fulREST ready

main.py

@@ -4,10 +4,13 @@ from contextlib import asynccontextmanager
 from typing import Optional, List
 
 from fastapi import FastAPI, HTTPException
+from sqlalchemy import select
 
 from app.core.discovery import DiscoveryService
 from app.core.state import state_manager
 from app.drivers.wiz import WizDriver
+from app.core.database import init_db, async_session
+from app.models.device import GroupModel, DeviceModel, GroupCreateSchema
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s | %(levelname)s | %(message)s"
@@ -20,11 +23,21 @@ wiz = WizDriver()
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    """
-    Жизненный цикл приложения: выполняется при старте и остановке.
-    """
+    """Жизненный цикл приложения."""
+    # 1. Инициализация БД (создание таблиц)
+    await init_db()
+    logger.info("🗄️ База данных инициализирована")
 
-    # Задача для периодического сканирования сети
+    # 2. Загрузка групп из БД в память
+    async with async_session() as session:
+        result = await session.execute(select(GroupModel))
+        groups = result.scalars().all()
+        for g in groups:
+            # Превращаем модель БД в формат стейта (если нужно)
+            state_manager.groups[g.id] = g
+            logger.info(f"📂 Загружена группа: {g.name} (MACs: {g.device_ids})")
+
+    # 3. Задача для периодического сканирования сети
     async def periodic_discovery():
         logger.info("🚀 Запущена фоновая служба Discovery")
 
@@ -36,43 +49,72 @@ async def lifespan(app: FastAPI):
                     state_manager.update_device(dev_data)
 
                 logger.info(
-                    f"📡 Сеть просканирована. Устройств в памяти: {len(state_manager.devices)}"
+                    f"📡 Сеть просканирована. Устройств онлайн: {len(state_manager.devices)}"
                 )
             except Exception as e:
                 logger.error(f"❌ Ошибка в цикле Discovery: {e}")
-
-            # Интервал сканирования (например, раз в 10 минут)
             await asyncio.sleep(600)
 
     discovery_task = asyncio.create_task(periodic_discovery())
 
     yield
 
-    # Логика при завершении работы сервера
     discovery_task.cancel()
     logger.info("🛑 Фоновая служба Discovery остановлена")
 
 
-app = FastAPI(
-    title="Ignis Core API",
-    description="Ядро системы умного дома для управления лампами WiZ/Gauss",
-    version="1.0.0",
-    lifespan=lifespan,
-)
+app = FastAPI(title="Ignis Core API", lifespan=lifespan)
 
-
-@app.get("/health")
-async def health():
-    """Проверка работоспособности сервиса."""
-    return {"status": "online", "devices_discovered": len(state_manager.devices)}
+# --- Эндпоинты Устройств ---
 
 
 @app.get("/devices")
 async def get_all_devices():
-    """Возвращает список всех обнаруженных устройств из стейта."""
+    """Список всех обнаруженных ламп."""
     return state_manager.devices
 
 
+@app.post("/control/device/{device_id}")
+async def control_device(device_id: str, state: bool, brightness: Optional[int] = None):
+    """Прямое управление лампой по MAC."""
+    device = state_manager.devices.get(device_id)
+    if not device:
+        raise HTTPException(status_code=404, detail="Лампа не в сети")
+
+    params = {"state": state}
+    if brightness:
+        params["dimming"] = brightness
+
+    result = await wiz.set_pilot(device.ip, params)
+    return {"device_id": device_id, "result": result}
+
+
+# --- Эндпоинты Групп ---
+
+
+@app.get("/groups")
+async def get_groups():
+    """Список всех созданных люстр/групп."""
+    return state_manager.groups
+
+
+@app.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}
+
+
 @app.post("/control/group/{group_id}")
 async def control_group(
     group_id: str,
@@ -80,19 +122,14 @@ async def control_group(
     brightness: Optional[int] = None,
     scene: Optional[str] = None,
 ):
-    """
-    Управление группой ламп (люстрой).
-    Параметры передаются в JSON или как query-params.
-    """
-    # 1. Получаем IP-адреса всех ламп в группе из стейт-менеджера
-    ips = state_manager.get_group_ips(group_id)
+    """Управление всей люстрой сразу."""
 
+    ips = state_manager.get_group_ips(group_id)
     if not ips:
         raise HTTPException(
-            status_code=404, detail=f"Группа '{group_id}' не найдена или пуста"
+            status_code=404, detail="Группа не найдена или лампы оффлайн"
         )
 
-    # 2. Формируем параметры команды для WiZ
     params = {}
     if state is not None:
         params["state"] = state
@@ -102,35 +139,13 @@ async def control_group(
         params["sceneId"] = wiz.SCENES[scene]
 
     if not params:
-        raise HTTPException(
-            status_code=400,
-            detail="Не указаны параметры управления (state, brightness или scene)",
-        )
+        raise HTTPException(status_code=400, detail="Никаких команд не передано")
 
-    # 3. Отправляем команды всем лампам группы параллельно
+    # Параллельная отправка всем лампам группы
     tasks = [wiz.set_pilot(ip, params) for ip in ips]
-    results = await asyncio.gather(*tasks, return_exceptions=True)
+    await asyncio.gather(*tasks)
 
-    return {
-        "group_id": group_id,
-        "affected_ips": ips,
-        "status": "commands_sent",
-        "params": params,
-    }
-
-
-@app.post("/control/device/{device_id}")
-async def control_device(device_id: str, state: bool):
-    """Прямое управление устройством по его ID (MAC-адресу)."""
-    device = state_manager.devices.get(device_id)
-
-    if not device:
-        raise HTTPException(
-            status_code=404, detail="Устройство не найдено в текущем стейте"
-        )
-
-    result = await wiz.set_pilot(device.ip, {"state": state})
-    return {"device_id": device_id, "ip": device.ip, "result": result}
+    return {"status": "ok", "group": group_id, "sent_to": ips}
 
 
 if __name__ == "__main__":