jeka/mega_hacs
1
0
Fork 0
mirror of https://github.com/andvikt/mega_hacs.git synced 2025-12-12 01:24:29 +05:00
Code Issues Packages Projects Releases Wiki Activity

fix multiple megas

Browse Source
This commit is contained in:
Andrey
2021-01-25 20:17:15 +03:00
parent 39c4ab0e3b
commit 1548e8c364
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
custom_components/mega/http.py
View File

@@ -26,8 +26,7 @@ class MegaView(HomeAssistantView):
def __init__(self, cfg: dict): def __init__(self, cfg: dict):
self._try = 0 self._try = 0
self.allowed_hosts = {'::1'} self.allowed_hosts = {'::1'}
self.callbacks: typing.DefaultDict[int, typing.List[typing.Callable[[dict], typing.Coroutine]]] \ self.callbacks = defaultdict(lambda: defaultdict(list))
= defaultdict(list)
self.templates: typing.Dict[str, typing.Dict[str, Template]] = { self.templates: typing.Dict[str, typing.Dict[str, Template]] = {
mid: { mid: {
pt: cfg[mid][pt][CONF_RESPONSE_TEMPLATE] pt: cfg[mid][pt][CONF_RESPONSE_TEMPLATE]
@@ -64,7 +63,7 @@ class MegaView(HomeAssistantView):
data = data.copy() data = data.copy()
ret = 'd' ret = 'd'
if port is not None: if port is not None:
for cb in self.callbacks[port]: for cb in self.callbacks[hub.id][port]:
cb(data) cb(data)
template: Template = self.templates.get(hub.id, {}).get(port) template: Template = self.templates.get(hub.id, {}).get(port)
if hub.update_all: if hub.update_all:

2
custom_components/mega/hub.py
View File

@@ -307,7 +307,7 @@ class MegaD:
if self.mqtt_inputs: if self.mqtt_inputs:
self._callbacks[port].append(callback) self._callbacks[port].append(callback)
else: else:
self.http.callbacks[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."""