Merge pull request #158 from andvikt/dev

Dev

.bumpversion.cfg

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

.gitignore

@@ -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

custom_components/mega/__init__.py

@@ -136,10 +136,10 @@ CONFIG_SCHEMA = vol.Schema(
                     CUSTOMIZE_PORT,
                     CUSTOMIZE_DS2413,
                 ),
-                vol.Optional(CONF_FILTER_VALUES): [cv.positive_float],
-                vol.Optional(CONF_FILTER_SCALE): cv.positive_float,
-                vol.Optional(CONF_FILTER_LOW): cv.positive_float,
-                vol.Optional(CONF_FILTER_HIGH): cv.positive_float,
+                vol.Optional(CONF_FILTER_VALUES): [vol.Coerce(float)],
+                vol.Optional(CONF_FILTER_SCALE): vol.Coerce(float),
+                vol.Optional(CONF_FILTER_LOW): vol.Coerce(float),
+                vol.Optional(CONF_FILTER_HIGH): vol.Coerce(float),
             },
             vol.Optional(CONF_1WBUS): [OWBUS]
         }

custom_components/mega/entities.py

@@ -8,6 +8,7 @@ from functools import partial
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_NAME
 from homeassistant.core import State
+from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.helpers.restore_state import RestoreEntity
 from . import hub as h
@@ -142,7 +143,7 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
         return self._customize
 
     @property
-    def device_info(self):
+    def device_info(self) -> DeviceInfo:
         if isinstance(self.port, list):
             pt_idx = self.id_suffix
         else:
@@ -151,19 +152,16 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
                 pt_idx, _ = _pt.split('e')
             else:
                 pt_idx = _pt
