Fix upb config flow connect (#121571)

This commit is contained in:
Glenn Waters 2024-07-09 02:32:34 -04:00 committed by Franck Nijhof
parent 50802f84f0
commit 5a04a886cf
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3

View File

@ -39,12 +39,13 @@ async def _validate_input(data):
url = _make_url_from_data(data)
upb = upb_lib.UpbPim({"url": url, "UPStartExportFile": file_path})
upb.connect(_connected_callback)
if not upb.config_ok:
_LOGGER.error("Missing or invalid UPB file: %s", file_path)
raise InvalidUpbFile
upb.connect(_connected_callback)
with suppress(TimeoutError):
async with asyncio.timeout(VALIDATE_TIMEOUT):
await connected_event.wait()