53 lines
2.0 KiB
Python
53 lines
2.0 KiB
Python
import logging
|
|
from inspect import iscoroutinefunction
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_USERNAME, CONF_PASSWORD
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryAuthFailed
|
|
from wyzeapy.exceptions import AccessTokenError, LoginError
|
|
from wyzeapy.wyze_auth_lib import Token
|
|
|
|
from .const import DOMAIN, ACCESS_TOKEN, REFRESH_TOKEN, REFRESH_TIME
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class TokenManager:
|
|
hass: HomeAssistant = None
|
|
config_entry: ConfigEntry = None
|
|
|
|
def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry):
|
|
TokenManager.hass = hass
|
|
TokenManager.config_entry = config_entry
|
|
|
|
@staticmethod
|
|
async def token_callback(token: Token = None):
|
|
_LOGGER.debug("TokenManager: Received new token, updating config entry.")
|
|
if TokenManager.hass.config_entries.async_entries(DOMAIN):
|
|
for entry in TokenManager.hass.config_entries.async_entries(DOMAIN):
|
|
TokenManager.hass.config_entries.async_update_entry(
|
|
entry,
|
|
data={
|
|
CONF_USERNAME: entry.data.get(CONF_USERNAME),
|
|
CONF_PASSWORD: entry.data.get(CONF_PASSWORD),
|
|
ACCESS_TOKEN: token.access_token,
|
|
REFRESH_TOKEN: token.refresh_token,
|
|
REFRESH_TIME: str(token.refresh_time),
|
|
},
|
|
)
|
|
|
|
|
|
def token_exception_handler(func):
|
|
async def inner_function(*args, **kwargs):
|
|
try:
|
|
if iscoroutinefunction(func):
|
|
await func(*args, **kwargs)
|
|
else:
|
|
func(*args, **kwargs)
|
|
except (AccessTokenError, LoginError) as err:
|
|
_LOGGER.error("TokenManager detected a login issue please re-login.")
|
|
raise ConfigEntryAuthFailed("Unable to login, please re-login.") from err
|
|
|
|
return inner_function
|