Afew Filter: Auto-tag thread filter

autotag_filter.py

#!/usr/bin/env python
# coding=utf-8

from __future__ import print_function, absolute_import, unicode_literals

from afew.filters.BaseFilter import Filter
from afew.FilterRegistry import register_filter


@register_filter
class AutoTagThreadFilter(Filter):
    message = 'Auto tag mails based on thread'
    query = ''

    def handle_message(self, message):
        try:
            tags = set()
            if message.get_thread_id():
                for message in self.database.get_thread(message.get_thread_id()):
                    tags.update(message.get_tags())
            if len(tags) != 0:
                self.add_tags(message, *tags)
        except Exception as err:
            self.log.exception("Skipping message with m-id: `{}`".format(message.get_header('Message-ID').encode('utf-8')))