pep/poezio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

affiliations: display all relevant pieces of information we get

Browse source 
- Don't fail if only one of the requests fail
- Change UI a bit

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2020-04-15 00:23:23 +02:00
parent a364e651e2
commit 2181690b9a
Signed by: pep
GPG key ID: DEDA74AEECA9D0F2
1 changed files with 48 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
poezio/tabs/muctab.py
View file

@ -1626,24 +1626,59 @@ class MucTab(ChatTab):
async def get_users_affiliations(self, jid: JID) -> None:
MUC_ADMIN_NS = 'http://jabber.org/protocol/muc#admin'
try:
iqs = await asyncio.gather(
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'owner'),
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'admin'),
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'member'),
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'outcast'),
return_exceptions=True,
)
except (IqError, IqTimeout) as exn:
self.core.room_error(exn.iq, jid)
all_errors = functools.reduce(
lambda acc, iq: acc and isinstance(iq, (IqError, IqTimeout)),
iqs,
True,
)
theme = get_theme()
aff_colors = {
'owner': theme.CHAR_AFFILIATION_OWNER,
'admin': theme.CHAR_AFFILIATION_ADMIN,
'member': theme.CHAR_AFFILIATION_MEMBER,
'none': theme.CHAR_AFFILIATION_NONE,
}
if all_errors:
self.add_message(
'Can\'t access affiliations',
highlight=True,
nickname='Error',
nick_color=theme.COLOR_ERROR_MSG,
typ=2,
)
self.core.refresh_window()
return None
self._text_buffer.add_message('Affiliations:')
self._text_buffer.add_message('Affiliations')
for iq in iqs:
if isinstance(iq, (IqError, IqTimeout)):
continue
query = iq.xml.find('{%s}query' % MUC_ADMIN_NS)
for item in query.findall('{%s}item' % MUC_ADMIN_NS):
items = query.findall('{%s}item' % MUC_ADMIN_NS)
if not items: # Nobody with this affiliation
continue
affiliation = items[0].get('affiliation')
aff_char = aff_colors[affiliation]
self._text_buffer.add_message(
'%s: %s' % (item.get('jid'), item.get('affiliation'))
' %s%s' % (aff_char, affiliation.capitalize()),
)
items = map(lambda i: i.get('jid'), items)
for ajid in sorted(items):
self._text_buffer.add_message(' %s' % ajid)
self.core.refresh_window()
return None