Home-IoT/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2024-11-29 02:04:13 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

update storage version from mdns (#5584)

Browse source
This commit is contained in:
Guillermo Ruffino 2023-10-23 16:02:29 -03:00 committed by GitHub
parent b898f75631
commit eead33b6f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
esphome/zeroconf.py
View file

@ -17,6 +17,8 @@ from zeroconf import (
current_time_millis, current_time_millis,
) )
from esphome.storage_json import StorageJSON, ext_storage_path
_CLASS_IN = 1 _CLASS_IN = 1
_FLAGS_QR_QUERY = 0x0000 # query _FLAGS_QR_QUERY = 0x0000 # query
_TYPE_A = 1 _TYPE_A = 1
@ -131,6 +133,7 @@ TXT_RECORD_PROJECT_NAME = b"project_name"
TXT_RECORD_PROJECT_VERSION = b"project_version" TXT_RECORD_PROJECT_VERSION = b"project_version"
TXT_RECORD_NETWORK = b"network" TXT_RECORD_NETWORK = b"network"
TXT_RECORD_FRIENDLY_NAME = b"friendly_name" TXT_RECORD_FRIENDLY_NAME = b"friendly_name"
TXT_RECORD_VERSION = b"version"
@dataclass @dataclass
@ -186,6 +189,10 @@ class DashboardImportDiscovery:
] ]
if any(key not in info.properties for key in required_keys): if any(key not in info.properties for key in required_keys):
# Not a dashboard import device # Not a dashboard import device
version = info.properties.get(TXT_RECORD_VERSION)
if version is not None:
version = version.decode()
self.update_device_mdns(node_name, version)
return return
import_url = info.properties[TXT_RECORD_PACKAGE_IMPORT_URL].decode() import_url = info.properties[TXT_RECORD_PACKAGE_IMPORT_URL].decode()
@ -208,6 +215,22 @@ class DashboardImportDiscovery:
def cancel(self) -> None: def cancel(self) -> None:
self.service_browser.cancel() self.service_browser.cancel()
def update_device_mdns(self, node_name: str, version: str):
storage_path = ext_storage_path(node_name + ".yaml")
storage_json = StorageJSON.load(storage_path)
if storage_json is not None:
storage_version = storage_json.esphome_version
if version != storage_version:
storage_json.esphome_version = version
storage_json.save(storage_path)
_LOGGER.info(
"Updated %s with mdns version %s (was %s)",
node_name,
version,
storage_version,
)
class EsphomeZeroconf(Zeroconf): class EsphomeZeroconf(Zeroconf):
def resolve_host(self, host: str, timeout=3.0): def resolve_host(self, host: str, timeout=3.0):