35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
|
from email.message import Message
|
||
|
|
|
||
|
|
from app.email import headers
|
||
|
|
from app.email_utils import add_or_replace_header
|
||
|
|
from app.handler.unsubscribe_encoder import (
|
||
|
|
UnsubscribeEncoder,
|
||
|
|
UnsubscribeAction,
|
||
|
|
)
|
||
|
|
from app.models import Alias, Contact
|
||
|
|
|
||
|
|
|
||
|
|
class UnsubscribeGenerator:
|
||
|
|
def add_header_to_message(
|
||
|
|
self, alias: Alias, contact: Contact, message: Message
|
||
|
|
) -> Message:
|
||
|
|
"""
|
||
|
|
Add List-Unsubscribe header
|
||
|
|
"""
|
||
|
|
user = alias.user
|
||
|
|
if user.one_click_unsubscribe_block_sender:
|
||
|
|
unsub_link = UnsubscribeEncoder.encode(
|
||
|
|
UnsubscribeAction.DisableContact, contact.id
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
unsub_link = UnsubscribeEncoder.encode(
|
||
|
|
UnsubscribeAction.DisableAlias, alias.id
|
||
|
|
)
|
||
|
|
|
||
|
|
add_or_replace_header(message, headers.LIST_UNSUBSCRIBE, f"<{unsub_link.link}>")
|
||
|
|
if not unsub_link.via_email:
|
||
|
|
add_or_replace_header(
|
||
|
|
message, headers.LIST_UNSUBSCRIBE_POST, "List-Unsubscribe=One-Click"
|
||
|
|
)
|
||
|
|
return message
|