From 41054296396def7d004011491e4671350289c079 Mon Sep 17 00:00:00 2001
From: Thibault Cohen <titilambert@gmail.com>
Date: Tue, 29 May 2018 10:23:12 -0400
Subject: [PATCH] Add asyncio support for Ebox (#14183)

* Fix Ebox sensor

* Fix #14183 comments

* Update ebox.py

* Update ebox.py

* Continue fixing comments
---
 homeassistant/components/sensor/ebox.py | 45 +++++++++++++++----------
 requirements_all.txt                    |  3 ++
 2 files changed, 30 insertions(+), 18 deletions(-)

diff --git a/homeassistant/components/sensor/ebox.py b/homeassistant/components/sensor/ebox.py
index aca2d7bdb9a..d7b867081a3 100644
--- a/homeassistant/components/sensor/ebox.py
+++ b/homeassistant/components/sensor/ebox.py
@@ -9,7 +9,6 @@ https://home-assistant.io/components/sensor.ebox/
 import logging
 from datetime import timedelta
 
-import requests
 import voluptuous as vol
 
 import homeassistant.helpers.config_validation as cv
@@ -18,9 +17,11 @@ from homeassistant.const import (
     CONF_USERNAME, CONF_PASSWORD,
     CONF_NAME, CONF_MONITORED_VARIABLES)
 from homeassistant.helpers.entity import Entity
+from homeassistant.util import Throttle
+from homeassistant.exceptions import PlatformNotReady
 
-# pylint: disable=import-error
-REQUIREMENTS = []  # ['pyebox==0.1.0'] - disabled because it breaks pip10
+
+REQUIREMENTS = ['pyebox==1.1.4']
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -32,7 +33,8 @@ PERCENT = '%'  # type: str
 DEFAULT_NAME = 'EBox'
 
 REQUESTS_TIMEOUT = 15
-SCAN_INTERVAL = timedelta(minutes=5)
+SCAN_INTERVAL = timedelta(minutes=15)
+MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=15)
 
 SENSOR_TYPES = {
     'usage': ['Usage', PERCENT, 'mdi:percent'],
@@ -62,25 +64,29 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
 })
 
 
-def setup_platform(hass, config, add_devices, discovery_info=None):
+async def async_setup_platform(hass, config, async_add_devices,
+                               discovery_info=None):
     """Set up the EBox sensor."""
     username = config.get(CONF_USERNAME)
     password = config.get(CONF_PASSWORD)
 
-    try:
-        ebox_data = EBoxData(username, password)
-        ebox_data.update()
-    except requests.exceptions.HTTPError as error:
-        _LOGGER.error("Failed login: %s", error)
-        return False
+    httpsession = hass.helpers.aiohttp_client.async_get_clientsession()
+    ebox_data = EBoxData(username, password, httpsession)
 
     name = config.get(CONF_NAME)
 
+    from pyebox.client import PyEboxError
+    try:
+        await ebox_data.async_update()
+    except PyEboxError as exp:
+        _LOGGER.error("Failed login: %s", exp)
+        raise PlatformNotReady
+
     sensors = []
     for variable in config[CONF_MONITORED_VARIABLES]:
         sensors.append(EBoxSensor(ebox_data, variable, name))
 
-    add_devices(sensors, True)
+    async_add_devices(sensors, True)
 
 
 class EBoxSensor(Entity):
@@ -116,9 +122,9 @@ class EBoxSensor(Entity):
         """Icon to use in the frontend, if any."""
         return self._icon
 
-    def update(self):
+    async def async_update(self):
         """Get the latest data from EBox and update the state."""
-        self.ebox_data.update()
+        await self.ebox_data.async_update()
         if self.type in self.ebox_data.data:
             self._state = round(self.ebox_data.data[self.type], 2)
 
@@ -126,18 +132,21 @@ class EBoxSensor(Entity):
 class EBoxData(object):
     """Get data from Ebox."""
 
-    def __init__(self, username, password):
+    def __init__(self, username, password, httpsession):
         """Initialize the data object."""
         from pyebox import EboxClient
-        self.client = EboxClient(username, password, REQUESTS_TIMEOUT)
+        self.client = EboxClient(username, password,
+                                 REQUESTS_TIMEOUT, httpsession)
         self.data = {}
 
-    def update(self):
+    @Throttle(MIN_TIME_BETWEEN_UPDATES)
+    async def async_update(self):
         """Get the latest data from Ebox."""
         from pyebox.client import PyEboxError
         try:
-            self.client.fetch_data()
+            await self.client.fetch_data()
         except PyEboxError as exp:
             _LOGGER.error("Error on receive last EBox data: %s", exp)
             return
+        # Update data
         self.data = self.client.get_data()
diff --git a/requirements_all.txt b/requirements_all.txt
index 3e7b966af4f..958b0f1027e 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -774,6 +774,9 @@ pydispatcher==2.0.5
 # homeassistant.components.android_ip_webcam
 pydroid-ipcam==0.8
 
+# homeassistant.components.sensor.ebox
+pyebox==1.1.4
+
 # homeassistant.components.climate.econet
 pyeconet==0.0.5