Added Code for fpl

custom_components/fpl/fplapi.py

@@ -0,0 +1,119 @@
+import asyncio
+import logging
+import re
+from datetime import timedelta, date
+
+import aiohttp
+import async_timeout
+from bs4 import BeautifulSoup
+
+_LOGGER = logging.getLogger(__name__)
+TIMEOUT = 5
+
+
+class FplApi(object):
+    """A class for getting energy usage information from Florida Power & Light."""
+
+    def __init__(self, username, password, is_tou, loop, session):
+        """Initialize the data retrieval. Session should have BasicAuth flag set."""
+        self._username = username
+        self._password = password
+        self._loop = loop
+        self._session = session
+        self._is_tou = is_tou
+        self._account_number = None
+        self._premise_number = None
+
+        self.yesterday_kwh = None
+        self.yesterday_dollars = None
+        self.mtd_kwh = None
+        self.mtd_dollars = None
+        self.projected_bill = None
+
+    async def login(self):
+        """login and get account information"""
+        async with async_timeout.timeout(TIMEOUT, loop=self._loop):
+            response = await self._session.get("https://www.fpl.com/api/resources/login",
+                                               auth=aiohttp.BasicAuth(self._username, self._password))
+
+        if (await response.json())["messages"][0]["messageCode"] != "login.success":
+            raise Exception('login failure')
+
+        async with async_timeout.timeout(TIMEOUT, loop=self._loop):
+            response = await self._session.get("https://www.fpl.com/api/resources/header")
+        json = await response.json()
+        self._account_number = json["data"]["selectedAccount"]["data"]["accountNumber"]
+        self._premise_number = json["data"]["selectedAccount"]["data"]["acctSecSettings"]["premiseNumber"]
+
+    async def async_get_yesterday_usage(self):
+        async with async_timeout.timeout(TIMEOUT, loop=self._loop):
+            response = await self._session.get(self._build_daily_url())
+
+        _LOGGER.debug("Response from API: %s", response.status)
+
+        if response.status != 200:
+            self.data = None
+            return
+
+        malformedXML = await response.read()
+
+        cleanerXML = str(malformedXML).replace(
+            '<?xml version="1.0" encoding="UTF-8" standalone="yes"?>', '', 1
+        ).split("<ARG>@@", 1)[0]
+
+        soup = BeautifulSoup(cleanerXML, 'html.parser')
+
+        tool_text = soup.find("dataset", seriesname="$") \
+            .find("set")["tooltext"]
+
+        match = re.search(r"\{br\}kWh Usage: (.*?) kWh \{br\}", tool_text)
+        if match:
+            self.yesterday_kwh = match.group(1)
+
+        match2 = re.search(r"\{br\}Approx\. Cost: (\$.*?) \{br\}", tool_text)
+        if match2:
+            self.yesterday_dollars = match2.group(1)
+
+    async def async_get_mtd_usage(self):
+        async with async_timeout.timeout(TIMEOUT, loop=self._loop):
+            response = await self._session.get(
+                "https://app.fpl.com/wps/myportal/EsfPortal")
+
+        soup = BeautifulSoup(await response.text(), 'html.parser')
+
+        self.mtd_kwh = soup.find(id="bpbsubcontainer") \
+            .find("table", class_="bpbtab_style_bill", width=430) \
+            .find_all("div", class_="bpbtabletxt")[-1].string
+
+        self.mtd_dollars = soup \
+            .find_all("div", class_="bpbusagebgnd")[1] \
+            .find("div", class_="bpbusagedollartxt").getText().strip()
+
+        self.projected_bill = soup.find(id="bpssmlsubcontainer") \
+            .find("div", class_="bpsmonthbillbgnd") \
+            .find("div", class_="bpsmnthbilldollartxt") \
+            .getText().strip().replace("$", "")
+
+    def _build_daily_url(self):
+        end_date = date.today()
+        start_date = end_date - timedelta(days=1)
+
+        return ("https://app.fpl.com/wps/PA_ESFPortalWeb/getDailyConsumption"
+                "?premiseNumber={premise_number}"
+                "&accountNumber={account_number}"
+                "&isTouUser={is_tou}"
+                "&startDate={start_date}"
+                "&endDate={end_date}"
+                "&userType=EXT"
+                "&isResidential=true"
+                "&certifiedDate=2000/01/01"
+                "&viewType=dollar"
+                "&tempType=max"
+                "&ecDayHumType=NoHum"
+                ).format(
+            premise_number=self._premise_number,
+            account_number=self._account_number,
+            is_tou=str(self._is_tou),
+            start_date=start_date.strftime("%Y%m%d"),
+            end_date=end_date.strftime("%Y%m%d"),
+        )