bachya/simplisafe-python

"""Test system functionality with an Auth0 code/verifier."""

import asyncio
import logging
import os

from aiohttp import ClientSession

from simplipy import API
from simplipy.errors import CannotConnectError, SimplipyError

_LOGGER = logging.getLogger()

SIMPLISAFE_REFRESH_TOKEN = os.getenv("SIMPLISAFE_REFRESH_TOKEN", "")


async def main() -> None:
    """Create the aiohttp session and run the example."""
    async with ClientSession() as session:
        logging.basicConfig(level=logging.DEBUG)
        if not SIMPLISAFE_REFRESH_TOKEN:
            _LOGGER.error(
                "You must specify a SIMPLISAFE_REFRESH_TOKEN in the environment."
            )
            return

        try:
            simplisafe = await API.async_from_refresh_token(
                SIMPLISAFE_REFRESH_TOKEN, session=session
            )

            if simplisafe.websocket:
                try:
                    await simplisafe.websocket.async_connect()
                except CannotConnectError as err:
                    _LOGGER.error(
                        "There was a error while connecting to the server: %s", err
                    )

                await simplisafe.websocket.async_listen()
        except SimplipyError as err:
            _LOGGER.error(err)
        except KeyboardInterrupt:
            pass


asyncio.run(main())