Compare commits

...

29 Commits

Author SHA1 Message Date
Викторов Андрей Германович
adf5cfe2f8 Bump version: 1.1.8b5 → 1.1.8b6 2023-10-15 14:10:59 +03:00
Викторов Андрей Германович
6bb897a26f - fix init of conf_http 2023-10-15 14:01:39 +03:00
Викторов Андрей Германович
4edc0dcb4f - исправлена логика filter_value 2023-10-15 13:56:44 +03:00
Викторов Андрей Германович
89d261bd9c - исправление логики filter_scale 2023-10-15 13:53:11 +03:00
Викторов Андрей Германович
58288de1dd Bump version: 1.1.8b4 → 1.1.8b5 2023-10-15 13:37:31 +03:00
Викторов Андрей Германович
ec5aeab559 fix update extender 2023-10-15 13:37:24 +03:00
Викторов Андрей Германович
d0659b84dc Bump version: 1.1.8b3 → 1.1.8b4 2023-10-14 18:58:17 +03:00
Викторов Андрей Германович
3cd38a57e0 fix updater service 2023-10-14 18:57:47 +03:00
Викторов Андрей Германович
261c628909 Bump version: 1.1.8b2 → 1.1.8b3 2023-10-11 14:01:30 +03:00
Викторов Андрей Германович
2af9cfbeaf fix rgb brightness memory 2023-10-11 14:01:23 +03:00
Викторов Андрей Германович
60fad1f20f Bump version: 1.1.8b1 → 1.1.8b2 2023-10-11 11:26:57 +03:00
Викторов Андрей Германович
5c18e395da fix color order 2023-10-11 11:26:15 +03:00
Викторов Андрей Германович
082c647110 Bump version: 1.1.8b0 → 1.1.8b1 2023-10-10 23:01:09 +03:00
Викторов Андрей Германович
21bc277b78 fix rgb order 2023-10-10 22:58:25 +03:00
Викторов Андрей Германович
00e62ee83f Bump version: 1.1.7 → 1.1.8b0 2023-10-10 21:01:59 +03:00
Викторов Андрей Германович
b62406210c fix rgb color 2023-10-10 21:01:21 +03:00
Викторов Андрей Германович
c45653b0ba Bump version: 1.1.7b0 → 1.1.7 2023-10-09 14:58:26 +03:00
Викторов Андрей Германович
683727b3ec Bump version: 1.1.6 → 1.1.7b0 2023-10-09 14:57:50 +03:00
Викторов Андрей Германович
2300d2750a . 2023-10-09 14:57:46 +03:00
Викторов Андрей Германович
ac8603f219 fix encoding issue 2023-10-09 14:55:15 +03:00
Викторов Андрей Германович
fbc1f31023 Bump version: 1.1.6b1 → 1.1.6 2023-08-21 08:55:54 +03:00
Викторов Андрей Германович
cd5ab3b689 Bump version: 1.1.6b0 → 1.1.6b1 2023-08-14 20:44:24 +03:00
Викторов Андрей Германович
0138f5e323 . 2023-08-14 20:43:26 +03:00
Викторов Андрей Германович
0ab4f6623c Bump version: 1.1.5 → 1.1.6b0 2023-08-14 20:38:08 +03:00
Викторов Андрей Германович
5664e1e929 . 2023-08-14 20:38:06 +03:00
Викторов Андрей Германович
bcf108fc9c . 2023-08-14 20:37:35 +03:00
Викторов Андрей Германович
ca3f90374b Bump version: 1.1.501 → 1.1.502b0 2023-08-14 20:37:08 +03:00
Викторов Андрей Германович
d15cce8061 Bump version: 1.1.5 → 1.1.501 2023-08-14 20:36:40 +03:00
Викторов Андрей Германович
4e83d81004 fix #152 2023-08-14 20:35:59 +03:00
8 changed files with 607 additions and 446 deletions

View File

@@ -1,5 +1,5 @@
[bumpversion] [bumpversion]
current_version = 1.1.5 current_version = 1.1.8b6
parse = (?P<major>\d+)(\.(?P<minor>\d+))(\.(?P<patch>\d+))(?P<release>[bf]*)(?P<build>\d*) parse = (?P<major>\d+)(\.(?P<minor>\d+))(\.(?P<patch>\d+))(?P<release>[bf]*)(?P<build>\d*)
commit = True commit = True
tag = True tag = True

18
.gitignore vendored Normal file
View File

@@ -0,0 +1,18 @@
*sync-conflict*
*.db
*.log*
*.yaml
**/__pycache__
test_*
temp_*
tr.py
experiment_*
example*
.HA_VERSION
.idea
.storage
site
.DS_Store
.venv
pyproject.toml
poetry.lock

View File

