Add night lamps support (#30)

Author: Mityushin

.gitignore

@@ -7,4 +7,7 @@ __pycache__
 /bin
 /lib
 /include
-/pyvenv.cfg
+/pyvenv.cfg
+.env.local
+venv/
+.idea

custom_components/sberdevices/light.py

@@ -30,6 +30,8 @@ def get_color_temp_range(device_type: str) -> (int, int):
             return 2000, 6500
         case "bulb":  # Sber A60 bulb
             return 2700, 6500
+        case "night_lamp":  # Sber night lamp
+            return 2700, 6500
         case _:
             return 2700, 6500
 
@@ -43,15 +45,15 @@ async def async_setup_entry(
 ) -> None:
     home: HomeAPI = hass.data[DOMAIN][entry.entry_id]["home"]
     await home.update_devices_cache()
+    light_types = ("bulb", "ledstrip", "night_lamp")
     async_add_entities(
         [
             SberLightEntity(
                 DeviceAPI(home, device["id"]),
-                "ledstrip" if "ledstrip" in device["image_set_type"] else "bulb",
+                next(t for t in light_types if t in device["image_set_type"]),
             )
             for device in home.get_cached_devices().values()
-            if "bulb" in device["image_set_type"]
-            or "ledstrip" in device["image_set_type"]  # TODO: lutiy kostyl'
+            if any(t in device["image_set_type"] for t in light_types)
         ]
     )
 

pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+pythonpath = .
+asyncio_mode = auto

requirements_dev.txt

@@ -0,0 +1 @@
+homeassistant

scripts/get_token.py

@@ -0,0 +1,56 @@
+"""
+Script to obtain SBER_ACCESS_TOKEN via OAuth2 flow.
+
+Run:
+    python scripts/get_token.py
+"""
+
+import asyncio
+import json
+import sys
+import os
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
+
+from custom_components.sberdevices.api import SberAPI
+
+
+async def main():
+    sber = SberAPI()
+
+    url = sber.create_authorization_url()
+    print("1. Откройте эту ссылку в браузере и авторизуйтесь в Сбербанк Онлайн:\n")
+    print(url)
+    print("\n2. После авторизации браузер перенаправит на URL вида:")
+    print("   companionapp://host?code=...&state=...\n")
+    print("3. Скопируйте ПОЛНЫЙ URL редиректа и вставьте сюда:\n")
+
+    redirect_url = input("URL: ").strip()
+
+    success = await sber.authorize_by_url(redirect_url)
+
+    if not success:
+        print("\nОшибка авторизации. Проверьте URL.")
+        return
+
+    token = sber.token
+
+    env_lines = [
+        f'SBER_ACCESS_TOKEN={token["access_token"]}',
+        f'SBER_REFRESH_TOKEN={token.get("refresh_token", "")}',
+        f'SBER_TOKEN_TYPE={token.get("token_type", "Bearer")}',
+        f'SBER_EXPIRES_AT={int(token.get("expires_at", 0))}',
+    ]
+
+    env_path = os.path.join(os.path.dirname(__file__), "..", ".env.local")
+    with open(env_path, "w") as f:
+        f.write("\n".join(env_lines) + "\n")
+
+    print(f"\nУспешно! Токен сохранён в .env.local\n")
+    for line in env_lines:
+        print(f"  {line}")
+    print(f"\nПолный токен (JSON):\n{json.dumps(token, indent=2)}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

tests/test_real_api.py

@@ -0,0 +1,63 @@
+"""
+Test that calls the real SberDevices API.
+
+Requires a valid OAuth token in .env.local (created by scripts/get_token.py)
+or in environment variables.
+
+Run:
+    pytest tests/test_real_api.py -s
+"""
+
+import json
+import os
+from pathlib import Path
+
+import pytest
+
+from custom_components.sberdevices.api import HomeAPI, SberAPI
+
+ENV_FILE = Path(__file__).resolve().parent.parent / ".env.local"
+
+
+def load_env_file():
+    """Load variables from .env.local if it exists."""
+    if not ENV_FILE.exists():
+        return
+    for line in ENV_FILE.read_text().splitlines():
+        line = line.strip()
+        if not line or line.startswith("#"):
+            continue
+        key, _, value = line.partition("=")
+        if key and _ and key not in os.environ:
+            os.environ[key] = value
+
+
+def load_token() -> dict:
+    load_env_file()
+
+    access_token = os.environ.get("SBER_ACCESS_TOKEN")
+    refresh_token = os.environ.get("SBER_REFRESH_TOKEN")
+
+    if not access_token:
+        pytest.skip("SBER_ACCESS_TOKEN not set — skipping real API test")
+
+    return {
+        "access_token": access_token,
+        "refresh_token": refresh_token or "",
+        "token_type": os.environ.get("SBER_TOKEN_TYPE", "Bearer"),
+        "expires_at": int(os.environ.get("SBER_EXPIRES_AT", "0")),
+    }
+
+
+@pytest.mark.asyncio
+async def test_get_device_tree():
+    """Fetch real device tree from SberDevices API."""
+    token = load_token()
+    sber = SberAPI(token=token)
+    home = HomeAPI(sber)
+
+    device_tree = await home.get_device_tree()
+
+    assert device_tree is not None
+    assert "devices" in device_tree
+    print(json.dumps(device_tree, indent=2, ensure_ascii=False))