mega_hacs/custom_components/mega/i2c.py

from dataclasses import dataclass, field
from urllib.parse import parse_qsl, urlparse
from bs4 import BeautifulSoup
from homeassistant.const import (
    DEVICE_CLASS_HUMIDITY,
    DEVICE_CLASS_TEMPERATURE,
    DEVICE_CLASS_ILLUMINANCE,
    DEVICE_CLASS_PRESSURE,
    PERCENTAGE,
    LIGHT_LUX,
    TEMP_CELSIUS,
    CONCENTRATION_PARTS_PER_MILLION,
    PRESSURE_BAR,
)
from collections import namedtuple

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


def parse_scan_page(page: str):
    ret = []
    req = []
    page = BeautifulSoup(page, features="lxml")
    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')
        if dev is None:
            continue
        classes = i2c_classes.get(dev, [])
        for i, c in enumerate(classes):
            if c is Skip:
                continue
            elif c is Request:
                req.append(params)
                continue
            elif isinstance(c, Request):
                if c.delay:
                    params = params.copy()
                    params['delay'] = c.delay
                req.append(params)
                continue
            elif isinstance(c, DeviceType):
                c, m, suffix = c
            else:
                continue
            suffix = suffix or c
            if 'addr' in params:
                suffix += f"_{params['addr']}" if suffix else str(params['addr'])
            if suffix:
                _dev = f'{dev}_{suffix}'
            else:
                _dev = dev
            params = params.copy()
            if i > 0:
                params['i2c_par'] = i

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


class Skip:
    pass


@dataclass
class Request:
    delay: float = None


i2c_classes = {
    'htu21d': [
        DeviceType(DEVICE_CLASS_HUMIDITY, PERCENTAGE, None),
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
    ],
    'sht31': [
        DeviceType(DEVICE_CLASS_HUMIDITY, PERCENTAGE, None),
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
    ],
    'max44009': [
        DeviceType(DEVICE_CLASS_ILLUMINANCE, LIGHT_LUX, None)
    ],
    'bh1750': [
        DeviceType(DEVICE_CLASS_ILLUMINANCE, LIGHT_LUX, None)
    ],
    'tsl2591': [
        DeviceType(DEVICE_CLASS_ILLUMINANCE, LIGHT_LUX, None)
    ],
    'bmp180': [
        DeviceType(DEVICE_CLASS_PRESSURE, PRESSURE_BAR, None),
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
    ],
    'bmx280': [
        DeviceType(DEVICE_CLASS_PRESSURE, PRESSURE_BAR, None),
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
        DeviceType(DEVICE_CLASS_HUMIDITY, PERCENTAGE, None)
    ],
    'mlx90614': [
        Skip,
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, 'temp'),
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, 'object'),
    ],
    'ptsensor': [
        Skip,
        Request(delay=1),  # запрос на измерение
        DeviceType(DEVICE_CLASS_PRESSURE, PRESSURE_BAR, None),
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
    ],
    'mcp9600': [
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),  # термопара
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),  # сенсор встроенный в микросхему
    ],
    't67xx': [
        DeviceType(None, CONCENTRATION_PARTS_PER_MILLION, None)  # для co2 нет класса в HA
    ],
    'tmp117': [
        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
    ]
}