Suez water

.coveragerc

@@ -581,6 +581,7 @@ omit =
     homeassistant/components/stiebel_eltron/*
     homeassistant/components/streamlabswater/*
     homeassistant/components/stride/notify.py
+    homeassistant/components/suez_water/*
     homeassistant/components/supervisord/sensor.py
     homeassistant/components/swiss_hydrological_data/sensor.py
     homeassistant/components/swiss_public_transport/sensor.py

CODEOWNERS

@@ -239,6 +239,7 @@ homeassistant/components/sql/* @dgomes
 homeassistant/components/statistics/* @fabaff
 homeassistant/components/stiebel_eltron/* @fucm
 homeassistant/components/stream/* @hunterjm
+homeassistant/components/suez_water/* @ooii
 homeassistant/components/sun/* @Swamp-Ig
 homeassistant/components/supla/* @mwegrzynek
 homeassistant/components/swiss_hydrological_data/* @fabaff

homeassistant/components/suez_water/__init__.py

@@ -0,0 +1 @@
+"""France Suez Water integration."""

homeassistant/components/suez_water/manifest.json

@@ -0,0 +1,8 @@
+{
+  "domain": "suez_water",
+  "name": "Suez Water Consumption Sensor",
+  "documentation": "https://www.home-assistant.io/components/suez_water",
+  "dependencies": [],
+  "codeowners": ["@ooii"],
+  "requirements": ["pysuez==0.1.17"]
+}

homeassistant/components/suez_water/sensor.py

@@ -0,0 +1,134 @@
+"""Sensor for Suez Water Consumption data."""
+from datetime import timedelta
+import logging
+
+import voluptuous as vol
+
+from pysuez.client import PySuezError
+from pysuez import SuezClient
+
+from homeassistant.components.sensor import PLATFORM_SCHEMA
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, VOLUME_LITERS
+import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.entity import Entity
+
+_LOGGER = logging.getLogger(__name__)
+CONF_COUNTER_ID = 'counter_id'
+
+SCAN_INTERVAL = timedelta(hours=12)
+
+COMPONENT_ICON = 'mdi:water-pump'
+COMPONENT_NAME = "Suez Water Client"
+
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Required(CONF_USERNAME): cv.string,
+    vol.Required(CONF_PASSWORD): cv.string,
+    vol.Required(CONF_COUNTER_ID): cv.string,
+})
+
+
+def setup_platform(hass, config, add_entities, discovery_info=None):
+    """Set up the sensor platform."""
+    username = config[CONF_USERNAME]
+    password = config[CONF_PASSWORD]
+    counter_id = config[CONF_COUNTER_ID]
+    try:
+        client = SuezClient(
+            username, password, counter_id)
+
+        if not client.check_credentials():
+            _LOGGER.warning("Wrong username and/or password")
+        else:
+            add_entities([SuezSensor(client)], True)
+
+    except PySuezError:
+        _LOGGER.warning("Unable to create Suez Client")
+        return
+
+
+class SuezSensor(Entity):
+    """Representation of a Sensor."""
+
+    def __init__(self, client):
+        """Initialize the data object."""
+        self._name = COMPONENT_NAME
+        self._attributes = {}
+        self._state = None
+        self._available = None
+        self._icon = COMPONENT_ICON
+        self.client = client
+        self.success = False
+
+    @property
+    def name(self):
+        """Return the name of the sensor."""
+        return self._name
+
+    @property
+    def state(self):
+        """Return the state of the sensor."""
+        return self._state
+
+    @property
+    def unit_of_measurement(self):
+        """Return the unit of measurement."""
+        return VOLUME_LITERS
+
+    @property
+    def device_state_attributes(self):
+        """Return the state attributes."""
+        return self._attributes
+
+    @property
+    def icon(self):
+        """Return the icon of the sensor."""
+        return self._icon
+
+    def _fetch_data(self):
+        """Fetch latest data from Suez."""
+        try:
+            self.client.update()
+            # _state holds the volume of consumed water during previous day
+            self._state = self.client.state
+            self.success = True
+            self._available = True
+            self._attributes[
+                'attribution'] = self.client.attributes[
+                    'attribution']
+            self._attributes['thisMonthConsumption'] = {}
+            for item in self.client.attributes['thisMonthConsumption']:
+                self._attributes['thisMonthConsumption'][
+                    item] = self.client.attributes[
+                        'thisMonthConsumption'][item]
+            self._attributes['previousMonthConsumption'] = {}
+            for item in self.client.attributes['previousMonthConsumption']:
+                self._attributes['previousMonthConsumption'][
+                    item] = self.client.attributes[
+                        'previousMonthConsumption'][item]
+            self._attributes[
+                'highestMonthlyConsumption'] = self.client.attributes[
+                    'highestMonthlyConsumption']
+            self._attributes[
+                'lastYearOverAll'] = self.client.attributes[
+                    'lastYearOverAll']
+            self._attributes[
+                'thisYearOverAll'] = self.client.attributes[
+                    'thisYearOverAll']
+            self._attributes['history'] = {}
+            for item in self.client.attributes['history']:
+                self._attributes[
+                    'history'][item] = self.client.attributes[
+                        'history'][item]
+
+        except PySuezError:
+            self.success = False
+            self._available = False
+            _LOGGER.warning("Unable to fetch data")
+            return
+
+    def update(self):
+        """Return the latest collected data from Linky."""
+        self._fetch_data()
+        _LOGGER.debug(
+            "Suez data state is: %s, and the success is %s",
+            self._state, self.success)

requirements_all.txt

@@ -342,6 +342,9 @@ datadog==0.15.0
 # homeassistant.components.metoffice
 datapoint==0.4.3
 
+# homeassistant.components.suez_water
+datetime==4.3
+
 # homeassistant.components.decora
 # decora==0.6
 
@@ -1375,6 +1378,9 @@ pystiebeleltron==0.0.1.dev2
 # homeassistant.components.stride
 pystride==0.1.7
 
+# homeassistant.components.suez_water
+pysuez==0.1.17
+
 # homeassistant.components.supla
 pysupla==0.0.3
 
@@ -1586,6 +1592,12 @@ recollect-waste==1.0.1
 # homeassistant.components.rainmachine
 regenmaschine==1.5.1
 
+# homeassistant.components.suez_water
+regex==2019.4.14
+
+# homeassistant.components.suez_water
+requests==2.21.0
+
 # homeassistant.components.python_script
 restrictedpython==4.0b8
 
@@ -1837,6 +1849,9 @@ vigilancemeteo==3.0.0
 # homeassistant.components.volkszaehler
 volkszaehler==0.1.2
 
+# homeassistant.components.suez_water
+voluptuous==0.11.5
+
 # homeassistant.components.volvooncall
 volvooncall==0.8.7