mirror of
https://github.com/andvikt/mega_hacs.git
synced 2025-12-10 16:44:28 +05:00
121 lines
3.2 KiB
Python
121 lines
3.2 KiB
Python
from urllib.parse import urlparse, parse_qsl
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
page = '''
|
|
<html><head></head><body><a href="/sec/?pt=33">Back</a><br>0x15 - <a href="/sec/?pt=33&scl=32&i2c_dev=t67xx">T67XX</a><br>0x40 - <a href="/sec/?pt=33&scl=32&i2c_dev=htu21d">HTU21D</a>/PCA9685/HM3301<br>0x4a - <a href="/sec/?pt=33&scl=32&i2c_dev=max44009">MAX44009</a><br>
|
|
</body></html>
|
|
'''
|
|
|
|
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,
|
|
)
|
|
|
|
|
|
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))
|
|
if 'i2c_dev' in params:
|
|
dev = params['i2c_dev']
|
|
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, tuple):
|
|
suffix, c = c
|
|
elif isinstance(c, str):
|
|
suffix = c
|
|
else:
|
|
suffix = ''
|
|
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,
|
|
})
|
|
req.append(params)
|
|
return req, ret
|
|
|
|
|
|
class Skip:
|
|
pass
|
|
|
|
|
|
class Request:
|
|
pass
|
|
|
|
|
|
i2c_classes = {
|
|
'htu21d': [
|
|
DEVICE_CLASS_HUMIDITY,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
],
|
|
'sht31': [
|
|
DEVICE_CLASS_HUMIDITY,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
],
|
|
'max44009': [
|
|
DEVICE_CLASS_ILLUMINANCE
|
|
],
|
|
'bh1750': [
|
|
DEVICE_CLASS_ILLUMINANCE
|
|
],
|
|
'tsl2591': [
|
|
DEVICE_CLASS_ILLUMINANCE
|
|
],
|
|
'bmp180': [
|
|
DEVICE_CLASS_PRESSURE,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
],
|
|
'bmx280': [
|
|
DEVICE_CLASS_PRESSURE,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
DEVICE_CLASS_HUMIDITY
|
|
],
|
|
'mlx90614': [
|
|
Skip,
|
|
('temp', DEVICE_CLASS_TEMPERATURE),
|
|
('object', DEVICE_CLASS_TEMPERATURE),
|
|
],
|
|
'ptsensor': [
|
|
Request, # запрос на измерение
|
|
DEVICE_CLASS_PRESSURE,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
],
|
|
'mcp9600': [
|
|
DEVICE_CLASS_TEMPERATURE, # термопара
|
|
DEVICE_CLASS_TEMPERATURE, # сенсор встроенный в микросхему
|
|
],
|
|
't67xx': [
|
|
None # для co2 нет класса в HA
|
|
],
|
|
'tmp117': [
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
]
|
|
}
|
|
|
|
print(parse_scan_page(page)) |