__init__.py

"""
1. Place file here: /config/custom_components/sonofftest/__init__.py
2. Add to configuration.yaml: `sonofftest: 192.168.1.123` (IP of sonoff)
3. Block the Sonoff from accessing the Internet
4. Restart Sonoff
5. Wait Sonoff LED to start blinking (two blinks every 2 seconds)
6. Restart HA
"""
import asyncio
import logging

import websockets

_LOGGER = logging.getLogger(__name__)

DOMAIN = 'sonofftest'


async def async_setup(hass, hass_config):
    host = hass_config[DOMAIN]

    try:
        _LOGGER.warning(f"connect to {host}")
        coro = websockets.connect(f"ws://{host}:8081/", subprotocols=['chat'])
        ws = await asyncio.wait_for(coro, 10)
        _LOGGER.warning("connected")
        await ws.send('ping')
        _LOGGER.warning("end")
    except Exception as e:
        _LOGGER.error(f"Error: {type(e)}")

    return True