mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-03 21:43:21 +00:00 
			
		
		
		
	This commit is a preparatory step for allowing organization owners to reset user preferences, refactors the `clear_scheduled_emails` function to support bulk operations.
		
			
				
	
	
		
			81 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			81 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from collections.abc import Callable
 | 
						|
 | 
						|
from django.http import HttpRequest, HttpResponse
 | 
						|
from django.shortcuts import render
 | 
						|
from django.views.decorators.csrf import csrf_exempt
 | 
						|
 | 
						|
from confirmation.models import Confirmation, ConfirmationKeyError, get_object_from_key
 | 
						|
from zerver.actions.user_settings import do_change_user_setting
 | 
						|
from zerver.context_processors import common_context
 | 
						|
from zerver.lib.send_email import clear_scheduled_emails
 | 
						|
from zerver.models import ScheduledEmail, UserProfile
 | 
						|
 | 
						|
 | 
						|
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], mark_object_used=False
 | 
						|
        )
 | 
						|
    except ConfirmationKeyError:
 | 
						|
        return render(request, "zerver/unsubscribe_link_error.html")
 | 
						|
 | 
						|
    assert isinstance(user_profile, UserProfile)
 | 
						|
    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_user_setting(
 | 
						|
        user_profile, "enable_offline_email_notifications", False, acting_user=user_profile
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
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_user_setting(user_profile, "enable_digest_emails", False, acting_user=user_profile)
 | 
						|
 | 
						|
 | 
						|
def do_login_unsubscribe(user_profile: UserProfile) -> None:
 | 
						|
    do_change_user_setting(user_profile, "enable_login_emails", False, acting_user=user_profile)
 | 
						|
 | 
						|
 | 
						|
def do_marketing_unsubscribe(user_profile: UserProfile) -> None:
 | 
						|
    do_change_user_setting(user_profile, "enable_marketing_emails", False, acting_user=user_profile)
 | 
						|
 | 
						|
 | 
						|
# 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": ("login", do_login_unsubscribe),
 | 
						|
    "marketing": ("marketing", do_marketing_unsubscribe),
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
# Login NOT required. These are for one-click unsubscribes.
 | 
						|
@csrf_exempt
 | 
						|
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")
 |