mirror of https://github.com/zulip/zulip.git
55 lines
2.5 KiB
Python
55 lines
2.5 KiB
Python
|
|
from django.conf import settings
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.shortcuts import render
|
|
from typing import Callable
|
|
|
|
from confirmation.models import Confirmation, get_object_from_key, \
|
|
render_confirmation_key_error, ConfirmationKeyException
|
|
from zerver.lib.actions import do_change_notification_settings, clear_scheduled_emails
|
|
from zerver.models import UserProfile, ScheduledEmail
|
|
from zerver.context_processors import common_context
|
|
|
|
def process_unsubscribe(request: HttpRequest, confirmation_key: str, subscription_type: str,
|
|
unsubscribe_function: Callable[[UserProfile], None]) -> HttpResponse:
|
|
try:
|
|
user_profile = get_object_from_key(confirmation_key, Confirmation.UNSUBSCRIBE)
|
|
except ConfirmationKeyException as exception:
|
|
return render_confirmation_key_error(request, exception)
|
|
|
|
unsubscribe_function(user_profile)
|
|
context = common_context(user_profile)
|
|
context.update({"subscription_type": subscription_type})
|
|
return render(request, 'zerver/unsubscribe_success.html', context=context)
|
|
|
|
# Email unsubscribe functions. All have the function signature
|
|
# processor(user_profile).
|
|
|
|
def do_missedmessage_unsubscribe(user_profile: UserProfile) -> None:
|
|
do_change_notification_settings(user_profile, 'enable_offline_email_notifications', False)
|
|
|
|
def do_welcome_unsubscribe(user_profile: UserProfile) -> None:
|
|
clear_scheduled_emails(user_profile.id, ScheduledEmail.WELCOME)
|
|
|
|
def do_digest_unsubscribe(user_profile: UserProfile) -> None:
|
|
do_change_notification_settings(user_profile, 'enable_digest_emails', False)
|
|
|
|
# The keys are part of the URL for the unsubscribe link and must be valid
|
|
# without encoding.
|
|
# The values are a tuple of (display name, unsubscribe function), where the
|
|
# display name is what we call this class of email in user-visible text.
|
|
email_unsubscribers = {
|
|
"missed_messages": ("missed messages", do_missedmessage_unsubscribe),
|
|
"welcome": ("welcome", do_welcome_unsubscribe),
|
|
"digest": ("digest", do_digest_unsubscribe)
|
|
}
|
|
|
|
# Login NOT required. These are for one-click unsubscribes.
|
|
def email_unsubscribe(request: HttpRequest, email_type: str,
|
|
confirmation_key: str) -> HttpResponse:
|
|
if email_type in email_unsubscribers:
|
|
display_name, unsubscribe_function = email_unsubscribers[email_type]
|
|
return process_unsubscribe(request, confirmation_key, display_name, unsubscribe_function)
|
|
|
|
return render(request, 'zerver/unsubscribe_link_error.html')
|