From d33c3927e76ab73a9a29c89704137010fbdebb7d Mon Sep 17 00:00:00 2001
From: hudashot <17239583+hudashot@users.noreply.github.com>
Date: Mon, 4 Dec 2017 10:11:45 +0000
Subject: [PATCH 1/2] Add more climate states to state_as_number

---
 homeassistant/helpers/state.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/homeassistant/helpers/state.py b/homeassistant/helpers/state.py
index 8b98bfadb68e80..254a48c3d0a8e8 100644
--- a/homeassistant/helpers/state.py
+++ b/homeassistant/helpers/state.py
@@ -21,7 +21,8 @@
     ATTR_HUMIDITY, ATTR_OPERATION_MODE, ATTR_SWING_MODE,
     SERVICE_SET_AUX_HEAT, SERVICE_SET_AWAY_MODE, SERVICE_SET_HOLD_MODE,
     SERVICE_SET_FAN_MODE, SERVICE_SET_HUMIDITY, SERVICE_SET_OPERATION_MODE,
-    SERVICE_SET_SWING_MODE, SERVICE_SET_TEMPERATURE)
+    SERVICE_SET_SWING_MODE, SERVICE_SET_TEMPERATURE, STATE_HEAT, STATE_COOL,
+    STATE_IDLE)
 from homeassistant.components.climate.ecobee import (
     ATTR_FAN_MIN_ON_TIME, SERVICE_SET_FAN_MIN_ON_TIME,
     ATTR_RESUME_ALL, SERVICE_RESUME_PROGRAM)
@@ -210,10 +211,11 @@ def state_as_number(state):
     Raises ValueError if this is not possible.
     """
     if state.state in (STATE_ON, STATE_LOCKED, STATE_ABOVE_HORIZON,
-                       STATE_OPEN, STATE_HOME):
+                       STATE_OPEN, STATE_HOME, STATE_HEAT, STATE_COOL):
         return 1
     elif state.state in (STATE_OFF, STATE_UNLOCKED, STATE_UNKNOWN,
-                         STATE_BELOW_HORIZON, STATE_CLOSED, STATE_NOT_HOME):
+                         STATE_BELOW_HORIZON, STATE_CLOSED, STATE_NOT_HOME,
+                         STATE_IDLE):
         return 0
 
     return float(state.state)

From 38cab87bea7af44da772c62da236517a8b0a733e Mon Sep 17 00:00:00 2001
From: hudashot <17239583+hudashot@users.noreply.github.com>
Date: Mon, 4 Dec 2017 10:11:34 +0000
Subject: [PATCH 2/2] Export climate metrics to Prometheus.

This adds climate_state and temperature_c metrics for each climate
device.
---
 homeassistant/components/prometheus.py | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/prometheus.py b/homeassistant/components/prometheus.py
index 0396cafd4ffaa4..ab0ac8841befdb 100644
--- a/homeassistant/components/prometheus.py
+++ b/homeassistant/components/prometheus.py
@@ -14,7 +14,8 @@
 from homeassistant.components import recorder
 from homeassistant.const import (
     CONF_DOMAINS, CONF_ENTITIES, CONF_EXCLUDE, CONF_INCLUDE, TEMP_CELSIUS,
-    EVENT_STATE_CHANGED, TEMP_FAHRENHEIT, CONTENT_TYPE_TEXT_PLAIN)
+    EVENT_STATE_CHANGED, TEMP_FAHRENHEIT, CONTENT_TYPE_TEXT_PLAIN,
+    ATTR_TEMPERATURE, ATTR_UNIT_OF_MEASUREMENT)
 from homeassistant import core as hacore
 from homeassistant.helpers import state as state_helper
 from homeassistant.util.temperature import fahrenheit_to_celsius
@@ -159,6 +160,26 @@ def _handle_lock(self, state):
         value = state_helper.state_as_number(state)
         metric.labels(**self._labels(state)).set(value)
 
+    def _handle_climate(self, state):
+        temp = state.attributes.get(ATTR_TEMPERATURE)
+        if temp:
+            unit = state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
+            if unit == TEMP_FAHRENHEIT:
+                temp = fahrenheit_to_celsius(temp)
+            metric = self._metric(
+                'temperature_c', self.prometheus_client.Gauge,
+                'Temperature in degrees Celsius')
+            metric.labels(**self._labels(state)).set(temp)
+
+        metric = self._metric(
+            'climate_state', self.prometheus_client.Gauge,
+            'State of the thermostat (0/1)')
+        try:
+            value = state_helper.state_as_number(state)
+            metric.labels(**self._labels(state)).set(value)
+        except ValueError:
+            pass
+
     def _handle_sensor(self, state):
         _sensor_types = {
             TEMP_CELSIUS: (
@@ -191,7 +212,7 @@ def _handle_sensor(self, state):
             ),
         }
 
-        unit = state.attributes.get('unit_of_measurement')
+        unit = state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
         metric = _sensor_types.get(unit)
 
         if metric is not None: