xep_0030: allow extra args in get_info_from_domain

This commit is contained in:
nicoco 2022-11-15 09:23:50 +01:00
parent 752f4258df
commit 6940e4276b

View file

@ -307,7 +307,7 @@ class XEP_0030(BasePlugin):
return self.api['has_identity'](jid, node, ifrom, data)
async def get_info_from_domain(self, domain=None, timeout=None,
cached=True, callback=None):
cached=True, callback=None, **iqkwargs):
"""Fetch disco#info of specified domain and one disco#items level below
"""
@ -316,12 +316,12 @@ class XEP_0030(BasePlugin):
if not cached or domain not in self.domain_infos:
infos = [self.get_info(
domain, timeout=timeout)]
domain, timeout=timeout, **iqkwargs)]
iq_items = await self.get_items(
domain, timeout=timeout)
domain, timeout=timeout, **iqkwargs)
items = iq_items['disco_items']['items']
infos += [
self.get_info(item[0], timeout=timeout)
self.get_info(item[0], timeout=timeout, **iqkwargs)
for item in items]
info_futures, _ = await asyncio.wait(
infos,