mirror of https://github.com/zulip/zulip.git
notifications.py: Replace get_user_profile_by_email.
Replace with get_user.
This commit is contained in:
parent
ac3e9256bb
commit
ba4ea7dd8a
|
@ -17,7 +17,7 @@ from zerver.models import (
|
|||
Stream,
|
||||
get_display_recipient,
|
||||
UserProfile,
|
||||
get_user_profile_by_email,
|
||||
get_user,
|
||||
get_user_profile_by_id,
|
||||
receives_offline_notifications,
|
||||
get_context_for_message,
|
||||
|
@ -400,8 +400,8 @@ def log_digest_event(msg):
|
|||
logging.basicConfig(filename=settings.DIGEST_LOG_PATH, level=logging.INFO)
|
||||
logging.info(msg)
|
||||
|
||||
def enqueue_welcome_emails(email, name):
|
||||
# type: (Text, Text) -> None
|
||||
def enqueue_welcome_emails(email, name, realm):
|
||||
# type: (Text, Text, Realm) -> None
|
||||
from zerver.context_processors import common_context
|
||||
if settings.WELCOME_EMAIL_SENDER is not None:
|
||||
# line break to avoid triggering lint rule
|
||||
|
@ -411,7 +411,7 @@ def enqueue_welcome_emails(email, name):
|
|||
from_name = None
|
||||
from_address = FromAddress.SUPPORT
|
||||
|
||||
user_profile = get_user_profile_by_email(email)
|
||||
user_profile = get_user(email, realm)
|
||||
unsubscribe_link = one_click_unsubscribe_link(user_profile, "welcome")
|
||||
context = common_context(user_profile)
|
||||
context.update({
|
||||
|
|
|
@ -787,9 +787,9 @@ class EmailUnsubscribeTests(ZulipTestCase):
|
|||
"""
|
||||
email = self.example_email("hamlet")
|
||||
user_profile = self.example_user('hamlet')
|
||||
|
||||
realm = get_realm("zulip")
|
||||
# Simulate a new user signing up, which enqueues 2 welcome e-mails.
|
||||
enqueue_welcome_emails(email, "King Hamlet")
|
||||
enqueue_welcome_emails(email, "King Hamlet", realm)
|
||||
self.assertEqual(2, len(ScheduledJob.objects.filter(
|
||||
type=ScheduledJob.EMAIL, filter_string__iexact=email)))
|
||||
|
||||
|
|
|
@ -157,6 +157,7 @@ class SignupWorker(QueueProcessingWorker):
|
|||
# type: (Mapping[str, Any]) -> None
|
||||
# This should clear out any invitation reminder emails
|
||||
clear_followup_emails_queue(data['email_address'])
|
||||
realm = Realm.objects.get(id=data['merge_fields']['REALM_ID'])
|
||||
if settings.MAILCHIMP_API_KEY and settings.PRODUCTION:
|
||||
endpoint = "https://%s.api.mailchimp.com/3.0/lists/%s/members" % \
|
||||
(settings.MAILCHIMP_API_KEY.split('-')[1], settings.ZULIP_FRIENDS_LIST_ID)
|
||||
|
@ -170,7 +171,7 @@ class SignupWorker(QueueProcessingWorker):
|
|||
else:
|
||||
r.raise_for_status()
|
||||
|
||||
enqueue_welcome_emails(data['email_address'], data['merge_fields']['NAME'])
|
||||
enqueue_welcome_emails(data['email_address'], data['merge_fields']['NAME'], realm)
|
||||
|
||||
@assign_queue('invites')
|
||||
class ConfirmationEmailWorker(QueueProcessingWorker):
|
||||
|
|
Loading…
Reference in New Issue