- add many i2c sensors

2025-12-13 10:04:28 +05:00 · 2023-10-15 21:09:46 +03:00
parent bd98fa216d
commit 39696b054f
4 changed files with 183 additions and 90 deletions
--- a/custom_components/mega/i2c.py
+++ b/custom_components/mega/i2c.py
@@ -16,27 +16,33 @@ from collections import namedtuple

 # DeviceType = namedtuple('DeviceType', 'device_class,unit_of_measurement,suffix')

+
@dataclass
 class DeviceType:
    device_class: typing.Optional[str] = None
    unit_of_measurement: typing.Optional[str] = None
    suffix: typing.Optional[str] = None
    delay: typing.Optional[float] = None
+    i2c_par: typing.Optional[int] = None
+    idx: typing.Optional[
+        int
+    ] = None  # на случай если все значения представлены одной строчкой (как с scd4x)


 def parse_scan_page(page: str):
    ret = []
    req = []
    page = BeautifulSoup(page, features="lxml")
-    for x in page.find_all('a'):
-        params = x.get('href')
+    for x in page.find_all("a"):
+        params = x.get("href")
        if params is None:
            continue
        params = dict(parse_qsl(urlparse(params).query))
-        dev = params.get('i2c_dev')
+        dev = params.get("i2c_dev")
        if dev is None:
            continue
        classes = i2c_classes.get(dev, [])
+        i2c_par, idx = (None, None)
        for i, c in enumerate(classes):
            _params = params.copy()
            if c is Skip:
@@ -46,31 +52,39 @@ def parse_scan_page(page: str):
                continue
            elif isinstance(c, Request):
                if c.delay:
-                    _params['delay'] = c.delay
+                    _params["delay"] = c.delay
                req.append(_params)
                continue
            elif isinstance(c, DeviceType):
-                c, m, suffix, delay = astuple(c)
+                c, m, suffix, delay, i2c_par, idx = astuple(c)
                if delay is not None:
-                    _params['delay'] = delay
+                    _params["delay"] = delay
            else:
                continue
            suffix = suffix or c
-            if 'addr' in _params:
-                suffix += f"_{_params['addr']}" if suffix else str(_params['addr'])
+            if "addr" in _params:
+                suffix += f"_{_params['addr']}" if suffix else str(_params["addr"])
            if suffix:
-                _dev = f'{dev}_{suffix}'
+                _dev = f"{dev}_{suffix}"
            else:
                _dev = dev
-            if i > 0:
-                _params['i2c_par'] = i
+            if i > 0 and i2c_par is None:
+                _params["i2c_par"] = i
+            # i2c_par может быть явно указан в DeviceType
+            elif i2c_par is not None and i2c_par > 0:
+                _params["i2c_par"] = i2c_par
+            # idx - тема фактически только для scd4x, означает номер внутри текстового значения разделенного знаком "/"
+            if idx is not None:
+                _params["idx"] = idx

-            ret.append({
-                'id_suffix': _dev,
-                'device_class': c,
-                'params': _params,
-                'unit_of_measurement': m,
-            })
+            ret.append(
+                {
+                    "id_suffix": _dev,
+                    "device_class": c,
+                    "params": _params,
+                    "unit_of_measurement": m,
+                }
+            )
            req.append(_params)
    return req, ret

@@ -85,67 +99,85 @@ class Request:


 i2c_classes = {
-    'htu21d': [
+    "htu21d": [
        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None),
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
    ],
-    'sht31': [
-        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None, delay=1.5),
+    "sht31": [
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None, delay=0.1),
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
    ],
-    'max44009': [
-        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)
-    ],
-    'bh1750': [
-        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)
-    ],
-    'tsl2591': [
-        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)
-    ],
-    'bmp180': [
+    "max44009": [DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)],
+    "bh1750": [DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)],
+    "tsl2591": [DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)],
+    "bmp180": [
        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
    ],
-    'bmx280': [
+    "bmx280": [
        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
-        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None)
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None),
    ],
-    'dps368': [
+    "dps368": [
        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
    ],
-    'mlx90614': [
+    "mlx90614": [
        Skip,
-        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, 'temp'),
-        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, 'object'),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, "temp"),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, "object"),
    ],
-    'ptsensor': [
+    "ptsensor": [
        Skip,
        Request(delay=3),  # запрос на измерение
        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
    ],
-    'mcp9600': [
+    "mcp9600": [
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),  # термопара
-        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),  # сенсор встроенный в микросхему
+        DeviceType(
+            SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None
+        ),  # сенсор встроенный в микросхему
    ],
-    't67xx': [
-        DeviceType(SensorDeviceClass.CO2, CONCENTRATION_PARTS_PER_MILLION, None)
-    ],
-    'tmp117': [
+    "t67xx": [DeviceType(SensorDeviceClass.CO2, CONCENTRATION_PARTS_PER_MILLION, None)],
+    "tmp117": [
        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
    ],
-    'ads1115': [
-        DeviceType(None, None, 'ch0'),
-        DeviceType(None, None, 'ch1'),
-        DeviceType(None, None, 'ch2'),
-        DeviceType(None, None, 'ch3'),
+    "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'),
+    "ads1015": [
+        DeviceType(None, None, "ch0"),
+        DeviceType(None, None, "ch1"),
+        DeviceType(None, None, "ch2"),
+        DeviceType(None, None, "ch3"),
+    ],
+    "opt3001": [
+        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None),
+    ],
+    "scd4x": [
+        DeviceType(SensorDeviceClass.CO2, CONCENTRATION_PARTS_PER_MILLION, None),
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
+    ],
+    "ina226": [
+        Skip,
+        DeviceType(SensorDeviceClass.CURRENT, "A", None),
+        DeviceType(SensorDeviceClass.VOLTAGE, "V", None),
+    ],
+    "scd4x": [
+        DeviceType(
+            SensorDeviceClass.CO2,
+            CONCENTRATION_PARTS_PER_MILLION,
+            None,
+            i2c_par=0,
+            idx=0,
+        ),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None, i2c_par=0, idx=1),
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None, i2c_par=0, idx=2),
    ],
 }