xep_0030: asyncio.wait takes tasks

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2023-02-15 11:55:57 +01:00
parent af07864cbb
commit 5c19f16287
Signed by: pep
GPG key ID: DEDA74AEECA9D0F2

View file

@ -315,13 +315,13 @@ class XEP_0030(BasePlugin):
domain = self.xmpp.boundjid.domain
if not cached or domain not in self.domain_infos:
infos = [self.get_info(
domain, timeout=timeout, **iqkwargs)]
infos = [asyncio.create_task(self.get_info(
domain, timeout=timeout, **iqkwargs))]
iq_items = await self.get_items(
domain, timeout=timeout, **iqkwargs)
items = iq_items['disco_items']['items']
infos += [
self.get_info(item[0], timeout=timeout, **iqkwargs)
asyncio.create_task(self.get_info(item[0], timeout=timeout, **iqkwargs))
for item in items]
info_futures, _ = await asyncio.wait(
infos,