upstream-zulip/zerver/management/commands/send_welcome_bot_message.py

from argparse import ArgumentParser
from typing import Any

from typing_extensions import override

from zerver.lib.management import ZulipBaseCommand
from zerver.lib.onboarding import send_initial_direct_messages_to_user


class Command(ZulipBaseCommand):
    help = """Sends the initial welcome bot message."""

    @override
    def add_arguments(self, parser: ArgumentParser) -> None:
        self.add_user_list_args(
            parser,
            help="Email addresses of user(s) to send welcome bot messages to.",
            all_users_help="Send to every user on the realm.",
        )
        self.add_realm_args(parser)

    @override
    def handle(self, *args: Any, **options: str) -> None:
        for user_profile in self.get_users(options, self.get_realm(options), is_bot=False):
            send_initial_direct_messages_to_user(user_profile)