core/homeassistant/components/home_connect/api.py

29 lines
953 B
Python

"""API for Home Connect bound to HASS OAuth."""
from aiohomeconnect.client import AbstractAuth
from aiohomeconnect.const import API_ENDPOINT
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_entry_oauth2_flow
from homeassistant.helpers.httpx_client import get_async_client
class AsyncConfigEntryAuth(AbstractAuth):
"""Provide Home Connect authentication tied to an OAuth2 based config entry."""
def __init__(
self,
hass: HomeAssistant,
oauth_session: config_entry_oauth2_flow.OAuth2Session,
) -> None:
"""Initialize Home Connect Auth."""
self.hass = hass
super().__init__(get_async_client(hass), host=API_ENDPOINT)
self.session = oauth_session
async def async_get_access_token(self) -> str:
"""Return a valid access token."""
await self.session.async_ensure_token_valid()
return self.session.token["access_token"]