mirror of
https://github.com/zulip/zulip.git
synced 2025-11-12 09:58:06 +00:00
0844c9a6ec
Note that a pretty common use case for this is a realm admin sending this to everyone after an import from HipChat or Slack. So this adds the realm_name to the title (so that there is something they might recognize) and kept the wording generic enough to accommodate the user not having clicked anything to get this email. Also strengthens the tests a bit to better test the complicated template logic.
54 lines
2.3 KiB
Python
54 lines
2.3 KiB
Python
|
|
import logging
|
|
from argparse import ArgumentParser
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from django.contrib.auth.tokens import default_token_generator
|
|
|
|
from zerver.forms import generate_password_reset_url
|
|
from zerver.lib.management import CommandError, ZulipBaseCommand
|
|
from zerver.lib.send_email import FromAddress, send_email
|
|
from zerver.models import UserProfile
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = """Send email to specified email address."""
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument('--entire-server', action="store_true", default=False,
|
|
help="Send to every user on the server. ")
|
|
self.add_user_list_args(parser,
|
|
help="Email addresses of user(s) to send password reset emails to.",
|
|
all_users_help="Send to every user on the realm.")
|
|
self.add_realm_args(parser)
|
|
|
|
def handle(self, *args: Any, **options: str) -> None:
|
|
if options["entire_server"]:
|
|
users = UserProfile.objects.filter(is_active=True, is_bot=False,
|
|
is_mirror_dummy=False)
|
|
else:
|
|
realm = self.get_realm(options)
|
|
try:
|
|
users = self.get_users(options, realm)
|
|
except CommandError as error:
|
|
if str(error) == "You have to pass either -u/--users or -a/--all-users.":
|
|
raise CommandError("You have to pass -u/--users or -a/--all-users or --entire-server.")
|
|
raise error
|
|
|
|
self.send(users)
|
|
|
|
def send(self, users: List[UserProfile]) -> None:
|
|
"""Sends one-use only links for resetting password to target users
|
|
|
|
"""
|
|
for user_profile in users:
|
|
context = {
|
|
'email': user_profile.email,
|
|
'reset_url': generate_password_reset_url(user_profile, default_token_generator),
|
|
'realm_uri': user_profile.realm.uri,
|
|
'realm_name': user_profile.realm.name,
|
|
'active_account_in_realm': True,
|
|
}
|
|
send_email('zerver/emails/password_reset', to_user_ids=[user_profile.id],
|
|
from_address=FromAddress.tokenized_no_reply_address(),
|
|
from_name="Zulip Account Security", context=context)
|