mirror of
https://github.com/zulip/zulip.git
synced 2025-11-11 01:16:19 +00:00
8e7ce7cc79
This is a preparatory commit for using isort for sorting all of our imports, merging changes to files where we can easily review the changes as something we're happy with. These are also files with relatively little active development, which means we don't expect much merge conflict risk from these changes.
53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any, List
|
|
|
|
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, is_bot=False)
|
|
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.delivery_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)
|