From d94efbe1a753494ebaff864f1574bacfdcff6a7d Mon Sep 17 00:00:00 2001
From: Andrey <andvikt@gmail.com>
Date: Tue, 20 Apr 2021 09:11:00 +0300
Subject: [PATCH] add ADS1115 add DEVICE_CLASS_CO2

---
 custom_components/mega/i2c.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/custom_components/mega/i2c.py b/custom_components/mega/i2c.py
index ef9cf9f..46fea6c 100644
--- a/custom_components/mega/i2c.py
+++ b/custom_components/mega/i2c.py
@@ -6,6 +6,7 @@ from homeassistant.const import (
     DEVICE_CLASS_TEMPERATURE,
     DEVICE_CLASS_ILLUMINANCE,
     DEVICE_CLASS_PRESSURE,
+    DEVICE_CLASS_CO2,
     PERCENTAGE,
     LIGHT_LUX,
     TEMP_CELSIUS,
@@ -123,9 +124,21 @@ i2c_classes = {
         DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),  # сенсор встроенный в микросхему
     ],
     't67xx': [
-        DeviceType(None, CONCENTRATION_PARTS_PER_MILLION, None)  # для co2 нет класса в HA
+        DeviceType(DEVICE_CLASS_CO2, CONCENTRATION_PARTS_PER_MILLION, None)
     ],
     'tmp117': [
         DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
-    ]
+    ],
+    'ads1115': [
+        DeviceType(None, None, 'ch0'),
+        DeviceType(None, None, 'ch1'),
+        DeviceType(None, None, 'ch2'),
+        DeviceType(None, None, 'ch3'),
+    ],
+    'ads1015': [
+        DeviceType(None, None, 'ch0'),
+        DeviceType(None, None, 'ch1'),
+        DeviceType(None, None, 'ch2'),
+        DeviceType(None, None, 'ch3'),
+    ],
 }