Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2024-11-26 12:45:26 +01:00
Code Issues Projects Releases Wiki Activity

Update records in one go (#119)

Browse Source
This commit is contained in:
Otto Winter 2021-10-13 10:56:29 +02:00 committed by GitHub
parent b3d1f13eeb
commit 5bfa90e7c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
aioesphomeapi/reconnect_logic.py
View File

@ -220,35 +220,42 @@ class ReconnectLogic(zeroconf.RecordUpdateListener): # type: ignore[misc,name-d
self._zc.async_remove_listener(self)
self._zc_listening = False
def _async_on_record(self, record: "zeroconf.DNSRecord") -> None: # type: ignore[name-defined]
if not isinstance(record, zeroconf.DNSPointer): # type: ignore[attr-defined]
# We only consider PTR records and match using the alias name
return
if self._is_stopped or self.name is None:
return
filter_alias = f"{self.name}._esphomelib._tcp.local."
if record.alias != filter_alias:
return
# This is a mDNS record from the device and could mean it just woke up
# Check if already connected, no lock needed for this access
if self._connected:
return
# Tell reconnection logic to retry connection attempt now (even before reconnect timer finishes)
_LOGGER.debug(
"%s: Triggering reconnect because of received mDNS record %s",
self._log_name,
record,
)
self._reconnect_event.set()
def async_update_records(
self,
zc: "zeroconf.Zeroconf", # pylint: disable=unused-argument
now: float, # pylint: disable=unused-argument
records: List["zeroconf.RecordUpdate"], # type: ignore[name-defined]
) -> None:
"""Listen to zeroconf updated mDNS records. This must be called from the eventloop."""
for update in records:
self._async_on_record(update.new)
"""Listen to zeroconf updated mDNS records. This must be called from the eventloop.
This is a mDNS record from the device and could mean it just woke up.
"""
# Check if already connected, no lock needed for this access and
# bail if either the already stopped or we haven't received device info yet
if (
self._connected
or self._reconnect_event.is_set()
or self._is_stopped
or self.name is None
):
return
filter_alias = f"{self.name}._esphomelib._tcp.local."
for record_update in records:
# We only consider PTR records and match using the alias name
if (
not isinstance(record_update.new, zeroconf.DNSPointer) # type: ignore[attr-defined]
or record_update.new.alias != filter_alias
):
continue
# Tell reconnection logic to retry connection attempt now (even before reconnect timer finishes)
_LOGGER.debug(
"%s: Triggering reconnect because of received mDNS record %s",
self._log_name,
record_update.new,
)
self._reconnect_event.set()
return