home-automation/apps/abstraction/vendors/tasmota.py

"""Tasmota vendor transformations."""

import logging
from typing import Any

logger = logging.getLogger(__name__)


def transform_relay_to_vendor(payload: dict[str, Any]) -> str:
    """Transform abstract relay payload to Tasmota format.
    
    Tasmota expects plain text 'on' or 'off' (not JSON).
    
    Example:
    - Abstract: {'power': 'on'}
    - Tasmota: 'on'
    """
    power = payload.get("power", "off")
    return power


def transform_relay_to_abstract(payload: str) -> dict[str, Any]:
    """Transform Tasmota relay payload to abstract format.
    
    Tasmota sends plain text 'ON' or 'OFF'.
    
    Example:
    - Tasmota: 'ON'
    - Abstract: {'power': 'on'}
    """
    return {"power": payload.strip().lower()}


# Registry of handlers for this vendor
HANDLERS = {
    ("relay", "to_vendor"): transform_relay_to_vendor,
    ("relay", "to_abstract"): transform_relay_to_abstract,
}