34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
from afew.filters.BaseFilter import Filter
|
||
|
|
from afew.FilterRegistry import register_filter
|
||
|
|
|
||
|
|
from khard.config import Config
|
||
|
|
|
||
|
|
import re
|
||
|
|
|
||
|
|
@register_filter
|
||
|
|
class TagOrContact(Filter):
|
||
|
|
message = 'Allow only tagged messages from non-contacts'
|
||
|
|
user = None
|
||
|
|
domain = None
|
||
|
|
|
||
|
|
def __init__(self, database, **kwargs):
|
||
|
|
super(TagOrContact, self).__init__(database, **kwargs)
|
||
|
|
self._extract_tag_re = re.compile(f'[^+]+(\+(?P<tag>[^@]+))?@[^@]+')
|
||
|
|
self._bare_email_re = re.compile(r'[^<]*<(?P<email>[^@<>]+@[^@<>]+)>')
|
||
|
|
self._contactbook = Config().abook
|
||
|
|
|
||
|
|
def handle_message(self, message):
|
||
|
|
recipient = message.get_header('Delivered-To')
|
||
|
|
match = self._extract_tag_re.fullmatch(recipient)
|
||
|
|
if match is None: return # wrong email
|
||
|
|
|
||
|
|
if match.group('tag'):
|
||
|
|
self.add_tags(message, match['tag'])
|
||
|
|
else:
|
||
|
|
sender = message.get_header('From')
|
||
|
|
if '<' in sender: sender = self._bare_email_re.fullmatch(sender)['email']
|
||
|
|
if self._contactbook.search(sender):
|
||
|
|
self.add_tags(message, 'contact')
|
||
|
|
else:
|
||
|
|
self.add_tags(message, 'killed')
|