server-tools/fetchmail_attach_from_folder/match_algorithm/email_domain.py

# -*- coding: utf-8 -*-
# Copyright - 2013-2018 Therp BV <https://therp.nl>.
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from .email_exact import EmailExact


class EmailDomain(EmailExact):
    """Search objects by domain name of email address.

    Beware of match_first here, this is most likely to get it wrong (gmail).
    """
    name = 'Domain of email address'

    def search_matches(self, folder, mail_message, mail_message_org):
        """Returns recordset of matching objects."""
        matches = super(EmailDomain, self).search_matches(
            folder, mail_message, mail_message_org)
        if not matches:
            object_model = folder.env[folder.model_id.model]
            domains = []
            for addr in self._get_mailaddresses(folder, mail_message):
                domains.append(addr.split('@')[-1])
            matches = object_model.search(
                self._get_mailaddress_search_domain(
                    folder, mail_message,
                    operator='like',
                    values=['%@' + domain for domain in set(domains)]),
                order=folder.model_order)
        return matches