Mercurial > libervia-backend
view src/stdui/ui_contact_list.py @ 2030:58a44574c9ad
misc: fixed 0.6.1 tag
author | Goffi <goffi@goffi.org> |
---|---|
date | Mon, 08 Aug 2016 21:32:21 +0200 |
parents | 2daf7b4c6756 |
children | c2fdee1bd908 |
line wrap: on
line source
#!/usr/bin/env python2 # -*- coding: utf-8 -*- # SAT standard user interface for managing contacts # Copyright (C) 2009-2016 Jérôme Poisson (goffi@goffi.org) # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from sat.core.i18n import _, D_ from sat.core.constants import Const as C from sat.tools import xml_tools from twisted.words.protocols.jabber import jid from xml.dom.minidom import Element import re class ContactList(object): """Add, update and remove contacts.""" def __init__(self, host): self.host = host self.__add_id = host.registerCallback(self._addContact, with_data=True) self.__update_id = host.registerCallback(self._updateContact, with_data=True) self.__confirm_delete_id = host.registerCallback(self._getConfirmRemoveXMLUI, with_data=True) host.importMenu((D_("Contacts"), D_("Add contact")), self._getAddDialogXMLUI, security_limit=2, help_string=D_("Add contact")) host.importMenu((D_("Contacts"), D_("Update contact")), self._getUpdateDialogXMLUI, security_limit=2, help_string=D_("Update contact")) host.importMenu((D_("Contacts"), D_("Remove contact")), self._getRemoveDialogXMLUI, security_limit=2, help_string=D_("Remove contact")) if 'MISC-ACCOUNT' in self.host.plugins: self.default_host = self.host.plugins['MISC-ACCOUNT'].getNewAccountDomain() else: self.default_host = 'example.net' def getContacts(self, profile): """Return a sorted list of the contacts for that profile @param profile: %(doc_profile)s @return: list[string] """ client = self.host.getClient(profile) ret = [contact.full() for contact in client.roster.getJids()] ret.sort() return ret def getGroups(self, new_groups=None, profile=C.PROF_KEY_NONE): """Return a sorted list of the groups for that profile @param new_group (list): add these groups to the existing ones @param profile: %(doc_profile)s @return: list[string] """ client = self.host.getClient(profile) ret = client.roster.getGroups() ret.sort() ret.extend([group for group in new_groups if group not in ret]) return ret def getGroupsOfContact(self, user_jid_s, profile): """Return all the groups of the given contact @param user_jid_s (string) @param profile: %(doc_profile)s @return: list[string] """ client = self.host.getClient(profile) return client.roster.getItem(jid.JID(user_jid_s)).groups def getGroupsOfAllContacts(self, profile): """Return a mapping between the contacts and their groups @param profile: %(doc_profile)s @return: dict (key: string, value: list[string]): - key: the JID userhost - value: list of groups """ client = self.host.getClient(profile) return {item.jid.userhost(): item.groups for item in client.roster.getItems()} def _data2elts(self, data): """Convert a contacts data dict to minidom Elements @param data (dict) @return list[Element] """ elts = [] for key in data: key_elt = Element('jid') key_elt.setAttribute('name', key) for value in data[key]: value_elt = Element('group') value_elt.setAttribute('name', value) key_elt.childNodes.append(value_elt) elts.append(key_elt) return elts def getDialogXMLUI(self, options, data, profile): """Generic method to return the XMLUI dialog for adding or updating a contact @param options (dict): parameters for the dialog, with the keys: - 'id': the menu callback id - 'title': deferred localized string - 'contact_text': deferred localized string @param data (dict) @param profile: %(doc_profile)s @return dict """ form_ui = xml_tools.XMLUI("form", title=options['title'], submit_id=options['id']) if 'message' in data: form_ui.addText(data['message']) form_ui.addDivider('dash') form_ui.addText(options['contact_text']) if options['id'] == self.__add_id: contact = data.get(xml_tools.formEscape('contact_jid'), '@%s' % self.default_host) form_ui.addString('contact_jid', value=contact) elif options['id'] == self.__update_id: contacts = self.getContacts(profile) list_ = form_ui.addList('contact_jid', options=contacts, selected=contacts[0]) elts = self._data2elts(self.getGroupsOfAllContacts(profile)) list_.setInternalCallback('groups_of_contact', fields=['contact_jid', 'groups_list'], data_elts=elts) form_ui.addDivider('blank') form_ui.addText(_("Select in which groups your contact is:")) selected_groups = [] if 'selected_groups' in data: selected_groups = data['selected_groups'] elif options['id'] == self.__update_id: try: selected_groups = self.getGroupsOfContact(contacts[0], profile) except IndexError: pass groups = self.getGroups(selected_groups, profile) form_ui.addList('groups_list', options=groups, selected=selected_groups, styles=['multi']) adv_list = form_ui.changeContainer("advanced_list", columns=3, selectable='no') form_ui.addLabel(D_("Add group")) form_ui.addString("add_group") button = form_ui.addButton('', value=D_('Add')) button.setInternalCallback('move', fields=['add_group', 'groups_list']) adv_list.end() form_ui.addDivider('blank') return {'xmlui': form_ui.toXml()} def _getAddDialogXMLUI(self, data, profile): """Get the dialog for adding contact @param data (dict) @param profile: %(doc_profile)s @return dict """ options = {'id': self.__add_id, 'title': D_('Add contact'), 'contact_text': D_("New contact identifier (JID):"), } return self.getDialogXMLUI(options, {}, profile) def _getUpdateDialogXMLUI(self, data, profile): """Get the dialog for updating contact @param data (dict) @param profile: %(doc_profile)s @return dict """ if not self.getContacts(profile): _dialog = xml_tools.XMLUI('popup', title=D_('Nothing to update')) _dialog.addText(_('Your contact list is empty.')) return {'xmlui': _dialog.toXml()} options = {'id': self.__update_id, 'title': D_('Update contact'), 'contact_text': D_("Which contact do you want to update?"), } return self.getDialogXMLUI(options, {}, profile) def _getRemoveDialogXMLUI(self, data, profile): """Get the dialog for removing contact @param data (dict) @param profile: %(doc_profile)s @return dict """ if not self.getContacts(profile): _dialog = xml_tools.XMLUI('popup', title=D_('Nothing to delete')) _dialog.addText(_('Your contact list is empty.')) return {'xmlui': _dialog.toXml()} form_ui = xml_tools.XMLUI("form", title=D_('Who do you want to remove from your contacts?'), submit_id=self.__confirm_delete_id) form_ui.addList('contact_jid', options=self.getContacts(profile)) return {'xmlui': form_ui.toXml()} def _getConfirmRemoveXMLUI(self, data, profile): """Get the confirmation dialog for removing contact @param data (dict) @param profile: %(doc_profile)s @return dict """ contact = data[xml_tools.formEscape('contact_jid')] cb = lambda data, profile: self._deleteContact(jid.JID(contact), profile) delete_id = self.host.registerCallback(cb, with_data=True, one_shot=True) form_ui = xml_tools.XMLUI("form", title=D_("Delete contact"), submit_id=delete_id) form_ui.addText(D_("Are you sure you want to remove %s from your contact list?") % contact) return {'xmlui': form_ui.toXml()} def _addContact(self, data, profile): """Add the selected contact @param data (dict) @param profile: %(doc_profile)s @return dict """ contact_jid_s = data[xml_tools.formEscape('contact_jid')] try: contact_jid = jid.JID(contact_jid_s) except (RuntimeError, jid.InvalidFormat, AttributeError): # TODO: replace '\t' by a constant (see tools.xmlui.XMLUI.onFormSubmitted) data['selected_groups'] = data[xml_tools.formEscape('groups_list')].split('\t') options = {'id': self.__add_id, 'title': D_('Add contact'), 'contact_text': D_('Please enter a valid JID (like "contact@%s"):') % self.default_host, } return self.getDialogXMLUI(options, data, profile) self.host.addContact(contact_jid, profile_key=profile) return self._updateContact(data, profile) # after adding, updating def _updateContact(self, data, profile): """Update the selected contact @param data (dict) @param profile: %(doc_profile)s @return dict """ contact_jid = jid.JID(data[xml_tools.formEscape('contact_jid')]) # TODO: replace '\t' by a constant (see tools.xmlui.XMLUI.onFormSubmitted) groups = data[xml_tools.formEscape('groups_list')].split('\t') self.host.updateContact(contact_jid, name='', groups=groups, profile_key=profile) return {} def _deleteContact(self, contact_jid, profile): """Delete the selected contact @param contact_jid (JID) @param profile: %(doc_profile)s @return dict """ self.host.delContact(contact_jid, profile_key=profile) return {}