feat: Stats. Closes #6

This commit is contained in:
Artem Kokos
2026-03-28 23:06:40 +07:00
parent c793b73fa2
commit 732313a61c
5 changed files with 303 additions and 12 deletions

View File

@@ -1,10 +1,13 @@
import asyncio
import json
import logging
from typing import Optional
from fastapi import APIRouter, Depends, HTTPException
from app.core.state import state_manager
from app.core.database import async_session
from app.drivers.wiz import WizDriver
from app.api.deps import verify_token
from app.api.deps import verify_token, AuthContext
from app.models.event_log import EventLog
logger = logging.getLogger(__name__)
@@ -12,9 +15,44 @@ router = APIRouter(dependencies=[Depends(verify_token)])
wiz = WizDriver()
async def _log_event(
auth: AuthContext, action: str, target_type: str, target_id: str, params: dict
):
"""Записать событие в лог."""
try:
async with async_session() as session:
event = EventLog(
key_name=auth.key_name,
action=action,
target_type=target_type,
target_id=target_id,
params=json.dumps(params, ensure_ascii=False) if params else None,
)
session.add(event)
await session.commit()
except Exception as e:
logger.error(f"Ошибка записи в лог: {e}")
def _classify_action(params: dict) -> str:
"""Определить тип действия по параметрам."""
if "state" in params and len(params) == 1:
return "toggle_on" if params["state"] else "toggle_off"
if "sceneId" in params or "scene" in params:
return "scene"
if "r" in params or "g" in params or "b" in params:
return "color"
if "temp" in params:
return "temperature"
if "dimming" in params:
return "brightness"
return "control"
@router.post("/device/{device_id}")
async def control_device(
device_id: str,
auth: AuthContext = Depends(verify_token),
state: Optional[bool] = None,
brightness: Optional[int] = None,
scene: Optional[str] = None,
@@ -44,12 +82,17 @@ async def control_device(
raise HTTPException(status_code=400, detail="Никаких команд не передано")
result = await wiz.set_pilot(device.ip, params)
# Логируем
await _log_event(auth, _classify_action(params), "device", device_id, params)
return {"device_id": device_id, "applied": params, "result": result}
@router.post("/group/{group_id}")
async def control_group(
group_id: str,
auth: AuthContext = Depends(verify_token),
state: Optional[bool] = None,
brightness: Optional[int] = None,
scene: Optional[str] = None,
@@ -76,8 +119,11 @@ async def control_group(
params["r"], params["g"], params["b"] = r or 0, g or 0, b or 0
tasks = [wiz.set_pilot(ip, params) for ip in ips]
await asyncio.gather(*tasks, return_exceptions=True)
# Логируем
await _log_event(auth, _classify_action(params), "group", group_id, params)
return {"status": "ok", "applied": params, "sent_to": ips}
@@ -88,18 +134,11 @@ async def blink_device(device_id: str):
raise HTTPException(status_code=404, detail="Лампа оффлайн")
try:
# 1. Получаем текущее состояние
current = await wiz.get_pilot(device.ip)
# Если не удалось получить статус, считаем что она выключена (False)
original_state = current.get("result", {}).get("state", False)
# 2. Инвертируем состояние
await wiz.set_pilot(device.ip, {"state": not original_state})
await asyncio.sleep(0.5)
# 3. Возвращаем как было
await wiz.set_pilot(device.ip, {"state": original_state})
return {"status": "blink_done", "original": original_state}
except Exception as e:
logger.error(f"Blink error: {e}")
@@ -127,11 +166,9 @@ async def get_group_status(group_id: str):
if not ips:
raise HTTPException(status_code=404, detail="Группа не найдена или оффлайн")
# Опрашиваем все лампы группы параллельно
tasks = [wiz.get_pilot(ip) for ip in ips]
results = await asyncio.gather(*tasks, return_exceptions=True)
# Формируем красивый ответ
status_report = []
for ip, res in zip(ips, results):
if isinstance(res, Exception):