60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
|
"""Simple implementation to call Home Assistant REST API."""
|
||
|
|
|
||
|
|
from homeassistant import exceptions
|
||
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
||
|
|
|
||
|
|
API_URL = "{proto}://{host}:{port}/api/remote_homeassistant/discovery"
|
||
|
|
|
||
|
|
|
||
|
|
class ApiProblem(exceptions.HomeAssistantError):
|
||
|
|
"""Error to indicate problem reaching API."""
|
||
|
|
|
||
|
|
|
||
|
|
class CannotConnect(exceptions.HomeAssistantError):
|
||
|
|
"""Error to indicate we cannot connect."""
|
||
|
|
|
||
|
|
|
||
|
|
class InvalidAuth(exceptions.HomeAssistantError):
|
||
|
|
"""Error to indicate there is invalid auth."""
|
||
|
|
|
||
|
|
|
||
|
|
class BadResponse(exceptions.HomeAssistantError):
|
||
|
|
"""Error to indicate a bad response was received."""
|
||
|
|
|
||
|
|
|
||
|
|
class UnsupportedVersion(exceptions.HomeAssistantError):
|
||
|
|
"""Error to indicate an unsupported version of Home Assistant."""
|
||
|
|
|
||
|
|
|
||
|
|
class EndpointMissing(exceptions.HomeAssistantError):
|
||
|
|
"""Error to indicate there is invalid auth."""
|
||
|
|
|
||
|
|
|
||
|
|
async def async_get_discovery_info(hass, host, port, secure, access_token, verify_ssl):
|
||
|
|
"""Get discovery information from server."""
|
||
|
|
url = API_URL.format(
|
||
|
|
proto="https" if secure else "http",
|
||
|
|
host=host,
|
||
|
|
port=port,
|
||
|
|
)
|
||
|
|
headers = {
|
||
|
|
"Authorization": "Bearer " + access_token,
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
}
|
||
|
|
session = async_get_clientsession(hass, verify_ssl)
|
||
|
|
|
||
|
|
# Fetch discovery info location for name and unique UUID
|
||
|
|
async with session.get(url, headers=headers) as resp:
|
||
|
|
if resp.status == 404:
|
||
|
|
raise EndpointMissing()
|
||
|
|
if 400 <= resp.status < 500:
|
||
|
|
raise InvalidAuth()
|
||
|
|
if resp.status != 200:
|
||
|
|
raise ApiProblem()
|
||
|
|
json = await resp.json()
|
||
|
|
if not isinstance(json, dict):
|
||
|
|
raise BadResponse(f"Bad response data: {json}")
|
||
|
|
if "uuid" not in json:
|
||
|
|
raise UnsupportedVersion()
|
||
|
|
return json
|