light_controller.py - master - drresist/xiabilight

light_controller.py

136 строк4 KB

drresist

create: PKG-INFO, SOURCES.txt, entry_points.txt, requires.txt, top_level.txt, dependency_links.txt, test_app.cpython-313-pytest-9.0.3.pyc, test_app.py, uv.lock, README.md, pyproject.toml, main.py, web_app.py, templates.py, camera_service.py, light_controller.py, color_analyzer.py, app_config.py, ambilight_ui_config.json

19 май 2026, 15:10

Верифицирован

19 май 2026, 15:105a71c14

from typing import Optional, Tuple

import numpy as np

try:
    from yeelight import Bulb
except ImportError:
    Bulb = None


class XiaomiLightController:
    def __init__(self):
        self.bulb = None
        self.ip: Optional[str] = None

        self.last_hsv: Optional[Tuple[int, int, int]] = None
        self.music_mode_active = False

    def connect(self, ip: str):
        if not ip:
            raise RuntimeError("Xiaomi IP is empty.")

        if Bulb is None:
            raise RuntimeError("yeelight package is missing. Run: pip install yeelight")

        if self.ip == ip and self.bulb is not None:
            return

        self.ip = ip
        self.bulb = Bulb(ip)
        self.bulb.turn_on()

    def ensure_music_mode(self, ip: str, enabled: bool, dry_run: bool):
        if dry_run:
            self.music_mode_active = enabled
            return

        self.connect(ip)

        if enabled and not self.music_mode_active:
            try:
                self.bulb.start_music()
                self.music_mode_active = True
                print("[LIGHT] Music mode enabled")
            except Exception as e:
                self.music_mode_active = False
                raise RuntimeError(f"Could not enable music mode: {e}")

        if not enabled and self.music_mode_active:
            try:
                self.bulb.stop_music()
            except Exception:
                pass

            self.music_mode_active = False
            print("[LIGHT] Music mode disabled")

    def set_hsv_color(
        self,
        ip: str,
        hsv: Tuple[int, int, int],
        transition_ms: int,
        dry_run: bool,
        music_mode: bool,
        force: bool = False,
    ):
        hue, sat, brightness = hsv

        hue = int(np.clip(hue, 0, 359))
        sat = int(np.clip(sat, 1, 100))
        brightness = int(np.clip(brightness, 1, 100))

        hsv_cmd = (hue, sat, brightness)

        if not force and self._is_tiny_change(hsv_cmd):
            return

        self.last_hsv = hsv_cmd

        if dry_run:
            mode = "music" if music_mode else "normal"
            print(f"[DRY RUN] mode={mode}, HSV={hsv_cmd}")
            return

        self.ensure_music_mode(ip, music_mode, dry_run=False)

        try:
            # More reliable than set_scene("hsv", ...) on many Xiaomi/Yeelight devices.
            self.bulb.set_hsv(
                hue,
                sat,
                brightness,
                effect="smooth",
                duration=transition_ms,
            )
        except Exception as e:
            raise RuntimeError(f"set_hsv failed: {e}")

    def test_color(
        self,
        ip: str,
        color: str,
        dry_run: bool,
    ):
        colors = {
            "red": (0, 100, 70),
            "green": (120, 100, 70),
            "blue": (240, 100, 70),
            "purple": (280, 100, 70),
            "warm": (30, 55, 60),
            "white": (30, 5, 80),
        }

        if color not in colors:
            raise ValueError(f"Unknown test color: {color}")

        self.set_hsv_color(
            ip=ip,
            hsv=colors[color],
            transition_ms=300,
            dry_run=dry_run,
            music_mode=False,
            force=True,
        )

    def _is_tiny_change(self, hsv_cmd: Tuple[int, int, int]) -> bool:
        if self.last_hsv is None:
            return False

        hue_diff = abs(hsv_cmd[0] - self.last_hsv[0])
        hue_diff = min(hue_diff, 360 - hue_diff)

        sat_diff = abs(hsv_cmd[1] - self.last_hsv[1])
        brightness_diff = abs(hsv_cmd[2] - self.last_hsv[2])

        return hue_diff < 4 and sat_diff < 4 and brightness_diff < 4