111 lines
3.0 KiB
Python
111 lines
3.0 KiB
Python
import logging
|
|
from datetime import datetime, timedelta
|
|
from .fplapi import FplApi
|
|
import aiohttp
|
|
import asyncio
|
|
from homeassistant.helpers.entity import Entity
|
|
from homeassistant import util
|
|
|
|
from homeassistant.const import (
|
|
CONF_NAME,
|
|
EVENT_CORE_CONFIG_UPDATE,
|
|
STATE_UNKNOWN,
|
|
CONF_USERNAME,
|
|
CONF_PASSWORD,
|
|
STATE_UNKNOWN,
|
|
)
|
|
from .const import DOMAIN, ICON, LOGIN_RESULT_OK
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
MIN_TIME_BETWEEN_SCANS = timedelta(minutes=30)
|
|
MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=60)
|
|
|
|
|
|
def setup(hass, config):
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
username = config_entry.data.get(CONF_USERNAME)
|
|
password = config_entry.data.get(CONF_PASSWORD)
|
|
|
|
session = aiohttp.ClientSession()
|
|
try:
|
|
api = FplApi(username, password, hass.loop, session)
|
|
result = await api.login()
|
|
|
|
fpl_accounts = []
|
|
|
|
if result == LOGIN_RESULT_OK:
|
|
accounts = await api.async_get_open_accounts()
|
|
for account in accounts:
|
|
_LOGGER.info(f"Adding fpl account: {account}")
|
|
fpl_accounts.append(FplSensor(hass, config_entry.data, account))
|
|
|
|
async_add_entities(fpl_accounts)
|
|
|
|
except Exception as e: # pylint: disable=broad-except
|
|
_LOGGER.error(f"Adding fpl accounts: {e}")
|
|
|
|
await session.close()
|
|
|
|
|
|
class FplSensor(Entity):
|
|
def __init__(self, hass, config, account):
|
|
self._config = config
|
|
self.username = config.get(CONF_USERNAME)
|
|
self.password = config.get(CONF_PASSWORD)
|
|
self._state = STATE_UNKNOWN
|
|
self.loop = hass.loop
|
|
|
|
self._account = account
|
|
self._data = None
|
|
|
|
async def async_added_to_hass(self):
|
|
await self.async_update()
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the ID of this device."""
|
|
id = "{}{}".format(DOMAIN, self._account)
|
|
return id
|
|
|
|
@property
|
|
def name(self):
|
|
return f"{DOMAIN.upper()} {self._account}"
|
|
|
|
@property
|
|
def state(self):
|
|
return self._state #
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
return " "
|
|
|
|
@property
|
|
def icon(self):
|
|
return ICON
|
|
|
|
@property
|
|
def state_attributes(self):
|
|
return self._data
|
|
|
|
@util.Throttle(MIN_TIME_BETWEEN_SCANS, MIN_TIME_BETWEEN_UPDATES)
|
|
async def async_update(self):
|
|
self._state = STATE_UNKNOWN
|
|
self._data = None
|
|
session = aiohttp.ClientSession()
|
|
try:
|
|
api = FplApi(self.username, self.password, self.loop, session)
|
|
await api.login()
|
|
self._data = await api.async_get_data(self._account)
|
|
if self._data["budget_bill"]:
|
|
self._state = self._data["projected_budget_bill"]
|
|
else:
|
|
self._state = self._data["projected_bill"]
|
|
except Exception: # pylint: disable=broad-except
|
|
pass
|
|
|
|
await session.close()
|