192 lines
6.3 KiB
Python
192 lines
6.3 KiB
Python
"""API de contrôle des LEDs WS2812b."""
|
|
|
|
import asyncio
|
|
import logging
|
|
from fastapi import APIRouter, Request, Query
|
|
from fastapi.responses import JSONResponse
|
|
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/leds/status")
|
|
async def led_status(request: Request):
|
|
led = request.app.state.led_service
|
|
return {"effect": led.current_effect}
|
|
|
|
|
|
@router.post("/leds/effect")
|
|
async def set_effect(
|
|
request: Request,
|
|
effect: str = Query(..., description="idle | countdown | capture | captured | finished | printing | error | disabled | off"),
|
|
):
|
|
"""Force un effet LED (admin uniquement, pas de vérification auth ici — à protéger via Zoraxy)."""
|
|
valid = ("idle", "countdown", "capture", "captured", "finished", "printing", "error", "disabled", "off")
|
|
if effect not in valid:
|
|
return JSONResponse({"error": f"Effet invalide. Valeurs: {valid}"}, status_code=400)
|
|
|
|
led = request.app.state.led_service
|
|
ws = request.app.state.ws_manager
|
|
led.play(effect)
|
|
await ws.broadcast({"type": "led_effect", "effect": effect})
|
|
return {"ok": True, "effect": effect}
|
|
|
|
|
|
@router.post("/leds/color")
|
|
async def set_color(
|
|
request: Request,
|
|
r: int = Query(default=0, ge=0, le=255),
|
|
g: int = Query(default=0, ge=0, le=255),
|
|
b: int = Query(default=0, ge=0, le=255),
|
|
):
|
|
"""Couleur fixe immédiate sur tout l'anneau."""
|
|
led = request.app.state.led_service
|
|
led.set_color(r, g, b)
|
|
return {"ok": True, "color": [r, g, b]}
|
|
|
|
|
|
@router.post("/leds/off")
|
|
async def leds_off(request: Request):
|
|
led = request.app.state.led_service
|
|
led.play("off")
|
|
return {"ok": True}
|
|
|
|
|
|
@router.get("/leds/brightness")
|
|
async def get_brightness(request: Request):
|
|
cfg = request.app.state.config.leds
|
|
return {"brightness": cfg.brightness, "auto_brightness": cfg.auto_brightness}
|
|
|
|
|
|
@router.put("/leds/brightness")
|
|
async def set_brightness(
|
|
request: Request,
|
|
value: int = Query(..., ge=0, le=255),
|
|
save: bool = Query(default=True),
|
|
):
|
|
"""Regle la luminosite globale (0-255) et l'applique immediatement."""
|
|
led = request.app.state.led_service
|
|
led.set_brightness(value)
|
|
if save:
|
|
config_svc = request.app.state.config_service
|
|
config_svc.save_led_brightness(value)
|
|
return {"ok": True, "brightness": value}
|
|
|
|
|
|
@router.put("/leds/auto-brightness")
|
|
async def set_auto_brightness(
|
|
request: Request,
|
|
enabled: bool = Query(...),
|
|
save: bool = Query(default=True),
|
|
):
|
|
"""Active ou desactive la luminosite automatique du flash selon l'ambiance."""
|
|
cfg = request.app.state.config.leds
|
|
cfg.auto_brightness = enabled
|
|
if save:
|
|
config_svc = request.app.state.config_service
|
|
config_svc.save_led_auto_brightness(enabled)
|
|
return {"ok": True, "auto_brightness": enabled}
|
|
|
|
|
|
@router.get("/leds/effects")
|
|
async def get_effects(request: Request):
|
|
"""Retourne la configuration de tous les effets LED."""
|
|
cfg = request.app.state.config.leds
|
|
return {
|
|
name: {
|
|
"color": effect.get("color", [0, 0, 0]),
|
|
"flash_duration": effect.get("flash_duration", 0.1),
|
|
"flashes": effect.get("flashes", 2),
|
|
"speed": effect.get("speed", 0.05),
|
|
"duration": effect.get("duration", 2.0),
|
|
}
|
|
for name, effect in cfg.effects.items()
|
|
}
|
|
|
|
|
|
@router.get("/leds/ambient-lux")
|
|
async def get_ambient_lux(request: Request):
|
|
"""Retourne la luminosité ambiante mesurée (dernière valeur connue)."""
|
|
led = request.app.state.led_service
|
|
lux = led._ambient_lux
|
|
return {
|
|
"ambient_lux": lux,
|
|
"known": lux >= 0,
|
|
"auto_brightness": request.app.state.config.leds.auto_brightness,
|
|
"description": (
|
|
"sombre (flash plein)" if lux < 50 and lux >= 0 else
|
|
"luminosité moyenne (flash adapté)" if lux < 180 and lux >= 0 else
|
|
"clair (flash réduit)" if lux >= 180 else
|
|
"non mesuré"
|
|
),
|
|
}
|
|
|
|
|
|
@router.post("/leds/measure-ambient")
|
|
async def measure_ambient_lux(request: Request):
|
|
"""Déclenche une mesure de luminosité ambiante via snapshot liveview.
|
|
Utile pour diagnostiquer si le flux liveview est accessible depuis JH-Photomaton.
|
|
"""
|
|
pb = request.app.state.photobooth_service
|
|
led = request.app.state.led_service
|
|
|
|
snapshot = await pb.get_liveview_snapshot()
|
|
if not snapshot:
|
|
return JSONResponse({
|
|
"ok": False,
|
|
"error": "Aucun endpoint liveview accessible sur photobooth-app. "
|
|
"Vérifiez que photobooth-app est démarré et que le flux vidéo est actif.",
|
|
}, status_code=503)
|
|
|
|
try:
|
|
from PIL import Image
|
|
import io
|
|
img = Image.open(io.BytesIO(snapshot)).convert("L").resize((80, 60))
|
|
pixels = list(img.getdata())
|
|
avg = sum(pixels) / len(pixels)
|
|
led.set_ambient_lux(avg)
|
|
return {
|
|
"ok": True,
|
|
"ambient_lux": round(avg, 1),
|
|
"description": (
|
|
"sombre → flash plein (255)" if avg < 50 else
|
|
f"moyen → flash {int(255 - (avg - 50) / 130 * 135)}" if avg < 180 else
|
|
"clair → flash réduit (80)"
|
|
),
|
|
}
|
|
except Exception as e:
|
|
return JSONResponse({"ok": False, "error": str(e)}, status_code=500)
|
|
|
|
|
|
@router.put("/leds/effect/{effect_name}")
|
|
async def update_effect(
|
|
request: Request,
|
|
effect_name: str,
|
|
r: int = Query(default=None, ge=0, le=255),
|
|
g: int = Query(default=None, ge=0, le=255),
|
|
b: int = Query(default=None, ge=0, le=255),
|
|
flash_duration: float = Query(default=None, ge=0.05, le=2.0),
|
|
flashes: int = Query(default=None, ge=1, le=5),
|
|
save: bool = Query(default=True),
|
|
):
|
|
"""Met a jour la configuration d'un effet LED (couleur, timings).
|
|
|
|
Exemple : PUT /api/leds/effect/capture?r=255&g=200&b=80&flash_duration=0.3&flashes=2
|
|
"""
|
|
cfg = request.app.state.config.leds
|
|
config_svc = request.app.state.config_service
|
|
|
|
effect_raw = cfg.effects.setdefault(effect_name, {})
|
|
|
|
if r is not None and g is not None and b is not None:
|
|
effect_raw["color"] = [r, g, b]
|
|
if flash_duration is not None:
|
|
effect_raw["flash_duration"] = flash_duration
|
|
if flashes is not None:
|
|
effect_raw["flashes"] = flashes
|
|
|
|
if save:
|
|
config_svc.save_led_effect(effect_name, effect_raw)
|
|
|
|
return {"ok": True, "effect": effect_name, "config": effect_raw}
|