gateway.py

Go to the documentation of this file.

"""Handle Yale connection setup and authentication."""
 
import logging
from pathlib import Path
 
from aiohttp import ClientSession
from yalexs.authenticator_common import Authentication, AuthenticationState
from yalexs.manager.gateway import Gateway
 
from homeassistant.helpers import config_entry_oauth2_flow
 
_LOGGER = logging.getLogger(__name__)
 
 
class YaleGateway(Gateway):
    """Handle the connection to Yale."""
 
    def __init__(
        self,
        config_path: Path,
        aiohttp_session: ClientSession,
        oauth_session: config_entry_oauth2_flow.OAuth2Session,
    ) -> None:
        """Init the connection."""
        super().__init__(config_path, aiohttp_session)
        self._oauth_session_oauth_session = oauth_session
 
    async def async_get_access_token(self) -> str:
        """Get access token."""
        await self._oauth_session_oauth_session.async_ensure_token_valid()
        return self._oauth_session_oauth_session.token["access_token"]
 
    async def async_refresh_access_token_if_needed(self) -> None:
        """Refresh the access token if needed."""
        await self._oauth_session_oauth_session.async_ensure_token_valid()
 
    async def async_authenticate(self) -> Authentication:
        """Authenticate with the details provided to setup."""
        await self._oauth_session_oauth_session.async_ensure_token_valid()
        self.authenticationauthentication = Authentication(
            AuthenticationState.AUTHENTICATED, None, None, None
        )
        return self.authenticationauthentication