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

plugins/contact: iterate all data forms, thx. mathieui

Browse source
This commit is contained in:
Georg Lukas 2021-01-22 15:37:32 +01:00
parent 12ab0a7e80
commit 5cfb0cbf43
1 changed files with 14 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
plugins/contact.py
View file

@ -32,25 +32,22 @@ class Plugin(BasePlugin):
self.api.information(message, 'Error')
return
info = iq['disco_info']
title = 'Contact Info'
contacts = []
for field in info['form']:
var = field['var']
if field['type'] == 'hidden' and var == 'FORM_TYPE':
form_type = field['value'][0]
if form_type != 'http://jabber.org/network/serverinfo':
self.api.information('Not a server: “%s”: %s' % (iq['from'], form_type), 'Error')
return
continue
if not var.endswith('-addresses'):
continue
var = var[:-10] # strip '-addresses'
sep = '\n ' + len(var) * ' '
field_value = field.get_value(convert=False)
value = sep.join(field_value) if isinstance(field_value, list) else field_value
contacts.append('%s: %s' % (var, value))
# iterate all data forms, in case there are multiple
for form in iq['disco_info']:
values = form.get_values()
if values['FORM_TYPE'][0] == 'http://jabber.org/network/serverinfo':
for var in values:
if not var.endswith('-addresses'):
continue
title = var[:-10] # strip '-addresses'
sep = '\n ' + len(title) * ' '
field_value = values[var]
if field_value:
value = sep.join(field_value) if isinstance(field_value, list) else field_value
contacts.append('%s: %s' % (title, value))
if contacts:
self.api.information('\n'.join(contacts), title)
self.api.information('\n'.join(contacts), 'Contact Info')
else:
self.api.information('No Contact Addresses for %s' % iq['from'], 'Error')