mirror of https://github.com/zulip/zulip.git
58 lines
2.6 KiB
Python
58 lines
2.6 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from zerver.lib.management import CommandError, ZulipBaseCommand
|
|
from zerver.lib.send_email import send_custom_email
|
|
from zerver.models import UserProfile
|
|
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = """
|
|
Send a custom email with Zulip branding to the specified users.
|
|
|
|
Useful to send a notice to all users of a realm or server.
|
|
|
|
The From and Subject headers can be provided in the body of the markdown
|
|
document used to generate the email, or on the command line."""
|
|
|
|
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. ")
|
|
parser.add_argument('--markdown-template-path', '--path',
|
|
dest='markdown_template_path',
|
|
required=True,
|
|
type=str,
|
|
help='Path to a markdown-format body for the email.')
|
|
parser.add_argument('--subject',
|
|
type=str,
|
|
help='Subject for the email. It can be declarated in markdown file in headers')
|
|
parser.add_argument('--from-name',
|
|
type=str,
|
|
help='From line for the email. It can be declarated in markdown file in headers')
|
|
parser.add_argument('--reply-to',
|
|
type=str,
|
|
help='Optional reply-to line for the email')
|
|
parser.add_argument('--admins-only',
|
|
help='Send only to organization administrators',
|
|
action='store_true')
|
|
|
|
self.add_user_list_args(parser,
|
|
help="Email addresses of user(s) to send 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
|
|
|
|
send_custom_email(users, options)
|