async def test_devices():
async with aiohttp.ClientSession() as client:
auth = AlmondLocalAuth('http://127.0.0.1:3000', client)
almond_api = WebAlmondAPI(auth)
await almond_api.async_create_simple_device('com.xkcd')
devices = await almond_api.async_list_devices()
assert any(d['kind'] == 'com.xkcd' for d in devices)
async def async_verify_local_connection(opp: core.OpenPeerPower, host: str):
"""Verify that a local connection works."""
websession = aiohttp_client.async_get_clientsession(opp)
api = WebAlmondAPI(AlmondLocalAuth(host, websession))
try:
with async_timeout.timeout(10):
await api.async_list_apps()
return True
except (asyncio.TimeoutError, ClientError):
return False
async def async_setup_entry(hass: HomeAssistant,
entry: config_entries.ConfigEntry):
"""Set up Almond config entry."""
websession = aiohttp_client.async_get_clientsession(hass)
if entry.data["type"] == TYPE_LOCAL:
auth = AlmondLocalAuth(entry.data["host"], websession)
else:
# OAuth2
implementation = (
await
config_entry_oauth2_flow.async_get_config_entry_implementation(
hass, entry))
oauth_session = config_entry_oauth2_flow.OAuth2Session(
hass, entry, implementation)
auth = AlmondOAuth(entry.data["host"], websession, oauth_session)
api = WebAlmondAPI(auth)
agent = AlmondAgent(hass, api, entry)
# Hass.io does its own configuration.
if not entry.data.get("is_hassio"):
# If we're not starting or local, set up Almond right away
if hass.state != CoreState.not_running or entry.data[
"type"] == TYPE_LOCAL:
await _configure_almond_for_ha(hass, entry, api)
else:
# OAuth2 implementations can potentially rely on the HA Cloud url.
# This url is not be available until 30 seconds after boot.
async def configure_almond(_now):
try:
await _configure_almond_for_ha(hass, entry, api)
except ConfigEntryNotReady:
_LOGGER.warning(
"Unable to configure Almond to connect to Home Assistant"
)
async def almond_hass_start(_event):
event.async_call_later(hass, ALMOND_SETUP_DELAY,
configure_almond)
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_START,
almond_hass_start)
conversation.async_set_agent(hass, agent)
return True
async def test_simple():
async with aiohttp.ClientSession() as client:
auth = AlmondLocalAuth('http://127.0.0.1:3000', client)
almond_api = WebAlmondAPI(auth)
# issue the first command and discard the result (this will cause
# Almond to spew the welcome messages)
await almond_api.async_converse_text("hello")
result = await almond_api.async_converse_text("hello")
assert result == {
"askSpecial": None,
"messages": [{
"type": "text",
"text": "Hi!",
"icon": None
}],
"conversationId": "stateless"
}
async def async_setup_entry(hass, entry):
"""Set up Almond config entry."""
websession = aiohttp_client.async_get_clientsession(hass)
if entry.data["type"] == TYPE_LOCAL:
auth = AlmondLocalAuth(entry.data["host"], websession)
else:
# OAuth2
implementation = await config_entry_oauth2_flow.async_get_config_entry_implementation(
hass, entry)
oauth_session = config_entry_oauth2_flow.OAuth2Session(
hass, entry, implementation)
auth = AlmondOAuth(entry.data["host"], websession, oauth_session)
api = WebAlmondAPI(auth)
agent = AlmondAgent(hass, api, entry)
# Hass.io does its own configuration of Almond.
if entry.data.get("is_hassio") or entry.data["type"] != TYPE_LOCAL:
conversation.async_set_agent(hass, agent)
return True
# Configure Almond to connect to Home Assistant
store = storage.Store(hass, STORAGE_VERSION, STORAGE_KEY)
data = await store.async_load()
if data is None:
data = {}
user = None
if "almond_user" in data:
user = await hass.auth.async_get_user(data["almond_user"])
if user is None:
user = await hass.auth.async_create_system_user(
"Almond", [GROUP_ID_ADMIN])
data["almond_user"] = user.id
await store.async_save(data)
refresh_token = await hass.auth.async_create_refresh_token(
user,
# Almond will be fine as long as we restart once every 5 years
access_token_expiration=timedelta(days=365 * 5),
)
# Create long lived access token
access_token = hass.auth.async_create_access_token(refresh_token)
# Store token in Almond
try:
with async_timeout.timeout(10):
await api.async_create_device({
"kind":
"io.home-assistant",
"hassUrl":
hass.config.api.base_url,
"accessToken":
access_token,
"refreshToken":
"",
# 5 years from now in ms.
"accessTokenExpires":
(time.time() + 60 * 60 * 24 * 365 * 5) * 1000,
})
except (asyncio.TimeoutError, ClientError) as err:
if isinstance(err, asyncio.TimeoutError):
msg = "Request timeout"
else:
msg = err
_LOGGER.warning("Unable to configure Almond: %s", msg)
await hass.auth.async_remove_refresh_token(refresh_token)
raise ConfigEntryNotReady
# Clear all other refresh tokens
for token in list(user.refresh_tokens.values()):
if token.id != refresh_token.id:
await hass.auth.async_remove_refresh_token(token)
conversation.async_set_agent(hass, agent)
return True
async def test_apps():
async with aiohttp.ClientSession() as client:
auth = AlmondLocalAuth('http://127.0.0.1:3000', client)
almond_api = WebAlmondAPI(auth)
assert await almond_api.async_list_apps() == []
