V1
This commit is contained in:
@@ -0,0 +1,187 @@
|
||||
"""Réglages V4L2 de la webcam locale pour l'interface Alfred.
|
||||
|
||||
API en lecture/écriture sécurisée par une liste blanche de contrôles, plages
|
||||
et états renvoyés par le pilote. Aucun accès shell ni commande arbitraire.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
import shutil
|
||||
import subprocess
|
||||
import tempfile
|
||||
import threading
|
||||
from pathlib import Path
|
||||
|
||||
_ALLOWED = {
|
||||
'brightness', 'contrast', 'saturation', 'hue', 'gamma', 'gain',
|
||||
'sharpness', 'backlight_compensation', 'power_line_frequency',
|
||||
'white_balance_automatic', 'white_balance_temperature',
|
||||
'auto_exposure', 'exposure_time_absolute', 'exposure_dynamic_framerate',
|
||||
}
|
||||
_AUTO = ('white_balance_automatic', 'auto_exposure')
|
||||
_LINE = re.compile(r'^\s*(\w+)\s+0x[\da-f]+\s+\((int|bool|menu)\)\s*:\s*(.*)$', re.I)
|
||||
_KV = re.compile(r'(min|max|step|default|value)=(-?\d+)')
|
||||
_MENU = re.compile(r'^\s*(-?\d+):\s*(.+?)\s*$')
|
||||
_LOCK = threading.RLock()
|
||||
|
||||
|
||||
class CameraSettingsError(RuntimeError):
|
||||
pass
|
||||
|
||||
|
||||
def _device() -> str:
|
||||
return os.getenv('CAMERA_DEVICE') or f"/dev/video{int(os.getenv('CAMERA_INDEX', '0'))}"
|
||||
|
||||
|
||||
def _config_path() -> Path:
|
||||
return Path(os.getenv('CAMERA_SETTINGS_FILE', str(
|
||||
Path(__file__).resolve().parent.parent / 'config' / 'camera_controls.json'
|
||||
))).expanduser()
|
||||
|
||||
|
||||
def _run(*args: str) -> str:
|
||||
binary = shutil.which('v4l2-ctl')
|
||||
if not binary:
|
||||
raise CameraSettingsError('v4l2-ctl est absent : installer le paquet v4l-utils.')
|
||||
try:
|
||||
result = subprocess.run(
|
||||
[binary, '-d', _device(), *args], capture_output=True,
|
||||
text=True, timeout=6, check=False
|
||||
)
|
||||
except (OSError, subprocess.TimeoutExpired) as exc:
|
||||
raise CameraSettingsError(f'Caméra indisponible ({exc}).') from exc
|
||||
if result.returncode != 0:
|
||||
explanation = (result.stderr or result.stdout or 'erreur inconnue').strip()
|
||||
raise CameraSettingsError(f'V4L2 : {explanation[:300]}')
|
||||
return result.stdout
|
||||
|
||||
|
||||
def _parse_controls(output: str) -> dict:
|
||||
controls: dict[str, dict] = {}
|
||||
current = None
|
||||
for line in output.splitlines():
|
||||
match = _LINE.match(line)
|
||||
if match:
|
||||
name, kind, raw = match.groups()
|
||||
current = name if name in _ALLOWED else None
|
||||
if current:
|
||||
props = {k: int(v) for k, v in _KV.findall(raw)}
|
||||
controls[current] = {
|
||||
'kind': kind.lower(), 'inactive': 'inactive' in raw,
|
||||
'menu': [], **props,
|
||||
}
|
||||
continue
|
||||
menu = _MENU.match(line)
|
||||
if menu and current and controls[current]['kind'] == 'menu':
|
||||
value, title = menu.groups()
|
||||
controls[current]['menu'].append({'value': int(value), 'label': title})
|
||||
return controls
|
||||
|
||||
|
||||
def _get_controls() -> dict:
|
||||
return _parse_controls(_run('--list-ctrls-menus'))
|
||||
|
||||
|
||||
def get_settings() -> dict:
|
||||
with _LOCK:
|
||||
return {'device': _device(), 'controls': _get_controls()}
|
||||
|
||||
|
||||
def _validate(control: object, value: object, controls: dict) -> int:
|
||||
if not isinstance(control, str) or control not in _ALLOWED or control not in controls:
|
||||
raise ValueError('Réglage caméra inconnu ou non pris en charge.')
|
||||
data = controls[control]
|
||||
if data['inactive']:
|
||||
raise ValueError('Réglage indisponible tant que le mode automatique est actif.')
|
||||
if type(value) is not int: # écarte bool et chaînes
|
||||
raise ValueError('Valeur entière attendue.')
|
||||
if data['kind'] == 'bool' and value not in (0, 1):
|
||||
raise ValueError('Valeur booléenne attendue (0 ou 1).')
|
||||
if data['kind'] == 'menu':
|
||||
choices = [v['value'] for v in data['menu']]
|
||||
if choices and value not in choices:
|
||||
raise ValueError('Valeur absente des options de la caméra.')
|
||||
minimum, maximum = data.get('min', 0), data.get('max', 1)
|
||||
step = data.get('step', 1)
|
||||
if not minimum <= value <= maximum or (value - minimum) % max(1, step):
|
||||
raise ValueError(f'Valeur hors plage ({minimum} à {maximum}, pas {step}).')
|
||||
return value
|
||||
|
||||
|
||||
def _write_config(controls: dict) -> None:
|
||||
path = _config_path()
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
snapshot = {name: props['value'] for name, props in controls.items()
|
||||
if not props['inactive'] and 'value' in props}
|
||||
fd = None
|
||||
temp = None
|
||||
try:
|
||||
fd, temp = tempfile.mkstemp(prefix='.camera_', suffix='.tmp', dir=path.parent)
|
||||
with os.fdopen(fd, 'w', encoding='utf-8') as stream:
|
||||
fd = None
|
||||
json.dump({'controls': snapshot}, stream, indent=2)
|
||||
stream.write('\n')
|
||||
os.replace(temp, path)
|
||||
temp = None
|
||||
except OSError as exc:
|
||||
raise CameraSettingsError(f'Impossible de mémoriser les réglages caméra : {exc}') from exc
|
||||
finally:
|
||||
if fd is not None:
|
||||
os.close(fd)
|
||||
if temp is not None and os.path.exists(temp):
|
||||
os.unlink(temp)
|
||||
|
||||
|
||||
def update_setting(control: object, value: object) -> dict:
|
||||
with _LOCK:
|
||||
controls = _get_controls()
|
||||
n = _validate(control, value, controls)
|
||||
_run('--set-ctrl', f'{control}={n}')
|
||||
refreshed = _get_controls() # réévalue les contrôles devenus actifs/inactifs
|
||||
_write_config(refreshed)
|
||||
return {'device': _device(), 'controls': refreshed}
|
||||
|
||||
|
||||
def reset_defaults() -> dict:
|
||||
"""Restaure les valeurs exposées par le pilote, sans forcer les champs inactifs."""
|
||||
with _LOCK:
|
||||
original = _get_controls()
|
||||
# D'abord les modes auto, puis les contrôles restés actifs.
|
||||
for name in [*_AUTO, *(c for c in original if c not in _AUTO)]:
|
||||
live = _get_controls()
|
||||
item = live.get(name)
|
||||
if not item or item['inactive'] or 'default' not in item:
|
||||
continue
|
||||
try:
|
||||
_run('--set-ctrl', f"{name}={item['default']}")
|
||||
except CameraSettingsError as exc:
|
||||
raise CameraSettingsError(f'Échec restauration {name} : {exc}') from exc
|
||||
refreshed = _get_controls()
|
||||
_write_config(refreshed)
|
||||
return {'device': _device(), 'controls': refreshed}
|
||||
|
||||
|
||||
def restore_saved_settings() -> None:
|
||||
"""Applique les valeurs persistées; une webcam absente ne bloque pas Flask."""
|
||||
with _LOCK:
|
||||
path = _config_path()
|
||||
if not path.is_file():
|
||||
return
|
||||
try:
|
||||
saved = json.loads(path.read_text(encoding='utf-8')).get('controls', {})
|
||||
if not isinstance(saved, dict):
|
||||
raise ValueError('Format invalide')
|
||||
for name in [*_AUTO, *(n for n in saved if n not in _AUTO)]:
|
||||
if name not in saved:
|
||||
continue
|
||||
live = _get_controls()
|
||||
try:
|
||||
n = _validate(name, saved[name], live)
|
||||
except ValueError:
|
||||
continue # matériel/pilote modifié ou contrôle devenu inactif
|
||||
_run('--set-ctrl', f'{name}={n}')
|
||||
print('[ALFRED CAMERA] Réglages restaurés.', flush=True)
|
||||
except (OSError, ValueError, CameraSettingsError) as exc:
|
||||
print(f'[ALFRED CAMERA] Restauration ignorée : {exc}', flush=True)
|
||||
Reference in New Issue
Block a user