mirror of https://github.com/zulip/zulip.git
60 lines
2.5 KiB
Python
60 lines
2.5 KiB
Python
from __future__ import absolute_import
|
|
|
|
import logging
|
|
from typing import Any, Dict, List, Optional, Text
|
|
|
|
from argparse import ArgumentParser
|
|
from zerver.models import UserProfile
|
|
from django.utils.http import urlsafe_base64_encode
|
|
from django.utils.encoding import force_bytes
|
|
|
|
from django.contrib.auth.tokens import default_token_generator, PasswordResetTokenGenerator
|
|
|
|
from zerver.lib.send_email import send_email, FromAddress
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = """Send email to specified email address."""
|
|
|
|
def add_arguments(self, parser):
|
|
# type: (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.")
|
|
self.add_realm_args(parser)
|
|
|
|
def handle(self, *args, **options):
|
|
# type: (*Any, **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)
|
|
users = self.get_users(options, realm)
|
|
|
|
self.send(users)
|
|
|
|
def send(self, users, subject_template_name='', email_template_name='',
|
|
use_https=True, token_generator=default_token_generator,
|
|
from_email=None, html_email_template_name=None):
|
|
# type: (List[UserProfile], str, str, bool, PasswordResetTokenGenerator, Optional[Text], Optional[str]) -> None
|
|
"""Sends one-use only links for resetting password to target users
|
|
|
|
"""
|
|
for user_profile in users:
|
|
context = {
|
|
'email': user_profile.email,
|
|
'domain': user_profile.realm.host,
|
|
'site_name': "zulipo",
|
|
'uid': urlsafe_base64_encode(force_bytes(user_profile.pk)),
|
|
'user': user_profile,
|
|
'token': token_generator.make_token(user_profile),
|
|
'protocol': 'https' if use_https else 'http',
|
|
}
|
|
|
|
logging.warning("Sending %s email to %s" % (email_template_name, user_profile.email,))
|
|
send_email('zerver/emails/password_reset', to_user_id=user_profile.id,
|
|
from_name="Zulip Account Security", from_address=FromAddress.NOREPLY,
|
|
context=context)
|