@@ -1,25 +1,57 @@
"""The mega integration.""" """The mega integration."""
import asyncio import asyncio
import logging import logging
import typing
from functools import partial from functools import partial
import voluptuous as vol import voluptuous as vol
from homeassistant.const import ( from homeassistant.const import (
CONF_NAME, CONF_DOMAIN, CONF_NAME,
CONF_UNIT_OF_MEASUREMENT, CONF_VALUE_TEMPLATE, CONF_DEVICE_CLASS, CONF_PORT CONF_DOMAIN,
CONF_UNIT_OF_MEASUREMENT,
CONF_VALUE_TEMPLATE,
CONF_DEVICE_CLASS,
CONF_PORT,
) )
from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers.service import bind_hass from homeassistant.helpers.service import bind_hass
from homeassistant.helpers import config_validation as cv from homeassistant.helpers import config_validation as cv
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from .const import DOMAIN, CONF_INVERT, CONF_RELOAD, PLATFORMS, CONF_PORTS, CONF_CUSTOM, CONF_SKIP, CONF_PORT_TO_SCAN, \ from .const import (
CONF_MQTT_INPUTS, CONF_HTTP, CONF_RESPONSE_TEMPLATE, CONF_ACTION, CONF_GET_VALUE, CONF_ALLOW_HOSTS, \ DOMAIN,
CONF_CONV_TEMPLATE, CONF_ALL, CONF_FORCE_D, CONF_DEF_RESPONSE, CONF_FORCE_I2C_SCAN, CONF_HEX_TO_FLOAT, \ CONF_INVERT,
RGB_COMBINATIONS, CONF_WS28XX, CONF_ORDER, CONF_SMOOTH, CONF_LED, CONF_WHITE_SEP, CONF_CHIP, CONF_RANGE, \ PLATFORMS,
CONF_FILTER_VALUES, CONF_FILTER_SCALE, CONF_FILTER_LOW, CONF_FILTER_HIGH, CONF_FILL_NA, CONF_MEGA_ID, CONF_ADDR, \ CONF_PORTS,
CONF_1WBUS CONF_CUSTOM,
CONF_SKIP,
CONF_HTTP,
CONF_RESPONSE_TEMPLATE,
CONF_ACTION,
CONF_GET_VALUE,
CONF_ALLOW_HOSTS,
CONF_CONV_TEMPLATE,
CONF_ALL,
CONF_FORCE_D,
CONF_DEF_RESPONSE,
CONF_FORCE_I2C_SCAN,
CONF_HEX_TO_FLOAT,
RGB_COMBINATIONS,
CONF_WS28XX,
CONF_ORDER,
CONF_SMOOTH,
CONF_LED,
CONF_WHITE_SEP,
CONF_CHIP,
CONF_RANGE,
CONF_FILTER_VALUES,
CONF_FILTER_SCALE,
CONF_FILTER_LOW,
CONF_FILTER_HIGH,
CONF_FILL_NA,
CONF_MEGA_ID,
CONF_ADDR,
CONF_1WBUS,
)
from .hub import MegaD from .hub import MegaD
from .config_flow import ConfigFlow from .config_flow import ConfigFlow
from .http import MegaView from .http import MegaView
@@ -28,58 +60,53 @@ _LOGGER = logging.getLogger(__name__)
_port_n = vol.Any(int, str) _port_n = vol.Any(int, str)
LED_LIGHT = \ LED_LIGHT = {
{ str: vol.Any(
str: vol.Any( {
{ vol.Required(CONF_PORTS): vol.Any(
vol.Required(CONF_PORTS): vol.Any( vol.ExactSequence([_port_n, _port_n, _port_n]),
vol.ExactSequence([_port_n, _port_n, _port_n]), vol.ExactSequence([_port_n, _port_n, _port_n, _port_n]),
vol.ExactSequence([_port_n, _port_n, _port_n, _port_n]), msg="ports must be [R, G, B] or [R, G, B, W] of integers 0..255",
msg='ports must be [R, G, B] or [R, G, B, W] of integers 0..255' ),
), vol.Optional(CONF_NAME): str,
vol.Optional(CONF_NAME): str, vol.Optional(CONF_WHITE_SEP, default=True): bool,
vol.Optional(CONF_WHITE_SEP, default=True): bool, vol.Optional(CONF_SMOOTH, default=1): cv.time_period_seconds,
vol.Optional(CONF_SMOOTH, default=1): cv.time_period_seconds, },
}, {
{ vol.Required(CONF_PORT): int,
vol.Required(CONF_PORT): int, vol.Required(CONF_WS28XX): True,
vol.Required(CONF_WS28XX): True, vol.Optional(CONF_CHIP, default=100): int,
vol.Optional(CONF_CHIP, default=100): int, vol.Optional(CONF_ORDER, default="rgb"): vol.Any(
vol.Optional(CONF_ORDER, default='rgb'): vol.Any(*RGB_COMBINATIONS, msg=f'order must be one of {RGB_COMBINATIONS}'), *RGB_COMBINATIONS, msg=f"order must be one of {RGB_COMBINATIONS}"
vol.Optional(CONF_SMOOTH, default=1): cv.time_period_seconds, ),
vol.Optional(CONF_NAME): str, vol.Optional(CONF_SMOOTH, default=1): cv.time_period_seconds,
}, vol.Optional(CONF_NAME): str,
) },
} )
}
CUSTOMIZE_PORT = { CUSTOMIZE_PORT = {
vol.Optional(CONF_SKIP, description='исключить порт из сканирования', default=False): bool, vol.Optional(
vol.Optional(CONF_FILL_NA, default='last'): vol.Any( CONF_SKIP, description="исключить порт из сканирования", default=False
'last', ): bool,
'none' vol.Optional(CONF_FILL_NA, default="last"): vol.Any("last", "none"),
), vol.Optional(CONF_RANGE, description="диапазон диммирования"): [
vol.Optional(CONF_RANGE, description='диапазон диммирования'): [
vol.Range(0, 255), vol.Range(0, 255),
vol.Range(0, 255), vol.Range(0, 255),
], ],
vol.Optional(CONF_INVERT, default=False): bool, vol.Optional(CONF_INVERT, default=False): bool,
vol.Optional(CONF_NAME): vol.Any(str, { vol.Optional(CONF_NAME): vol.Any(str, {vol.Required(str): str}),
vol.Required(str): str vol.Optional(CONF_DOMAIN): vol.Any("light", "switch"),
}), vol.Optional(
vol.Optional(CONF_DOMAIN): vol.Any('light', 'switch'), CONF_UNIT_OF_MEASUREMENT,
vol.Optional(CONF_UNIT_OF_MEASUREMENT, description='единицы измерений, либо строка либо мепинг'): description="единицы измерений, либо строка либо мепинг",
vol.Any(str, { ): vol.Any(str, {vol.Required(str): str}),
vol.Required(str): str vol.Optional(CONF_DEVICE_CLASS): vol.Any(str, {vol.Required(str): str}),
}),
vol.Optional(CONF_DEVICE_CLASS):
vol.Any(str, {
vol.Required(str): str
}),
vol.Optional( vol.Optional(
CONF_RESPONSE_TEMPLATE, CONF_RESPONSE_TEMPLATE,
description='шаблон ответа когда на этот порт приходит' description="шаблон ответа когда на этот порт приходит" "сообщение из меги ",
'сообщение из меги '): cv.template, ): cv.template,
vol.Optional(CONF_ACTION): cv.script_action, # пока не реализовано vol.Optional(CONF_ACTION): cv.script_action, # пока не реализовано
vol.Optional(CONF_GET_VALUE, default=True): bool, vol.Optional(CONF_GET_VALUE, default=True): bool,
vol.Optional(CONF_CONV_TEMPLATE): cv.template, vol.Optional(CONF_CONV_TEMPLATE): cv.template,
vol.Optional(CONF_VALUE_TEMPLATE): cv.template, vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
@@ -96,43 +123,48 @@ CUSTOMIZE_PORT = {
vol.Optional(CONF_DEVICE_CLASS): str, vol.Optional(CONF_DEVICE_CLASS): str,
vol.Optional(CONF_UNIT_OF_MEASUREMENT): str, vol.Optional(CONF_UNIT_OF_MEASUREMENT): str,
vol.Optional(CONF_VALUE_TEMPLATE): cv.template, vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
} },
} }
CUSTOMIZE_DS2413 = { CUSTOMIZE_DS2413 = {
vol.Optional(str.lower, description='адрес и индекс устройства'): CUSTOMIZE_PORT vol.Optional(str.lower, description="адрес и индекс устройства"): CUSTOMIZE_PORT
} }
def extender(x): def extender(x):
if isinstance(x, str) and 'e' in x: if isinstance(x, str) and "e" in x:
return x return x
else: else:
raise ValueError('must has "e" in port name') raise ValueError('must has "e" in port name')
OWBUS = vol.Schema({
vol.Required(CONF_PORT): vol.Any(vol.Coerce(int), vol.Coerce(str)), OWBUS = vol.Schema(
vol.Required(CONF_MEGA_ID): vol.Coerce(str), {
vol.Required(CONF_ADDR): [str], vol.Required(CONF_PORT): vol.Any(vol.Coerce(int), vol.Coerce(str)),
}) vol.Required(CONF_MEGA_ID): vol.Coerce(str),
vol.Required(CONF_ADDR): [str],
}
)
CONFIG_SCHEMA = vol.Schema( CONFIG_SCHEMA = vol.Schema(
{ {
DOMAIN: { DOMAIN: {
vol.Optional(CONF_ALLOW_HOSTS): [str], vol.Optional(CONF_ALLOW_HOSTS): [str],
vol.Optional('entities'): { vol.Optional("entities"): {
vol.Optional(str): vol.Any( vol.Optional(str): vol.Any(CUSTOMIZE_PORT, CUSTOMIZE_DS2413)
CUSTOMIZE_PORT, },
CUSTOMIZE_DS2413 vol.Optional(vol.Any(str, int), description="id меги из веб-интерфейса"): {
)},
vol.Optional(vol.Any(str, int), description='id меги из веб-интерфейса'): {
vol.Optional(CONF_FORCE_D, description='Принудительно слать d после срабатывания входа', default=False): bool,
vol.Optional( vol.Optional(
CONF_DEF_RESPONSE, CONF_FORCE_D,
description='Ответ по умолчанию', description="Принудительно слать d после срабатывания входа",
default=None default=False,
): bool,
vol.Optional(
CONF_DEF_RESPONSE, description="Ответ по умолчанию", default=None
): vol.Any(cv.template, None), ): vol.Any(cv.template, None),
vol.Optional(CONF_LED): LED_LIGHT, vol.Optional(CONF_LED): LED_LIGHT,
vol.Optional(vol.Any(int, extender), description='номер порта'): vol.Any( vol.Optional(
vol.Any(int, extender), description="номер порта"
): vol.Any(
CUSTOMIZE_PORT, CUSTOMIZE_PORT,
CUSTOMIZE_DS2413, CUSTOMIZE_DS2413,
), ),
@@ -141,14 +173,14 @@ CONFIG_SCHEMA = vol.Schema(
vol.Optional(CONF_FILTER_LOW): vol.Coerce(float), vol.Optional(CONF_FILTER_LOW): vol.Coerce(float),
vol.Optional(CONF_FILTER_HIGH): vol.Coerce(float), vol.Optional(CONF_FILTER_HIGH): vol.Coerce(float),
}, },
vol.Optional(CONF_1WBUS): [OWBUS] vol.Optional(CONF_1WBUS): [OWBUS],
} }
}, },
extra=vol.ALLOW_EXTRA, extra=vol.ALLOW_EXTRA,
) )
ALIVE_STATE = 'alive' ALIVE_STATE = "alive"
DEF_ID = 'def' DEF_ID = "def"
_POLL_TASKS = {} _POLL_TASKS = {}
_hubs = {} _hubs = {}
_subs = {} _subs = {}
@@ -162,29 +194,40 @@ async def async_setup(hass: HomeAssistant, config: dict):
view.allowed_hosts |= set(config.get(DOMAIN, {}).get(CONF_ALLOW_HOSTS, [])) view.allowed_hosts |= set(config.get(DOMAIN, {}).get(CONF_ALLOW_HOSTS, []))
hass.http.register_view(view) hass.http.register_view(view)
hass.services.async_register( hass.services.async_register(
DOMAIN, 'save', partial(_save_service, hass), schema=vol.Schema({ DOMAIN,
vol.Optional('mega_id'): str "save",
}) partial(_save_service, hass),
schema=vol.Schema({vol.Optional("mega_id"): str}),
) )
hass.services.async_register( hass.services.async_register(
DOMAIN, 'get_port', partial(_get_port, hass), schema=vol.Schema({ DOMAIN,
vol.Optional('mega_id'): str, "get_port",
vol.Optional('port'): vol.Any(int, [int]), partial(_get_port, hass),
}) schema=vol.Schema(
{
vol.Optional("mega_id"): str,
vol.Optional("port"): vol.Any(int, [int]),
}
),
) )
hass.services.async_register( hass.services.async_register(
DOMAIN, 'run_cmd', partial(_run_cmd, hass), schema=vol.Schema({ DOMAIN,
vol.Optional('port'): int, "run_cmd",
vol.Required('cmd'): str, partial(_run_cmd, hass),
vol.Optional('mega_id'): str, schema=vol.Schema(
}) {
vol.Optional("port"): int,
vol.Required("cmd"): str,
vol.Optional("mega_id"): str,
}
),
) )
return True return True
async def get_hub(hass, entry): async def get_hub(hass, entry):
id = entry.data.get('id', entry.entry_id) id = entry.data.get("id", entry.entry_id)
data = dict(entry.data) data = dict(entry.data)
data.update(entry.options or {}) data.update(entry.options or {})
data.update(id=id) data.update(id=id)
@@ -194,7 +237,7 @@ async def get_hub(hass, entry):
async def _add_mega(hass: HomeAssistant, entry: ConfigEntry): async def _add_mega(hass: HomeAssistant, entry: ConfigEntry):
id = entry.data.get('id', entry.entry_id) id = entry.data.get("id", entry.entry_id)
hub = await get_hub(hass, entry) hub = await get_hub(hass, entry)
hub.fw = await hub.get_fw() hub.fw = await hub.get_fw()
hass.data[DOMAIN][id] = hub hass.data[DOMAIN][id] = hub
@@ -213,9 +256,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
await hub.start() await hub.start()
for platform in PLATFORMS: for platform in PLATFORMS:
hass.async_create_task( hass.async_create_task(
hass.config_entries.async_forward_entry_setup( hass.config_entries.async_forward_entry_setup(entry, platform)
entry, platform
)
) )
await hub.updater.async_refresh() await hub.updater.async_refresh()
return True return True
@@ -237,11 +278,11 @@ async def updater(hass: HomeAssistant, entry: ConfigEntry):
async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None: async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Handle removal of an entry.""" """Handle removal of an entry."""
id = entry.data.get('id', entry.entry_id) id = entry.data.get("id", entry.entry_id)
hub: MegaD = hass.data[DOMAIN].get(id) hub: MegaD = hass.data[DOMAIN].get(id)
if hub is None: if hub is None:
return True return True
_LOGGER.debug(f'remove {id}') _LOGGER.debug(f"remove {id}")
_hubs.pop(id, None) _hubs.pop(id, None)
hass.data[DOMAIN].pop(id, None) hass.data[DOMAIN].pop(id, None)
hass.data[DOMAIN][CONF_ALL].pop(id, None) hass.data[DOMAIN][CONF_ALL].pop(id, None)
@@ -255,19 +296,24 @@ async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
await hub.stop() await hub.stop()
return True return True
async_unload_entry = async_remove_entry async_unload_entry = async_remove_entry
async def async_migrate_entry(hass, config_entry: ConfigEntry): async def async_migrate_entry(hass, config_entry: ConfigEntry):
"""Migrate old entry.""" """Migrate old entry."""
_LOGGER.debug("Migrating from version %s to version %s", config_entry.version, ConfigFlow.VERSION) _LOGGER.debug(
"Migrating from version %s to version %s",
config_entry.version,
ConfigFlow.VERSION,
)
hub = await get_hub(hass, config_entry) hub = await get_hub(hass, config_entry)
new = dict(config_entry.data) new = dict(config_entry.data)
await hub.start() await hub.start()
cfg = await hub.get_config() cfg = await hub.get_config()
await hub.stop() await hub.stop()
new.update(cfg) new.update(cfg)
_LOGGER.debug(f'new config: %s', new) _LOGGER.debug(f"new config: %s", new)
config_entry.data = new config_entry.data = new
config_entry.version = ConfigFlow.VERSION config_entry.version = ConfigFlow.VERSION
@@ -277,7 +323,7 @@ async def async_migrate_entry(hass, config_entry: ConfigEntry):
async def _save_service(hass: HomeAssistant, call: ServiceCall): async def _save_service(hass: HomeAssistant, call: ServiceCall):
mega_id = call.data.get('mega_id') mega_id = call.data.get("mega_id")
if mega_id: if mega_id:
hub: MegaD = hass.data[DOMAIN][mega_id] hub: MegaD = hass.data[DOMAIN][mega_id]
await hub.save() await hub.save()
@@ -289,8 +335,8 @@ async def _save_service(hass: HomeAssistant, call: ServiceCall):
@bind_hass @bind_hass
async def _get_port(hass: HomeAssistant, call: ServiceCall): async def _get_port(hass: HomeAssistant, call: ServiceCall):
port = call.data.get('port') port = call.data.get("port")
mega_id = call.data.get('mega_id') mega_id = call.data.get("mega_id")
if mega_id: if mega_id:
hub: MegaD = hass.data[DOMAIN][mega_id] hub: MegaD = hass.data[DOMAIN][mega_id]
if port is None: if port is None:
@@ -300,6 +346,7 @@ async def _get_port(hass: HomeAssistant, call: ServiceCall):
elif isinstance(port, list): elif isinstance(port, list):
for x in port: for x in port:
await hub.get_port(x) await hub.get_port(x)
hub.updater.async_set_updated_data(hub.values)
else: else:
for hub in hass.data[DOMAIN][CONF_ALL].values(): for hub in hass.data[DOMAIN][CONF_ALL].values():
if not isinstance(hub, MegaD): if not isinstance(hub, MegaD):
@@ -311,12 +358,13 @@ async def _get_port(hass: HomeAssistant, call: ServiceCall):
elif isinstance(port, list): elif isinstance(port, list):
for x in port: for x in port:
await hub.get_port(x) await hub.get_port(x)
hub.updater.async_set_updated_data(hub.values)
@bind_hass @bind_hass
async def _run_cmd(hass: HomeAssistant, call: ServiceCall): async def _run_cmd(hass: HomeAssistant, call: ServiceCall):
mega_id = call.data.get('mega_id') mega_id = call.data.get("mega_id")
cmd = call.data.get('cmd') cmd = call.data.get("cmd")
if mega_id: if mega_id:
hub: MegaD = hass.data[DOMAIN][mega_id] hub: MegaD = hass.data[DOMAIN][mega_id]
await hub.request(cmd=cmd) await hub.request(cmd=cmd)

View File

@@ -155,7 +155,7 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
return DeviceInfo( return DeviceInfo(
identifiers={ identifiers={
# Serial numbers are unique identifiers within a specific domain # Serial numbers are unique identifiers within a specific domain
(DOMAIN, f'{self._mega_id}'), #pt_idx (DOMAIN, f'{self._mega_id}', pt_idx)
}, },
name=self.name, name=self.name,
manufacturer='ab-log.ru', manufacturer='ab-log.ru',

View File

@@ -12,48 +12,47 @@ from bs4 import BeautifulSoup
from homeassistant.components.sensor import SensorDeviceClass from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( TEMP_CELSIUS, PERCENTAGE, LIGHT_LUX from homeassistant.const import TEMP_CELSIUS, PERCENTAGE, LIGHT_LUX
)
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .config_parser import parse_config, DS2413, MCP230, MCP230_OUT, MCP230_IN, PCA9685 from .config_parser import parse_config, DS2413, MCP230, MCP230_OUT, MCP230_IN, PCA9685
from .const import ( from .const import (
TEMP, HUM, PRESS, TEMP,
LUX, PATT_SPLIT, DOMAIN, HUM,
CONF_HTTP, EVENT_BINARY_SENSOR, CONF_CUSTOM, CONF_FORCE_D, CONF_DEF_RESPONSE, PATT_FW, CONF_FORCE_I2C_SCAN, PRESS,
REMOVE_CONFIG LUX,
PATT_SPLIT,
DOMAIN,
CONF_HTTP,
EVENT_BINARY_SENSOR,
CONF_CUSTOM,
CONF_FORCE_D,
CONF_DEF_RESPONSE,
PATT_FW,
CONF_FORCE_I2C_SCAN,
REMOVE_CONFIG,
) )
from .entities import set_events_off, BaseMegaEntity, MegaOutPort, safe_int from .entities import set_events_off, BaseMegaEntity, MegaOutPort, safe_int
from .exceptions import CannotConnect, NoPort from .exceptions import CannotConnect, NoPort
from .i2c import parse_scan_page from .i2c import parse_scan_page
from .tools import make_ints, int_ignore, PriorityLock from .tools import make_ints, int_ignore, PriorityLock
TEMP_PATT = re.compile(r'temp:([01234567890\.]+)') TEMP_PATT = re.compile(r"temp:([01234567890\.]+)")
HUM_PATT = re.compile(r'hum:([01234567890\.]+)') HUM_PATT = re.compile(r"hum:([01234567890\.]+)")
PRESS_PATT = re.compile(r'press:([01234567890\.]+)') PRESS_PATT = re.compile(r"press:([01234567890\.]+)")
LUX_PATT = re.compile(r'lux:([01234567890\.]+)') LUX_PATT = re.compile(r"lux:([01234567890\.]+)")
PATTERNS = { PATTERNS = {TEMP: TEMP_PATT, HUM: HUM_PATT, PRESS: PRESS_PATT, LUX: LUX_PATT}
TEMP: TEMP_PATT, UNITS = {TEMP: TEMP_CELSIUS, HUM: PERCENTAGE, PRESS: "mmHg", LUX: LIGHT_LUX}
HUM: HUM_PATT,
PRESS: PRESS_PATT,
LUX: LUX_PATT
}
UNITS = {
TEMP: TEMP_CELSIUS,
HUM: PERCENTAGE,
PRESS: 'mmHg',
LUX: LIGHT_LUX
}
CLASSES = { CLASSES = {
TEMP: SensorDeviceClass.TEMPERATURE, TEMP: SensorDeviceClass.TEMPERATURE,
HUM: SensorDeviceClass.HUMIDITY, HUM: SensorDeviceClass.HUMIDITY,
PRESS: SensorDeviceClass.PRESSURE, PRESS: SensorDeviceClass.PRESSURE,
LUX: SensorDeviceClass.ILLUMINANCE LUX: SensorDeviceClass.ILLUMINANCE,
} }
I2C_DEVICE_TYPES = { I2C_DEVICE_TYPES = {
"2": LUX, # BH1750 "2": LUX, # BH1750
"3": LUX, # TSL2591 "3": LUX, # TSL2591
"7": LUX, # MAX44009 "7": LUX, # MAX44009
"70": LUX, # OPT3001 "70": LUX, # OPT3001
} }
@@ -62,45 +61,45 @@ class MegaD:
"""MegaD Hub""" """MegaD Hub"""
def __init__( def __init__(
self, self,
hass: HomeAssistant, hass: HomeAssistant,
loop: asyncio.AbstractEventLoop, loop: asyncio.AbstractEventLoop,
host: str, host: str,
password: str, password: str,
lg: logging.Logger, lg: logging.Logger,
id: str, id: str,
config: ConfigEntry = None, config: ConfigEntry = None,
mqtt_id: str = None, mqtt_id: str = None,
scan_interval=60, scan_interval=60,
port_to_scan=0, port_to_scan=0,
nports=38, nports=38,
update_all: bool=True, update_all: bool = True,
poll_outs: bool=False, poll_outs: bool = False,
fake_response: bool=True, fake_response: bool = True,
force_d: bool=None, force_d: bool = None,
allow_hosts: str=None, allow_hosts: str = None,
protected=True, protected=True,
restore_on_restart=False, restore_on_restart=False,
extenders=None, extenders=None,
ext_in=None, ext_in=None,
ext_acts=None, ext_acts=None,
i2c_sensors=None, i2c_sensors=None,
new_naming=False, new_naming=False,
update_time=False, update_time=False,
smooth: list=None, smooth: list = None,
**kwargs, **kwargs,
): ):
"""Initialize.""" """Initialize."""
self.skip_ports = set() self.skip_ports = set()
if config is not None: if config is not None:
lg.debug(f'load config: %s', config.data) lg.debug(f"load config: %s", config.data)
self.config = config self.config = config
self.http = hass.data.get(DOMAIN, {}).get(CONF_HTTP) self.http = hass.data.get(DOMAIN, {}).get(CONF_HTTP)
if not self.http is None: if not self.http is None:
self.http.allowed_hosts |= {host} self.http.allowed_hosts |= {host}
self.http.hubs[host] = self self.http.hubs[host] = self
if len(self.http.hubs) == 1: if len(self.http.hubs) == 1:
self.http.hubs['__def'] = self self.http.hubs["__def"] = self
if mqtt_id: if mqtt_id:
self.http.hubs[mqtt_id] = self self.http.hubs[mqtt_id] = self
self.smooth = smooth or [] self.smooth = smooth or []
@@ -134,7 +133,9 @@ class MegaD:
self.sensors = [] self.sensors = []
self.port_to_scan = port_to_scan self.port_to_scan = port_to_scan
self.last_update = datetime.now() self.last_update = datetime.now()
self._callbacks: typing.DefaultDict[int, typing.List[typing.Callable[[dict], typing.Coroutine]]] = defaultdict(list) self._callbacks: typing.DefaultDict[
int, typing.List[typing.Callable[[dict], typing.Coroutine]]
] = defaultdict(list)
self._loop = loop self._loop = loop
self._customize = None self._customize = None
self.values = {} self.values = {}
@@ -144,10 +145,12 @@ class MegaD:
self.lg, self.lg,
name="megad", name="megad",
update_method=self.poll, update_method=self.poll,
update_interval=timedelta(seconds=self.poll_interval) if self.poll_interval else None, update_interval=timedelta(seconds=self.poll_interval)
if self.poll_interval
else None,
) )
self.updaters = [] self.updaters = []
self.fw = '' self.fw = ""
self.notifiers = defaultdict(asyncio.Condition) self.notifiers = defaultdict(asyncio.Condition)
if not mqtt_id: if not mqtt_id:
_id = host.split(".")[-1] _id = host.split(".")[-1]
@@ -159,11 +162,11 @@ class MegaD:
self.customize[CONF_FORCE_D] = force_d self.customize[CONF_FORCE_D] = force_d
try: try:
if allow_hosts is not None and DOMAIN in hass.data: if allow_hosts is not None and DOMAIN in hass.data:
allow_hosts = set(allow_hosts.split(';')) allow_hosts = set(allow_hosts.split(";"))
hass.data[DOMAIN][CONF_HTTP].allowed_hosts |= allow_hosts hass.data[DOMAIN][CONF_HTTP].allowed_hosts |= allow_hosts
hass.data[DOMAIN][CONF_HTTP].protected = protected hass.data[DOMAIN][CONF_HTTP].protected = protected
except Exception: except Exception:
self.lg.exception('while setting allowed hosts') self.lg.exception("while setting allowed hosts")
self.binary_sensors = [] self.binary_sensors = []
async def start(self): async def start(self):
@@ -183,7 +186,7 @@ class MegaD:
self.lg.debug(self.sensors) self.lg.debug(self.sensors)
ports = [] ports = []
for x in self.sensors: for x in self.sensors:
if only_list and x.http_cmd != 'list': if only_list and x.http_cmd != "list":
continue continue
if x.port in ports: if x.port in ports:
continue continue
@@ -211,22 +214,24 @@ class MegaD:
@property @property
def is_online(self): def is_online(self):
return (datetime.now() - self.last_update).total_seconds() < (self.poll_interval + 10) return (datetime.now() - self.last_update).total_seconds() < (
self.poll_interval + 10
)
def _warn_offline(self): def _warn_offline(self):
if self.online: if self.online:
self.lg.warning('mega is offline') self.lg.warning("mega is offline")
self.hass.states.async_set( self.hass.states.async_set(
f'mega.{self.id}', f"mega.{self.id}",
'offline', "offline",
) )
self.online = False self.online = False
def _notify_online(self): def _notify_online(self):
if not self.online: if not self.online:
self.hass.states.async_set( self.hass.states.async_set(
f'mega.{self.id}', f"mega.{self.id}",
'online', "online",
) )
self.online = True self.online = True
@@ -236,13 +241,10 @@ class MegaD:
:return: :return:
""" """
for x in self.ds2413_ports: for x in self.ds2413_ports:
self.lg.debug(f'poll ds2413 for %s', x) self.lg.debug(f"poll ds2413 for %s", x)
try: try:
await self.get_port( await self.get_port(
port=x, port=x, force_http=True, http_cmd="list", conv=False
force_http=True,
http_cmd='list',
conv=False
) )
except asyncio.TimeoutError: except asyncio.TimeoutError:
continue continue
@@ -251,7 +253,7 @@ class MegaD:
""" """
Polling ports Polling ports
""" """
self.lg.debug('poll') self.lg.debug("poll")
if self._update_time: if self._update_time:
await self.update_time() await self.update_time()
for x in self.i2c_sensors: for x in self.i2c_sensors:
@@ -264,7 +266,7 @@ class MegaD:
for x in self.extenders: for x in self.extenders:
ret = await self._update_extender(x) ret = await self._update_extender(x)
if not isinstance(ret, dict): if not isinstance(ret, dict):
self.lg.warning(f'wrong updater result: {ret} from extender {x}') self.lg.warning(f"wrong updater result: {ret} from extender {x}")
continue continue
self.values.update(ret) self.values.update(ret)
@@ -275,14 +277,14 @@ class MegaD:
async def get_mqtt_id(self): async def get_mqtt_id(self):
async with aiohttp.request( async with aiohttp.request(
'get', f'http://{self.host}/{self.sec}/?cf=2' "get", f"http://{self.host}/{self.sec}/?cf=2"
) as req: ) as req:
data = await req.text() data = await req.text(encoding="iso-8859-5")
data = BeautifulSoup(data, features="lxml") data = BeautifulSoup(data, features="lxml")
_id = data.find(attrs={'name': 'mdid'}) _id = data.find(attrs={"name": "mdid"})
if _id: if _id:
_id = _id['value'] _id = _id["value"]
return _id or 'megad/' + self.host.split('.')[-1] return _id or "megad/" + self.host.split(".")[-1]
async def get_fw(self): async def get_fw(self):
data = await self.request() data = await self.request()
@@ -292,84 +294,90 @@ class MegaD:
return await self.request(pt=port, cmd=cmd) return await self.request(pt=port, cmd=cmd)
async def request(self, priority=0, **kwargs): async def request(self, priority=0, **kwargs):
cmd = '&'.join([f'{k}={v}' for k, v in kwargs.items() if v is not None]) cmd = "&".join([f"{k}={v}" for k, v in kwargs.items() if v is not None])
url = f"http://{self.host}/{self.sec}" url = f"http://{self.host}/{self.sec}"
if cmd: if cmd:
url = f"{url}/?{cmd}" url = f"{url}/?{cmd}"
self.lg.debug('request: %s', url) self.lg.debug("request: %s", url)
async with self._http_lck(priority): async with self._http_lck(priority):
for _ntry in range(3): for _ntry in range(3):
try: try:
async with aiohttp.request("get", url=url, timeout=aiohttp.ClientTimeout(total=5)) as req: async with aiohttp.request(
"get", url=url, timeout=aiohttp.ClientTimeout(total=5)
) as req:
if req.status != 200: if req.status != 200:
self.lg.warning('%s returned %s (%s)', url, req.status, await req.text()) self.lg.warning(
"%s returned %s (%s)",
url,
req.status,
await req.text(encoding="iso-8859-5"),
)
return None return None
else: else:
ret = await req.text() ret = await req.text(encoding="iso-8859-5")
self.lg.debug('response %s', ret) self.lg.debug("response %s", ret)
return ret return ret
except asyncio.TimeoutError: except asyncio.TimeoutError:
self.lg.warning(f'timeout while requesting {url}') self.lg.warning(f"timeout while requesting {url}")
# raise # raise
await asyncio.sleep(1) await asyncio.sleep(1)
raise asyncio.TimeoutError('after 3 tries') raise asyncio.TimeoutError("after 3 tries")
async def save(self): async def save(self):
await self.send_command(cmd='s') await self.send_command(cmd="s")
def parse_response(self, ret, cmd='get'): def parse_response(self, ret, cmd="get"):
if ret is None: if ret is None:
raise NoPort() raise NoPort()
if 'busy' in ret: if "busy" in ret:
return None return None
if ':' in ret: if ":" in ret:
if ';' in ret: if ";" in ret:
ret = ret.split(';') ret = ret.split(";")
elif '/' in ret and not cmd == 'list': elif "/" in ret and not cmd == "list":
ret = ret.split('/') ret = ret.split("/")
else: else:
ret = [ret] ret = [ret]
ret = {'value': dict([ ret = {"value": dict([x.split(":") for x in ret if x.count(":") == 1])}
x.split(':') for x in ret if x.count(':') == 1 elif "ON" in ret:
])} ret = {"value": "ON"}
elif 'ON' in ret: elif "OFF" in ret:
ret = {'value': 'ON'} ret = {"value": "OFF"}
elif 'OFF' in ret:
ret = {'value': 'OFF'}
else: else:
ret = {'value': ret} ret = {"value": ret}
return ret return ret
async def get_port(self, port, force_http=False, http_cmd='get', conv=True): async def get_port(self, port, force_http=False, http_cmd="get", conv=True):
""" """
Запрос состояния порта. Состояние всегда возвращается в виде объекта, всегда сохраняется в центральное Запрос состояния порта. Состояние всегда возвращается в виде объекта, всегда сохраняется в центральное
хранилище values хранилище values
""" """
self.lg.debug(f'get port %s', port) self.lg.debug(f"get port %s", port)
if http_cmd == 'list' and conv: if http_cmd == "list" and conv:
await self.request(pt=port, cmd='conv') await self.request(pt=port, cmd="conv")
await asyncio.sleep(1) await asyncio.sleep(1)
ret = self.parse_response(await self.request(pt=port, cmd=http_cmd), cmd=http_cmd) ret = self.parse_response(
await self.request(pt=port, cmd=http_cmd), cmd=http_cmd
)
ntry = 0 ntry = 0
while http_cmd == 'list' and ret is None and ntry < 3: while http_cmd == "list" and ret is None and ntry < 3:
await asyncio.sleep(1) await asyncio.sleep(1)
ret = self.parse_response(await self.request(pt=port, cmd=http_cmd)) ret = self.parse_response(await self.request(pt=port, cmd=http_cmd))
ntry += 1 ntry += 1
self.lg.debug('parsed: %s', ret) self.lg.debug("parsed: %s", ret)
self.values[port] = ret self.values[port] = ret
return ret return ret
@property @property
def ports(self): def ports(self):
return {e.port for e in self.entities} return {e.port for e in self.entities}
async def get_all_ports(self, only_out=False, check_skip=False): async def get_all_ports(self, only_out=False, check_skip=False):
try: try:
ret = await self.request(cmd='all') ret = await self.request(cmd="all")
except asyncio.TimeoutError: except asyncio.TimeoutError:
return return
for port, x in enumerate(ret.split(';')): for port, x in enumerate(ret.split(";")):
if port in self.ds2413_ports: if port in self.ds2413_ports:
continue continue
if check_skip and not port in self.ports: if check_skip and not port in self.ports:
@@ -387,22 +395,20 @@ class MegaD:
def _process_msg(self, msg): def _process_msg(self, msg):
try: try:
d = msg.topic.split('/') d = msg.topic.split("/")
port = d[-1] port = d[-1]
except ValueError: except ValueError:
self.lg.warning('can not process %s', msg) self.lg.warning("can not process %s", msg)
return return
if port == 'cmd': if port == "cmd":
return return
try: try:
port = int_ignore(port) port = int_ignore(port)
except: except:
self.lg.warning('can not process %s', msg) self.lg.warning("can not process %s", msg)
return return
self.lg.debug( self.lg.debug("process incomming message: %s", msg)
'process incomming message: %s', msg
)
value = None value = None
try: try:
value = json.loads(msg.payload) value = json.loads(msg.payload)
@@ -413,13 +419,13 @@ class MegaD:
cb(value) cb(value)
if isinstance(value, dict): if isinstance(value, dict):
value = value.copy() value = value.copy()
value['mega_id'] = self.id value["mega_id"] = self.id
self.hass.bus.async_fire( self.hass.bus.async_fire(
EVENT_BINARY_SENSOR, EVENT_BINARY_SENSOR,
value, value,
) )
except Exception as exc: except Exception as exc:
self.lg.warning(f'could not parse json ({msg.payload}): {exc}') self.lg.warning(f"could not parse json ({msg.payload}): {exc}")
return return
finally: finally:
asyncio.run_coroutine_threadsafe(self._notify(port, value), self.loop) asyncio.run_coroutine_threadsafe(self._notify(port, value), self.loop)
@@ -427,14 +433,15 @@ class MegaD:
def subscribe(self, port, callback): def subscribe(self, port, callback):
port = int_ignore(port) port = int_ignore(port)
self.lg.debug( self.lg.debug(
f'subscribe %s', port, f"subscribe %s",
port,
) )
self.http.callbacks[self.id][port].append(callback) self.http.callbacks[self.id][port].append(callback)
async def authenticate(self) -> bool: async def authenticate(self) -> bool:
"""Test if we can authenticate with the host.""" """Test if we can authenticate with the host."""
async with aiohttp.request("get", url=f"http://{self.host}/{self.sec}") as req: async with aiohttp.request("get", url=f"http://{self.host}/{self.sec}") as req:
if "Unauthorized" in await req.text(): if "Unauthorized" in await req.text(encoding="iso-8859-5"):
return False return False
else: else:
if req.status != 200: if req.status != 200:
@@ -442,21 +449,21 @@ class MegaD:
return True return True
async def get_port_page(self, port): async def get_port_page(self, port):
url = f'http://{self.host}/{self.sec}/?pt={port}' url = f"http://{self.host}/{self.sec}/?pt={port}"
self.lg.debug(f'get page for port {port} {url}') self.lg.debug(f"get page for port {port} {url}")
async with aiohttp.request('get', url) as req: async with aiohttp.request("get", url) as req:
return await req.text() return await req.text(encoding="iso-8859-5")
async def scan_port(self, port): async def scan_port(self, port):
data = await self.request(pt=port) data = await self.request(pt=port)
return parse_config(data) return parse_config(data)
async def scan_ports(self, nports=37): async def scan_ports(self, nports=37):
for x in range(0, nports+1): for x in range(0, nports + 1):
ret = await self.scan_port(x) ret = await self.scan_port(x)
if ret: if ret:
yield x, ret yield x, ret
self.nports = nports+1 self.nports = nports + 1
async def _update_extender(self, port): async def _update_extender(self, port):
""" """
@@ -465,12 +472,12 @@ class MegaD:
:return: :return:
""" """
try: try:
values = await self.request(pt=port, cmd='get') values = await self.request(pt=port, cmd="get")
except asyncio.TimeoutError: except asyncio.TimeoutError:
return return
ret = {} ret = {}
for i, x in enumerate(values.split(';')): for i, x in enumerate(values.split(";")):
ret[f'{port}e{i}'] = x ret[f"{port}e{i}" if not self.new_naming else f"{port:02d}e{i:02d}"] = x
return ret return ret
async def _update_i2c(self, params): async def _update_i2c(self, params):
@@ -479,130 +486,161 @@ class MegaD:
:param params: параметры url :param params: параметры url
:return: :return:
""" """
pt = params.get('pt') pt = params.get("pt")
if pt in self.skip_ports: if pt in self.skip_ports:
return return
if pt is not None: if pt is not None:
pass pass
_params = tuple(params.items()) _params = tuple(params.items())
delay = None delay = None
if 'delay' in params: if "delay" in params:
delay = params.pop('delay') delay = params.pop("delay")
try: try:
ret = { ret = {_params: await self.request(**params)}
_params: await self.request(**params)
}
except asyncio.TimeoutError: except asyncio.TimeoutError:
return return
self.lg.debug('i2c response: %s', ret) self.lg.debug("i2c response: %s", ret)
if delay: if delay:
self.lg.debug('delay %s', delay) self.lg.debug("delay %s", delay)
await asyncio.sleep(delay) await asyncio.sleep(delay)
return ret return ret
async def get_config(self, nports=37): async def get_config(self, nports=37):
ret = defaultdict(lambda: defaultdict(list)) ret = defaultdict(lambda: defaultdict(list))
ret['mqtt_id'] = await self.get_mqtt_id() ret["mqtt_id"] = await self.get_mqtt_id()
ret['extenders'] = extenders = [] ret["extenders"] = extenders = []
ret['ext_in'] = ext_int = {} ret["ext_in"] = ext_int = {}
ret['ext_acts'] = ext_acts = {} ret["ext_acts"] = ext_acts = {}
ret['i2c_sensors'] = i2c_sensors = [] ret["i2c_sensors"] = i2c_sensors = []
ret['smooth'] = smooth = [] ret["smooth"] = smooth = []
async for port, cfg in self.scan_ports(nports): async for port, cfg in self.scan_ports(nports):
_cust = self.customize.get(port) _cust = self.customize.get(port)
if not isinstance(_cust, dict): if not isinstance(_cust, dict):
_cust = {} _cust = {}
if cfg.pty == "0": if cfg.pty == "0":
ret['binary_sensor'][port].append({}) ret["binary_sensor"][port].append({})
elif cfg.pty == "1" and (cfg.m in ['0', '1', '3'] or cfg.m is None): elif cfg.pty == "1" and (cfg.m in ["0", "1", "3"] or cfg.m is None):
if cfg.misc is not None: if cfg.misc is not None:
smooth.append(port) smooth.append(port)
ret['light'][port].append({'dimmer': cfg.m == '1', 'smooth': safe_int(cfg.misc)}) ret["light"][port].append(
{"dimmer": cfg.m == "1", "smooth": safe_int(cfg.misc)}
)
elif cfg == DS2413: elif cfg == DS2413:
# ds2413 # ds2413
_data = await self.get_port(port=port, force_http=True, http_cmd='list', conv=False) _data = await self.get_port(
data = _data.get('value', {}) port=port, force_http=True, http_cmd="list", conv=False
)
data = _data.get("value", {})
if not isinstance(data, dict): if not isinstance(data, dict):
self.lg.warning(f'can not add ds2413 on port {port}, it has wrong data: {_data}') self.lg.warning(
f"can not add ds2413 on port {port}, it has wrong data: {_data}"
)
continue continue
for addr, state in data.items(): for addr, state in data.items():
ret['light'][port].extend([ ret["light"][port].extend(
{"index": 0, "addr": addr, "id_suffix": f'{addr}_a', "http_cmd": 'ds2413'}, [
{"index": 1, "addr": addr, "id_suffix": f'{addr}_b', "http_cmd": 'ds2413'}, {
]) "index": 0,
"addr": addr,
"id_suffix": f"{addr}_a",
"http_cmd": "ds2413",
},
{
"index": 1,
"addr": addr,
"id_suffix": f"{addr}_b",
"http_cmd": "ds2413",
},
]
)
elif cfg == MCP230: elif cfg == MCP230:
extenders.append(port) extenders.append(port)
if cfg.inta: if cfg.inta:
ext_int[int_ignore(cfg.inta)] = port ext_int[int_ignore(cfg.inta)] = port
values = await self.request(pt=port, cmd='get') values = await self.request(pt=port, cmd="get")
values = values.split(';') values = values.split(";")
for n in range(len(values)): for n in range(len(values)):
ext_page = await self.request(pt=port, ext=n) ext_page = await self.request(pt=port, ext=n)
ext_cfg = parse_config(ext_page) ext_cfg = parse_config(ext_page)
pt = f'{port}e{n}' if not self.new_naming else f'{port:02d}e{n:02d}' pt = f"{port}e{n}" if not self.new_naming else f"{port:02d}e{n:02d}"
if ext_cfg.ety == '1': if ext_cfg.ety == "1":
ret['light'][pt].append({}) ret["light"][pt].append({})
elif ext_cfg.ety == '0': elif ext_cfg.ety == "0":
if ext_cfg.eact: if ext_cfg.eact:
ext_acts[pt] = ext_cfg.eact ext_acts[pt] = ext_cfg.eact
ret['binary_sensor'][pt].append({}) ret["binary_sensor"][pt].append({})
elif cfg == PCA9685: elif cfg == PCA9685:
extenders.append(port) extenders.append(port)
values = await self.request(pt=port, cmd='get') values = await self.request(pt=port, cmd="get")
values = values.split(';') values = values.split(";")
for n in range(len(values)): for n in range(len(values)):
pt = f'{port}e{n}' pt = f"{port}e{n}"
name = pt if not self.new_naming else f'{port:02}e{n:02}' name = pt if not self.new_naming else f"{port:02}e{n:02}"
ret['light'][pt].append({'dimmer': True, 'dimmer_scale': 16, 'name': f'{self.id}_{name}'}) ret["light"][pt].append(
if cfg.pty == '4': # and (cfg.gr == '0' or _cust.get(CONF_FORCE_I2C_SCAN)) {
"dimmer": True,
"dimmer_scale": 16,
"name": f"{self.id}_{name}",
}
)
if cfg.pty == "4": # and (cfg.gr == '0' or _cust.get(CONF_FORCE_I2C_SCAN))
# i2c в режиме ANY # i2c в режиме ANY
scan = cfg.src.find('a', text='I2C Scan') scan = cfg.src.find("a", text="I2C Scan")
self.lg.debug(f'find scan link: %s', scan) self.lg.debug(f"find scan link: %s", scan)
if scan is not None: if scan is not None:
page = await self.request(pt=port, cmd='scan') page = await self.request(pt=port, cmd="scan")
req, parsed = parse_scan_page(page) req, parsed = parse_scan_page(page)
self.lg.debug(f'scan results: %s', (req, parsed)) self.lg.debug(f"scan results: %s", (req, parsed))
ret['i2c'][port].extend(parsed) ret["i2c"][port].extend(parsed)
i2c_sensors.extend(req) i2c_sensors.extend(req)
elif cfg.pty == '4' and cfg.m == '2': elif cfg.pty == "4" and cfg.m == "2":
# scl исключаем из сканирования # scl исключаем из сканирования
continue continue
elif cfg.pty is None and nports < 30: elif cfg.pty is None and nports < 30:
# вроде как это ADC на 328 меге # вроде как это ADC на 328 меге
ret['sensor'][port].append(dict()) ret["sensor"][port].append(dict())
elif cfg.pty in ('3', '2', '4'): elif cfg.pty in ("3", "2", "4"):
http_cmd = 'get' http_cmd = "get"
if cfg.d == '5' and cfg.pty == '3': if cfg.d == "5" and cfg.pty == "3":
# 1-wire bus # 1-wire bus
values = await self.get_port(port, force_http=True, http_cmd='list') values = await self.get_port(port, force_http=True, http_cmd="list")
http_cmd = 'list' http_cmd = "list"
else: else:
values = await self.get_port(port, force_http=True) values = await self.get_port(port, force_http=True)
if values is None or (isinstance(values, dict) and str(values.get('value')) in ('', 'None')): if values is None or (
values = await self.get_port(port, force_http=True, http_cmd='list') isinstance(values, dict)
http_cmd = 'list' and str(values.get("value")) in ("", "None")
self.lg.debug(f'values: %s', values) ):
values = await self.get_port(
port, force_http=True, http_cmd="list"
)
http_cmd = "list"
self.lg.debug(f"values: %s", values)
if values is None: if values is None:
self.lg.warning(f'port {port} is of type sensor but response is None, skipping it') self.lg.warning(
f"port {port} is of type sensor but response is None, skipping it"
)
continue continue
if isinstance(values, dict) and 'value' in values: if isinstance(values, dict) and "value" in values:
values = values['value'] values = values["value"]
if isinstance(values, str) and TEMP_PATT.search(values): if isinstance(values, str) and TEMP_PATT.search(values):
values = {TEMP: values} values = {TEMP: values}
elif not isinstance(values, dict): elif not isinstance(values, dict):
if cfg.pty == '4' and cfg.d in I2C_DEVICE_TYPES: if cfg.pty == "4" and cfg.d in I2C_DEVICE_TYPES:
values = {I2C_DEVICE_TYPES.get(cfg.m): values} values = {I2C_DEVICE_TYPES.get(cfg.m): values}
else: else:
values = {None: values} values = {None: values}
for key in values: for key in values:
self.lg.debug(f'add sensor {key}') self.lg.debug(f"add sensor {key}")
ret['sensor'][port].append(dict( ret["sensor"][port].append(
key=key, dict(
unit_of_measurement=UNITS.get(key, UNITS[TEMP]), key=key,
device_class=CLASSES.get(key, CLASSES[TEMP]), unit_of_measurement=UNITS.get(key, UNITS[TEMP]),
id_suffix=key, device_class=CLASSES.get(key, CLASSES[TEMP]),
http_cmd=http_cmd, id_suffix=key,
)) http_cmd=http_cmd,
)
)
return ret return ret
async def restore_states(self): async def restore_states(self):
@@ -614,10 +652,7 @@ class MegaD:
await x.async_turn_off() await x.async_turn_off()
async def update_time(self): async def update_time(self):
await self.request( await self.request(cf=7, stime=datetime.now().strftime("%H:%M:%S"))
cf=7,
stime=datetime.now().strftime('%H:%M:%S')
)
async def reload(self, reload_entry=True): async def reload(self, reload_entry=True):
new = await self.get_config(nports=self.nports) new = await self.get_config(nports=self.nports)
@@ -625,14 +660,14 @@ class MegaD:
for x in REMOVE_CONFIG: for x in REMOVE_CONFIG:
cfg.pop(x, None) cfg.pop(x, None)
cfg.update(new) cfg.update(new)
self.lg.debug(f'new config: %s', cfg) self.lg.debug(f"new config: %s", cfg)
self.config.data = cfg self.config.data = cfg
if reload_entry: if reload_entry:
await self.hass.config_entries.async_reload(self.config.entry_id) await self.hass.config_entries.async_reload(self.config.entry_id)
return cfg return cfg
def _wrap_port_smooth(self, from_, to_, time): def _wrap_port_smooth(self, from_, to_, time):
self.lg.debug('dim from %s to %s for %s seconds', from_, to_, time) self.lg.debug("dim from %s to %s for %s seconds", from_, to_, time)
if time <= 0: if time <= 0:
return return
beg = datetime.now() beg = datetime.now()
@@ -645,15 +680,15 @@ class MegaD:
yield val yield val
async def smooth_dim( async def smooth_dim(
self, self,
*config: typing.Tuple[typing.Any, int, int], *config: typing.Tuple[typing.Any, int, int],
time: float, time: float,
jitter: int = 50, jitter: int = 50,
ws=False, ws=False,
updater=None, updater=None,
can_smooth_hardware=False, can_smooth_hardware=False,
max_values=None, max_values=None,
chip=None, chip=None,
): ):
""" """
Плавное диммирование силами сервера, сразу нескольких портов (одной командой) Плавное диммирование силами сервера, сразу нескольких портов (одной командой)
@@ -697,7 +732,9 @@ class MegaD:
continue continue
if not ws: if not ws:
cmd = dict( cmd = dict(
cmd=';'.join([f'{pt}:{_next_val[i]}' for i, (pt, _, _) in enumerate(config)]) cmd=";".join(
[f"{pt}:{_next_val[i]}" for i, (pt, _, _) in enumerate(config)]
)
) )
await self.request(**cmd) await self.request(**cmd)
else: else:
@@ -705,7 +742,9 @@ class MegaD:
cmd = dict( cmd = dict(
pt=config[0][0], pt=config[0][0],
chip=chip, chip=chip,
ws=''.join([hex(x).split('x')[1].rjust(2, '0').upper() for x in _next_val]) ws="".join(
[hex(x).split("x")[1].rjust(2, "0").upper() for x in _next_val]
),
) )
await self.request(**cmd) await self.request(**cmd)

View File

@@ -16,7 +16,9 @@ from homeassistant.components.light import (
SUPPORT_BRIGHTNESS, SUPPORT_BRIGHTNESS,
LightEntity, LightEntity,
SUPPORT_TRANSITION, SUPPORT_TRANSITION,
SUPPORT_COLOR, ColorMode, LightEntityFeature, SUPPORT_COLOR,
ColorMode,
LightEntityFeature,
# SUPPORT_WHITE_VALUE # SUPPORT_WHITE_VALUE
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@@ -36,7 +38,15 @@ from .const import (
CONF_SWITCH, CONF_SWITCH,
DOMAIN, DOMAIN,
CONF_CUSTOM, CONF_CUSTOM,
CONF_SKIP, CONF_LED, CONF_WS28XX, CONF_PORTS, CONF_WHITE_SEP, CONF_SMOOTH, CONF_ORDER, CONF_CHIP, RGB, CONF_SKIP,
CONF_LED,
CONF_WS28XX,
CONF_PORTS,
CONF_WHITE_SEP,
CONF_SMOOTH,
CONF_ORDER,
CONF_CHIP,
RGB,
) )
from .tools import int_ignore, map_reorder_rgb from .tools import int_ignore, map_reorder_rgb
@@ -64,13 +74,17 @@ PLATFORM_SCHEMA = LIGHT_SCHEMA.extend(
async def async_setup_platform(hass, config, add_entities, discovery_info=None): async def async_setup_platform(hass, config, add_entities, discovery_info=None):
lg.warning('mega integration does not support yaml for lights, please use UI configuration') lg.warning(
"mega integration does not support yaml for lights, please use UI configuration"
)
return True return True
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_devices): async def async_setup_entry(
hass: HomeAssistant, config_entry: ConfigEntry, async_add_devices
):
mid = config_entry.data[CONF_ID] mid = config_entry.data[CONF_ID]
hub: MegaD = hass.data['mega'][mid] hub: MegaD = hass.data["mega"][mid]
devices = [] devices = []
customize = hass.data.get(DOMAIN, {}).get(CONF_CUSTOM, {}).get(mid, {}) customize = hass.data.get(DOMAIN, {}).get(CONF_CUSTOM, {}).get(mid, {})
skip = [] skip = []
@@ -78,23 +92,29 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
for entity_id, conf in customize[CONF_LED].items(): for entity_id, conf in customize[CONF_LED].items():
ports = conf.get(CONF_PORTS) or [conf.get(CONF_PORT)] ports = conf.get(CONF_PORTS) or [conf.get(CONF_PORT)]
skip.extend(ports) skip.extend(ports)
devices.append(MegaRGBW( devices.append(
mega=hub, MegaRGBW(
port=ports, mega=hub,
name=entity_id, port=ports,
customize=conf, name=entity_id,
id_suffix=entity_id, customize=conf,
config_entry=config_entry id_suffix=entity_id,
)) config_entry=config_entry,
for port, cfg in config_entry.data.get('light', {}).items(): )
)
for port, cfg in config_entry.data.get("light", {}).items():
port = int_ignore(port) port = int_ignore(port)
c = customize.get(port, {}) c = customize.get(port, {})
if c.get(CONF_SKIP, False) or port in skip or c.get(CONF_DOMAIN, 'light') != 'light': if (
c.get(CONF_SKIP, False)
or port in skip
or c.get(CONF_DOMAIN, "light") != "light"
):
continue continue
for data in cfg: for data in cfg:
hub.lg.debug(f'add light on port %s with data %s', port, data) hub.lg.debug(f"add light on port %s with data %s", port, data)
light = MegaLight(mega=hub, port=port, config_entry=config_entry, **data) light = MegaLight(mega=hub, port=port, config_entry=config_entry, **data)
if '<' in light.name: if "<" in light.name:
continue continue
devices.append(light) devices.append(light)
@@ -102,17 +122,14 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
class MegaLight(MegaOutPort, LightEntity): class MegaLight(MegaOutPort, LightEntity):
@property @property
def supported_features(self): def supported_features(self):
return ( return (SUPPORT_BRIGHTNESS if self.dimmer else 0) | (
(SUPPORT_BRIGHTNESS if self.dimmer else 0) | SUPPORT_TRANSITION if self.dimmer else 0
(SUPPORT_TRANSITION if self.dimmer else 0)
) )
class MegaRGBW(LightEntity, BaseMegaEntity): class MegaRGBW(LightEntity, BaseMegaEntity):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self._is_on = None self._is_on = None
@@ -123,7 +140,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
self._task: asyncio.Task = None self._task: asyncio.Task = None
self._restore = None self._restore = None
self.smooth: timedelta = self.customize[CONF_SMOOTH] self.smooth: timedelta = self.customize[CONF_SMOOTH]
self._color_order = self.customize.get(CONF_ORDER, 'rgb') self._color_order = self.customize.get(CONF_ORDER, "rgb")
self._last_called: float = 0 self._last_called: float = 0
self._max_values = None self._max_values = None
@@ -146,12 +163,11 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
def is_ws(self): def is_ws(self):
return self.customize.get(CONF_WS28XX) return self.customize.get(CONF_WS28XX)
@property @property
def supported_color_modes(self) -> set[ColorMode] | set[str] | None: def supported_color_modes(self) -> set[ColorMode] | set[str] | None:
return { return {
ColorMode.BRIGHTNESS, ColorMode.BRIGHTNESS,
ColorMode.RGB if len(self.port) != 4 else ColorMode.RGBW ColorMode.RGB if len(self.port) != 4 else ColorMode.RGBW,
} }
@property @property
@@ -164,7 +180,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
@property @property
def white_value(self): def white_value(self):
# if self.supported_features & SUPPORT_WHITE_VALUE: # if self.supported_features & SUPPORT_WHITE_VALUE:
return float(self.get_attribute('white_value', 0)) return float(self.get_attribute("white_value", 0))
@property @property
def rgb_color(self) -> tuple[int, int, int] | None: def rgb_color(self) -> tuple[int, int, int] | None:
@@ -177,15 +193,15 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
@property @property
def brightness(self): def brightness(self):
return float(self.get_attribute('brightness', 0)) return float(self.get_attribute("brightness", 0))
@property @property
def hs_color(self): def hs_color(self):
return self.get_attribute('hs_color', [0, 0]) return self.get_attribute("hs_color", [0, 0])
@property @property
def is_on(self): def is_on(self):
return self.get_attribute('is_on', False) return self.get_attribute("is_on", False)
@property @property
def supported_features(self): def supported_features(self):
@@ -195,7 +211,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
if not self.is_on: if not self.is_on:
return [0 for x in range(len(self.port))] if not self.is_ws else [0] * 3 return [0 for x in range(len(self.port))] if not self.is_ws else [0] * 3
rgb = colorsys.hsv_to_rgb( rgb = colorsys.hsv_to_rgb(
self.hs_color[0]/360, self.hs_color[1]/100, self.brightness / 255 self.hs_color[0] / 360, self.hs_color[1] / 100, self.brightness / 255
) )
rgb = [x for x in rgb] rgb = [x for x in rgb]
if self.white_value is not None: if self.white_value is not None:
@@ -203,9 +219,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
if not self.customize.get(CONF_WHITE_SEP): if not self.customize.get(CONF_WHITE_SEP):
white = white * (self.brightness / 255) white = white * (self.brightness / 255)
rgb.append(white / 255) rgb.append(white / 255)
rgb = [ rgb = [round(x * self.max_values[i]) for i, x in enumerate(rgb)]
round(x * self.max_values[i]) for i, x in enumerate(rgb)
]
if self.is_ws: if self.is_ws:
# восстанавливаем мэпинг # восстанавливаем мэпинг
rgb = map_reorder_rgb(rgb, RGB, self._color_order) rgb = map_reorder_rgb(rgb, RGB, self._color_order)
@@ -215,7 +229,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
if (time.time() - self._last_called) < 0.1: if (time.time() - self._last_called) < 0.1:
return return
self._last_called = time.time() self._last_called = time.time()
self.lg.debug(f'turn on %s with kwargs %s', self.entity_id, kwargs) self.lg.debug(f"turn on %s with kwargs %s", self.entity_id, kwargs)
if self._restore is not None: if self._restore is not None:
self._restore.update(kwargs) self._restore.update(kwargs)
kwargs = self._restore kwargs = self._restore
@@ -231,9 +245,9 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
return return
self._last_called = time.time() self._last_called = time.time()
self._restore = { self._restore = {
'hs_color': self.hs_color, "hs_color": self.hs_color,
'brightness': self.brightness, "brightness": self.brightness,
'white_value': self.white_value, "white_value": self.white_value,
} }
_before = self.get_rgbw() _before = self.get_rgbw()
self._is_on = False self._is_on = False
@@ -242,18 +256,22 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
self._task = asyncio.create_task(self.set_color(_before, **kwargs)) self._task = asyncio.create_task(self.set_color(_before, **kwargs))
async def set_color(self, _before, **kwargs): async def set_color(self, _before, **kwargs):
transition = kwargs.get('transition') transition = kwargs.get("transition")
update_state = transition is not None and transition > 3 update_state = transition is not None and transition > 3
_after = None
for item, value in kwargs.items(): for item, value in kwargs.items():
setattr(self, f'_{item}', value) setattr(self, f"_{item}", value)
_after = self.get_rgbw() if item == "rgb_color":
self._rgb_color = tuple(_after[:3]) _after = map_reorder_rgb(value, RGB, self._color_order)
self._hs_color = colorsys.rgb_to_hsv(*value)
_after = _after or self.get_rgbw()
self._rgb_color = map_reorder_rgb(tuple(_after[:3]), self._color_order, RGB)
if transition is None: if transition is None:
transition = self.smooth.total_seconds() transition = self.smooth.total_seconds()
ratio = self.calc_speed_ratio(_before, _after) ratio = self.calc_speed_ratio(_before, _after)
transition = transition * ratio transition = transition * ratio
self.async_write_ha_state() self.async_write_ha_state()
ports = self.port if not self.is_ws else self.port*3 ports = self.port if not self.is_ws else self.port * 3
config = [(port, _before[i], _after[i]) for i, port in enumerate(ports)] config = [(port, _before[i], _after[i]) for i, port in enumerate(ports)]
try: try:
await self.mega.smooth_dim( await self.mega.smooth_dim(
@@ -269,7 +287,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
except asyncio.CancelledError: except asyncio.CancelledError:
return return
except: except:
self.lg.exception('while dimming') self.lg.exception("while dimming")
async def async_will_remove_from_hass(self) -> None: async def async_will_remove_from_hass(self) -> None:
await super().async_will_remove_from_hass() await super().async_will_remove_from_hass()
@@ -284,10 +302,10 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
w = None w = None
rgb = rgbw rgb = rgbw
if self.is_ws: if self.is_ws:
rgb = map_reorder_rgb( rgb = map_reorder_rgb(rgb, self._color_order, RGB)
rgb, self._color_order, RGB h, s, v = colorsys.rgb_to_hsv(
) *[x / self.max_values[i] for i, x in enumerate(rgb)]
h, s, v = colorsys.rgb_to_hsv(*[x/self.max_values[i] for i, x in enumerate(rgb)]) )
h *= 360 h *= 360
s *= 100 s *= 100
v *= 255 v *= 255
@@ -296,7 +314,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
self._brightness = v self._brightness = v
if w is not None: if w is not None:
if not self.customize.get(CONF_WHITE_SEP): if not self.customize.get(CONF_WHITE_SEP):
w = w/(self._brightness / 255) w = w / (self._brightness / 255)
else: else:
w = w w = w
w = w / (self.max_values[-1] / 255) w = w / (self.max_values[-1] / 255)
@@ -321,7 +339,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
return return
data = data.get(x, None) data = data.get(x, None)
if isinstance(data, dict): if isinstance(data, dict):
data = data.get('value') data = data.get("value")
data = safe_int(data) data = safe_int(data)
if data is None: if data is None:
return return
@@ -338,4 +356,4 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
ret = r ret = r
else: else:
ret = max([r, ret]) ret = max([r, ret])
return ret return ret

View File

@@ -15,5 +15,5 @@
"@andvikt" "@andvikt"
], ],
"issue_tracker": "https://github.com/andvikt/mega_hacs/issues", "issue_tracker": "https://github.com/andvikt/mega_hacs/issues",
"version": "v1.1.5" "version": "v1.1.8b6"
} }

View File

@@ -4,7 +4,9 @@ import voluptuous as vol
import struct import struct
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
PLATFORM_SCHEMA as SENSOR_SCHEMA, SensorEntity, SensorDeviceClass PLATFORM_SCHEMA as SENSOR_SCHEMA,
SensorEntity,
SensorDeviceClass,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
@@ -12,35 +14,46 @@ from homeassistant.const import (
CONF_PORT, CONF_PORT,
CONF_UNIQUE_ID, CONF_UNIQUE_ID,
CONF_ID, CONF_ID,
CONF_TYPE, CONF_UNIT_OF_MEASUREMENT, CONF_VALUE_TEMPLATE, CONF_TYPE,
CONF_UNIT_OF_MEASUREMENT,
CONF_VALUE_TEMPLATE,
CONF_DEVICE_CLASS, CONF_DEVICE_CLASS,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.template import Template from homeassistant.helpers.template import Template
from .entities import MegaPushEntity from .entities import MegaPushEntity
from .const import CONF_KEY, TEMP, HUM, W1, W1BUS, CONF_CONV_TEMPLATE, CONF_HEX_TO_FLOAT, DOMAIN, CONF_CUSTOM, \ from .const import (
CONF_SKIP, CONF_FILTER_VALUES, CONF_FILTER_SCALE, CONF_FILTER_LOW, CONF_FILTER_HIGH, CONF_FILL_NA CONF_KEY,
TEMP,
HUM,
W1,
W1BUS,
CONF_CONV_TEMPLATE,
CONF_HEX_TO_FLOAT,
DOMAIN,
CONF_CUSTOM,
CONF_SKIP,
CONF_FILTER_VALUES,
CONF_FILTER_SCALE,
CONF_FILTER_LOW,
CONF_FILTER_HIGH,
CONF_FILL_NA,
)
from .hub import MegaD from .hub import MegaD
import re import re
from .tools import int_ignore from .tools import int_ignore
lg = logging.getLogger(__name__) lg = logging.getLogger(__name__)
TEMP_PATT = re.compile(r'temp:([01234567890\.]+)') TEMP_PATT = re.compile(r"temp:([01234567890\.]+)")
HUM_PATT = re.compile(r'hum:([01234567890\.]+)') HUM_PATT = re.compile(r"hum:([01234567890\.]+)")
PATTERNS = { PATTERNS = {
TEMP: TEMP_PATT, TEMP: TEMP_PATT,
HUM: HUM_PATT, HUM: HUM_PATT,
} }
UNITS = { UNITS = {TEMP: "°C", HUM: "%"}
TEMP: '°C', CLASSES = {TEMP: SensorDeviceClass.TEMPERATURE, HUM: SensorDeviceClass.HUMIDITY}
HUM: '%'
}
CLASSES = {
TEMP: SensorDeviceClass.TEMPERATURE,
HUM: SensorDeviceClass.HUMIDITY
}
# Validation of the user's configuration # Validation of the user's configuration
_ITEM = { _ITEM = {
vol.Required(CONF_PORT): int, vol.Required(CONF_PORT): int,
@@ -50,18 +63,18 @@ _ITEM = {
W1, W1,
W1BUS, W1BUS,
), ),
vol.Optional(CONF_KEY, default=''): str, vol.Optional(CONF_KEY, default=""): str,
} }
PLATFORM_SCHEMA = SENSOR_SCHEMA.extend( PLATFORM_SCHEMA = SENSOR_SCHEMA.extend(
{ {vol.Optional(str, description="mega id"): [_ITEM]},
vol.Optional(str, description="mega id"): [_ITEM]
},
extra=vol.ALLOW_EXTRA, extra=vol.ALLOW_EXTRA,
) )
async def async_setup_platform(hass, config, add_entities, discovery_info=None): async def async_setup_platform(hass, config, add_entities, discovery_info=None):
lg.warning('mega integration does not support yaml for sensors, please use UI configuration') lg.warning(
"mega integration does not support yaml for sensors, please use UI configuration"
)
return True return True
@@ -72,19 +85,23 @@ def _make_entity(config_entry, mid: str, port: int, conf: dict):
mega_id=mid, mega_id=mid,
port=port, port=port,
patt=PATTERNS.get(key), patt=PATTERNS.get(key),
unit_of_measurement=UNITS.get(key, UNITS[TEMP]), # TODO: make other units, make options in config flow unit_of_measurement=UNITS.get(
key, UNITS[TEMP]
), # TODO: make other units, make options in config flow
device_class=CLASSES.get(key, CLASSES[TEMP]), device_class=CLASSES.get(key, CLASSES[TEMP]),
id_suffix=key, id_suffix=key,
config_entry=config_entry config_entry=config_entry,
) )
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_devices): async def async_setup_entry(
hass: HomeAssistant, config_entry: ConfigEntry, async_add_devices
):
mid = config_entry.data[CONF_ID] mid = config_entry.data[CONF_ID]
hub: MegaD = hass.data['mega'][mid] hub: MegaD = hass.data["mega"][mid]
devices = [] devices = []
customize = hass.data.get(DOMAIN, {}).get(CONF_CUSTOM, {}).get(mid, {}) customize = hass.data.get(DOMAIN, {}).get(CONF_CUSTOM, {}).get(mid, {})
for tp in ['sensor', 'i2c']: for tp in ["sensor", "i2c"]:
for port, cfg in config_entry.data.get(tp, {}).items(): for port, cfg in config_entry.data.get(tp, {}).items():
port = int_ignore(port) port = int_ignore(port)
c = customize.get(port, {}) c = customize.get(port, {})
@@ -92,14 +109,19 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
hub.skip_ports |= {port} hub.skip_ports |= {port}
continue continue
for data in cfg: for data in cfg:
hub.lg.debug(f'add sensor on port %s with data %s, constructor: %s', port, data, _constructors[tp]) hub.lg.debug(
f"add sensor on port %s with data %s, constructor: %s",
port,
data,
_constructors[tp],
)
sensor = _constructors[tp]( sensor = _constructors[tp](
mega=hub, mega=hub,
port=port, port=port,
config_entry=config_entry, config_entry=config_entry,
**data, **data,
) )
if '<' in sensor.name: if "<" in sensor.name:
continue continue
devices.append(sensor) devices.append(sensor)
@@ -107,64 +129,76 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
class FilterBadValues(MegaPushEntity, SensorEntity): class FilterBadValues(MegaPushEntity, SensorEntity):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self._prev_value = None self._prev_value = None
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def filter_value(self, value): def filter_value(self, value):
if value is None:
return
try: try:
if value \ if (
in self.filter_values \ value in self.filter_values
or (self.filter_low is not None and value < self.filter_low) \ or (self.filter_low is not None and value < self.filter_low)
or (self.filter_high is not None and value > self.filter_high) \ or (self.filter_high is not None and value > self.filter_high)
or ( or (
self._prev_value is not None self._prev_value is not None
and self.filter_scale is not None and self.filter_scale is not None
and abs(value)
> 2 # при переходе через 0 каждое небольшое изменение будет иметь слишком большой эффект
and ( and (
abs(value - self._prev_value) / self._prev_value > self.filter_scale abs((value - self._prev_value) / self._prev_value)
> self.filter_scale
) )
)
): ):
if self.fill_na == 'last': if self.fill_na == "last":
value = self._prev_value value = self._prev_value
else: else:
value = None value = None
self._prev_value = value self._prev_value = value
return value return value
except Exception as exc: except Exception as exc:
lg.exception(f'while parsing value') lg.exception(f"while parsing value")
return None return None
@property @property
def filter_values(self): def filter_values(self):
return self.customize.get(CONF_FILTER_VALUES, self.mega.customize.get(CONF_FILTER_VALUES, [])) return self.customize.get(
CONF_FILTER_VALUES, self.mega.customize.get(CONF_FILTER_VALUES, [])
)
@property @property
def filter_scale(self): def filter_scale(self):
return self.customize.get(CONF_FILTER_SCALE, self.mega.customize.get(CONF_FILTER_SCALE, None)) return self.customize.get(
CONF_FILTER_SCALE, self.mega.customize.get(CONF_FILTER_SCALE, None)
)
@property @property
def filter_low(self): def filter_low(self):
return self.customize.get(CONF_FILTER_LOW, self.mega.customize.get(CONF_FILTER_LOW, None)) return self.customize.get(
CONF_FILTER_LOW, self.mega.customize.get(CONF_FILTER_LOW, None)
)
@property @property
def filter_high(self): def filter_high(self):
return self.customize.get(CONF_FILTER_HIGH, self.mega.customize.get(CONF_FILTER_HIGH, None)) return self.customize.get(
CONF_FILTER_HIGH, self.mega.customize.get(CONF_FILTER_HIGH, None)
)
@property @property
def fill_na(self): def fill_na(self):
return self.customize.get(CONF_FILL_NA, 'last') return self.customize.get(CONF_FILL_NA, "last")
class MegaI2C(FilterBadValues): class MegaI2C(FilterBadValues):
def __init__( def __init__(
self, self,
*args, *args,
device_class: str, device_class: str,
params: dict, params: dict,
unit_of_measurement: str = None, unit_of_measurement: str = None,
**kwargs **kwargs,
): ):
self._device_class = device_class self._device_class = device_class
self._params = tuple(params.items()) self._params = tuple(params.items())
@@ -183,9 +217,11 @@ class MegaI2C(FilterBadValues):
@property @property
def extra_state_attributes(self): def extra_state_attributes(self):
attrs = super().extra_state_attributes or {} attrs = super().extra_state_attributes or {}
attrs.update({ attrs.update(
'i2c_id': self.id_suffix, {
}) "i2c_id": self.id_suffix,
}
)
return attrs return attrs
@property @property
@@ -202,22 +238,24 @@ class MegaI2C(FilterBadValues):
ret = self.mega.values.get(self._params) ret = self.mega.values.get(self._params)
if self.customize.get(CONF_HEX_TO_FLOAT): if self.customize.get(CONF_HEX_TO_FLOAT):
try: try:
ret = struct.unpack('!f', bytes.fromhex(ret))[0] ret = struct.unpack("!f", bytes.fromhex(ret))[0]
except: except:
self.lg.warning(f'could not convert {ret} form hex to float') self.lg.warning(f"could not convert {ret} form hex to float")
tmpl: Template = self.customize.get(CONF_CONV_TEMPLATE, self.customize.get(CONF_VALUE_TEMPLATE)) tmpl: Template = self.customize.get(
CONF_CONV_TEMPLATE, self.customize.get(CONF_VALUE_TEMPLATE)
)
try: try:
ret = float(ret) ret = float(ret)
if tmpl is not None and self.hass is not None: if tmpl is not None and self.hass is not None:
tmpl.hass = self.hass tmpl.hass = self.hass
ret = tmpl.async_render({'value': ret}) ret = tmpl.async_render({"value": ret})
except: except:
ret = ret ret = ret
ret = self.filter_value(ret) ret = self.filter_value(ret)
if ret is not None: if ret is not None:
return str(ret) return str(ret)
except Exception: except Exception:
lg.exception('while getting value') lg.exception("while getting value")
return None return None
@property @property
@@ -226,14 +264,8 @@ class MegaI2C(FilterBadValues):
class Mega1WSensor(FilterBadValues): class Mega1WSensor(FilterBadValues):
def __init__( def __init__(
self, self, unit_of_measurement=None, device_class=None, key=None, *args, **kwargs
unit_of_measurement=None,
device_class=None,
key=None,
*args,
**kwargs
): ):
""" """
1-wire sensor entity 1-wire sensor entity
@@ -264,7 +296,7 @@ class Mega1WSensor(FilterBadValues):
@property @property
def unique_id(self): def unique_id(self):
if self.key: if self.key:
return super().unique_id + f'_{self.key}' return super().unique_id + f"_{self.key}"
else: else:
return super().unique_id return super().unique_id
@@ -284,23 +316,27 @@ class Mega1WSensor(FilterBadValues):
def native_value(self): def native_value(self):
try: try:
ret = None ret = None
if not hasattr(self, 'key'): if not hasattr(self, "key"):
return None return None
if self.key: if self.key:
try: try:
ret = self.mega.values.get(self.port, {}) ret = self.mega.values.get(self.port, {})
if isinstance(ret, dict): if isinstance(ret, dict):
ret = ret.get('value', {}) ret = ret.get("value", {})
if isinstance(ret, dict): if isinstance(ret, dict):
ret = ret.get(self.key) ret = ret.get(self.key)
except: except:
self.lg.error(self.mega.values.get(self.port, {}).get('value', {})) self.lg.error(self.mega.values.get(self.port, {}).get("value", {}))
return return
else: else:
ret = self.mega.values.get(self.port, {}).get('value') ret = self.mega.values.get(self.port, {}).get("value")
if ret is None and self.fill_na == 'fill_na' and self.prev_value is not None: if (
ret is None
and self.fill_na == "fill_na"
and self.prev_value is not None
):
ret = self.prev_value ret = self.prev_value
elif ret is None and self.fill_na == 'fill_na' and self._state is not None: elif ret is None and self.fill_na == "fill_na" and self._state is not None:
ret = self._state.state ret = self._state.state
try: try:
ret = float(ret) ret = float(ret)
@@ -310,15 +346,17 @@ class Mega1WSensor(FilterBadValues):
ret = self.prev_value ret = self.prev_value
if self.customize.get(CONF_HEX_TO_FLOAT): if self.customize.get(CONF_HEX_TO_FLOAT):
try: try:
ret = struct.unpack('!f', bytes.fromhex(ret))[0] ret = struct.unpack("!f", bytes.fromhex(ret))[0]
except: except:
self.lg.warning(f'could not convert {ret} form hex to float') self.lg.warning(f"could not convert {ret} form hex to float")
tmpl: Template = self.customize.get(CONF_CONV_TEMPLATE, self.customize.get(CONF_VALUE_TEMPLATE)) tmpl: Template = self.customize.get(
CONF_CONV_TEMPLATE, self.customize.get(CONF_VALUE_TEMPLATE)
)
try: try:
ret = float(ret) ret = float(ret)
if tmpl is not None and self.hass is not None: if tmpl is not None and self.hass is not None:
tmpl.hass = self.hass tmpl.hass = self.hass
ret = tmpl.async_render({'value': ret}) ret = tmpl.async_render({"value": ret})
except: except:
pass pass
ret = self.filter_value(ret) ret = self.filter_value(ret)
@@ -326,7 +364,7 @@ class Mega1WSensor(FilterBadValues):
if ret is not None: if ret is not None:
return str(ret) return str(ret)
except Exception: except Exception:
lg.exception('while parsing state') lg.exception("while parsing state")
return None return None
@property @property
@@ -342,6 +380,6 @@ class Mega1WSensor(FilterBadValues):
_constructors = { _constructors = {
'sensor': Mega1WSensor, "sensor": Mega1WSensor,
'i2c': MegaI2C, "i2c": MegaI2C,
} }