-        return {
-            "identifiers": {
+        return DeviceInfo(
+            identifiers={
                 # Serial numbers are unique identifiers within a specific domain
-                (DOMAIN, f'{self._mega_id}', pt_idx),
+                (DOMAIN, f'{self._mega_id}', pt_idx)
             },
-            "config_entries": [
-                self.config_entry,
-            ],
-            "name": f'{self._mega_id} port {pt_idx}' if not isinstance(self.port, list) else f'{self._mega_id} {pt_idx}',
-            "manufacturer": 'ab-log.ru',
-            "sw_version": self.mega.fw,
-            "via_device": (DOMAIN, self._mega_id),
-        }
+            name=self.name,
+            manufacturer='ab-log.ru',
+            sw_version=self.mega.fw,
+            via_device=(DOMAIN, self._mega_id),
+        )
 
     @property
     def lg(self) -> logging.Logger:

custom_components/mega/hub.py

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

custom_components/mega/i2c.py

@@ -1,13 +1,10 @@
 import typing
-from dataclasses import dataclass, field, astuple
+from dataclasses import dataclass, astuple
 from urllib.parse import parse_qsl, urlparse
 from bs4 import BeautifulSoup
+
+from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.const import (
-    DEVICE_CLASS_HUMIDITY,
-    DEVICE_CLASS_TEMPERATURE,
-    DEVICE_CLASS_ILLUMINANCE,
-    DEVICE_CLASS_PRESSURE,
-    DEVICE_CLASS_CO2,
     PERCENTAGE,
     LIGHT_LUX,
     TEMP_CELSIUS,
@@ -16,11 +13,11 @@ from homeassistant.const import (
 )
 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
@@ -89,55 +86,55 @@ class Request:
 
 i2c_classes = {
     'htu21d': [
-        DeviceType(DEVICE_CLASS_HUMIDITY, PERCENTAGE, None),
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
     ],
     'sht31': [
-        DeviceType(DEVICE_CLASS_HUMIDITY, PERCENTAGE, None, delay=1.5),
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None, delay=1.5),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
     ],
     'max44009': [
-        DeviceType(DEVICE_CLASS_ILLUMINANCE, LIGHT_LUX, None)
+        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)
     ],
     'bh1750': [
-        DeviceType(DEVICE_CLASS_ILLUMINANCE, LIGHT_LUX, None)
+        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)
     ],
     'tsl2591': [
-        DeviceType(DEVICE_CLASS_ILLUMINANCE, LIGHT_LUX, None)
+        DeviceType(SensorDeviceClass.ILLUMINANCE, LIGHT_LUX, None)
     ],
     'bmp180': [
-        DeviceType(DEVICE_CLASS_PRESSURE, PRESSURE_BAR, None),
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
+        DeviceType(SensorDeviceClass.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)
+        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.HUMIDITY, PERCENTAGE, None)
     ],
     'dps368': [
-        DeviceType(DEVICE_CLASS_PRESSURE, PRESSURE_BAR, None),
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
     ],
     'mlx90614': [
         Skip,
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, 'temp'),
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, 'object'),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, 'temp'),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, 'object'),
     ],
     'ptsensor': [
         Skip,
         Request(delay=3),  # запрос на измерение
-        DeviceType(DEVICE_CLASS_PRESSURE, PRESSURE_BAR, None),
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.PRESSURE, PRESSURE_BAR, None),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
     ],
     'mcp9600': [
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),  # термопара
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),  # сенсор встроенный в микросхему
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),  # термопара
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),  # сенсор встроенный в микросхему
     ],
     't67xx': [
-        DeviceType(DEVICE_CLASS_CO2, CONCENTRATION_PARTS_PER_MILLION, None)
+        DeviceType(SensorDeviceClass.CO2, CONCENTRATION_PARTS_PER_MILLION, None)
     ],
     'tmp117': [
-        DeviceType(DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None),
+        DeviceType(SensorDeviceClass.TEMPERATURE, TEMP_CELSIUS, None),
     ],
     'ads1115': [
         DeviceType(None, None, 'ch0'),

custom_components/mega/light.py

@@ -16,7 +16,9 @@ from homeassistant.components.light import (
     SUPPORT_BRIGHTNESS,
     LightEntity,
     SUPPORT_TRANSITION,
-    SUPPORT_COLOR, ColorMode, LightEntityFeature,
+    SUPPORT_COLOR,
+    ColorMode,
+    LightEntityFeature,
     # SUPPORT_WHITE_VALUE
 )
 from homeassistant.config_entries import ConfigEntry
@@ -36,7 +38,15 @@ from .const import (
     CONF_SWITCH,
     DOMAIN,
     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
 
@@ -64,13 +74,17 @@ PLATFORM_SCHEMA = LIGHT_SCHEMA.extend(
 
 
 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
 
 
-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]
-    hub: MegaD = hass.data['mega'][mid]
+    hub: MegaD = hass.data["mega"][mid]
     devices = []
     customize = hass.data.get(DOMAIN, {}).get(CONF_CUSTOM, {}).get(mid, {})
     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():
             ports = conf.get(CONF_PORTS) or [conf.get(CONF_PORT)]
             skip.extend(ports)
-            devices.append(MegaRGBW(
-                mega=hub,
-                port=ports,
-                name=entity_id,
-                customize=conf,
-                id_suffix=entity_id,
-                config_entry=config_entry
-            ))
-    for port, cfg in config_entry.data.get('light', {}).items():
+            devices.append(
+                MegaRGBW(
+                    mega=hub,
+                    port=ports,
+                    name=entity_id,
+                    customize=conf,
+                    id_suffix=entity_id,
+                    config_entry=config_entry,
+                )
+            )
+    for port, cfg in config_entry.data.get("light", {}).items():
         port = int_ignore(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
         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)
-            if '<' in light.name:
+            if "<" in light.name:
                 continue
             devices.append(light)
 
@@ -102,17 +122,14 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
 
 
 class MegaLight(MegaOutPort, LightEntity):
-
     @property
     def supported_features(self):
-        return (
-            (SUPPORT_BRIGHTNESS if self.dimmer else 0) |
-            (SUPPORT_TRANSITION if self.dimmer else 0)
+        return (SUPPORT_BRIGHTNESS if self.dimmer else 0) | (
+            SUPPORT_TRANSITION if self.dimmer else 0
         )
 
 
 class MegaRGBW(LightEntity, BaseMegaEntity):
-
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self._is_on = None
@@ -123,7 +140,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
         self._task: asyncio.Task = None
         self._restore = None
         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._max_values = None
 
@@ -131,7 +148,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
     def max_values(self) -> list:
         if self._max_values is None:
             if self.is_ws:
-                self._max_values = [255] * 3
+                self._max_values = [255] * 4
             else:
                 self._max_values = [
                     255 if isinstance(x, int) else 4095 for x in self.port
@@ -146,12 +163,11 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
     def is_ws(self):
         return self.customize.get(CONF_WS28XX)
 
-
     @property
     def supported_color_modes(self) -> set[ColorMode] | set[str] | None:
         return {
             ColorMode.BRIGHTNESS,
-            ColorMode.RGB if len(self.port) != 4 else ColorMode.RGBW
+            ColorMode.RGB if len(self.port) != 4 else ColorMode.RGBW,
         }
 
     @property
@@ -164,7 +180,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
     @property
     def white_value(self):
         # if self.supported_features & SUPPORT_WHITE_VALUE:
-        return float(self.get_attribute('white_value', 0))
+        return float(self.get_attribute("white_value", 0))
 
     @property
     def rgb_color(self) -> tuple[int, int, int] | None:
@@ -177,15 +193,15 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
 
     @property
     def brightness(self):
-        return float(self.get_attribute('brightness', 0))
+        return float(self.get_attribute("brightness", 0))
 
     @property
     def hs_color(self):
-        return self.get_attribute('hs_color', [0, 0])
+        return self.get_attribute("hs_color", [0, 0])
 
     @property
     def is_on(self):
-        return self.get_attribute('is_on', False)
+        return self.get_attribute("is_on", False)
 
     @property
     def supported_features(self):
@@ -195,7 +211,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
         if not self.is_on:
             return [0 for x in range(len(self.port))] if not self.is_ws else [0] * 3
         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]
         if self.white_value is not None:
@@ -203,9 +219,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
             if not self.customize.get(CONF_WHITE_SEP):
                 white = white * (self.brightness / 255)
             rgb.append(white / 255)
-        rgb = [
-            round(x * self.max_values[i]) for i, x in enumerate(rgb)
-        ]
+        rgb = [round(x * self.max_values[i]) for i, x in enumerate(rgb)]
         if self.is_ws:
             # восстанавливаем мэпинг
             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:
             return
         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:
             self._restore.update(kwargs)
             kwargs = self._restore
@@ -231,9 +245,9 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
             return
         self._last_called = time.time()
         self._restore = {
-            'hs_color': self.hs_color,
-            'brightness': self.brightness,
-            'white_value': self.white_value,
+            "hs_color": self.hs_color,
+            "brightness": self.brightness,
+            "white_value": self.white_value,
         }
         _before = self.get_rgbw()
         self._is_on = False
@@ -242,18 +256,21 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
         self._task = asyncio.create_task(self.set_color(_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
+        _after = None
         for item, value in kwargs.items():
-            setattr(self, f'_{item}', value)
-        _after = self.get_rgbw()
+            setattr(self, f"_{item}", value)
+            if item == "rgb_color":
+                _after = map_reorder_rgb(value, RGB, self._color_order)
+        _after = _after or self.get_rgbw()
         self._rgb_color = tuple(_after[:3])
         if transition is None:
             transition = self.smooth.total_seconds()
             ratio = self.calc_speed_ratio(_before, _after)
             transition = transition * ratio
         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)]
         try:
             await self.mega.smooth_dim(
@@ -269,7 +286,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
         except asyncio.CancelledError:
             return
         except:
-            self.lg.exception('while dimming')
+            self.lg.exception("while dimming")
 
     async def async_will_remove_from_hass(self) -> None:
         await super().async_will_remove_from_hass()
@@ -284,10 +301,10 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
             w = None
             rgb = rgbw
         if self.is_ws:
-            rgb = map_reorder_rgb(
-                rgb, self._color_order, RGB
-            )
-        h, s, v = colorsys.rgb_to_hsv(*[x/self.max_values[i] for i, x in enumerate(rgb)])
+            rgb = map_reorder_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 *= 360
         s *= 100
         v *= 255
@@ -296,7 +313,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
             self._brightness = v
         if w is not None:
             if not self.customize.get(CONF_WHITE_SEP):
-                w = w/(self._brightness / 255)
+                w = w / (self._brightness / 255)
             else:
                 w = w
             w = w / (self.max_values[-1] / 255)
@@ -321,7 +338,7 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
                 return
             data = data.get(x, None)
             if isinstance(data, dict):
-                data = data.get('value')
+                data = data.get("value")
             data = safe_int(data)
             if data is None:
                 return
@@ -338,4 +355,4 @@ class MegaRGBW(LightEntity, BaseMegaEntity):
                 ret = r
             else:
                 ret = max([r, ret])
-        return ret
+        return ret

custom_components/mega/manifest.json

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

custom_components/mega/sensor.py

@@ -4,9 +4,7 @@ import voluptuous as vol
 import struct
 
 from homeassistant.components.sensor import (
-    PLATFORM_SCHEMA as SENSOR_SCHEMA,
-    DEVICE_CLASS_TEMPERATURE,
-    DEVICE_CLASS_HUMIDITY, SensorEntity
+    PLATFORM_SCHEMA as SENSOR_SCHEMA, SensorEntity, SensorDeviceClass
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
@@ -40,8 +38,8 @@ UNITS = {
     HUM: '%'
 }
 CLASSES = {
-    TEMP: DEVICE_CLASS_TEMPERATURE,
-    HUM: DEVICE_CLASS_HUMIDITY
+    TEMP: SensorDeviceClass.TEMPERATURE,
+    HUM: SensorDeviceClass.HUMIDITY
 }
 # Validation of the user's configuration
 _ITEM = {
@@ -118,15 +116,15 @@ class FilterBadValues(MegaPushEntity, SensorEntity):
         try:
             if value \
                     in self.filter_values \
-               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 (
+                    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._prev_value is not None
                     and self.filter_scale is not None
                     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':
                     value = self._prev_value
                 else:
@@ -231,8 +229,8 @@ class Mega1WSensor(FilterBadValues):
 
     def __init__(
             self,
-            unit_of_measurement,
-            device_class,
+            unit_of_measurement=None,
+            device_class=None,
             key=None,
             *args,
             **kwargs

docs/index.md

@@ -12,7 +12,7 @@
 Если вам понравилась интеграция, не забудьте поставить звезду на гитхабе - вам не сложно, а мне приятно ) А если
 интеграция очень понравилась - еще приятнее, если вы воспользуетесь кнопкой доната )
 
-Обновление прошивки MegaD можно делать прямо из HA с помощью [аддона](https://github.com/andvikt/mega_addon.git)
+Обновление прошивки MegaD можно делать из HA с помощью [аддона](https://github.com/andvikt/mega_addon.git)
 
 ## Основные особенности {: #mains }
 - Настройка в [веб-интерфейсе](settings.md) + [yaml](yaml.md)

readme.md

@@ -1,7 +1,7 @@
 
 
 # MegaD HomeAssistant integration
-[![hacs_badge](https://img.shields.io/badge/HACS-Custom-orange.svg)](https://github.com/custom-components/hacs)
+[![hacs_badge](https://img.shields.io/badge/HACS-Custom-orange.svg)](https://github.com/hacs/integration)
 [![Donate](https://img.shields.io/badge/donate-Yandex-red.svg)](https://yoomoney.ru/to/410013955329136)
 
 Интеграция с [MegaD-2561, MegaD-328](https://www.ab-log.ru/smart-house/ethernet/megad-2561)