zulip/zilencer/management/commands/add_new_user.py

from typing import Any

from django.core.management.base import CommandParser

from zerver.lib.actions import do_create_user
from zerver.lib.management import ZulipBaseCommand
from zerver.models import Realm, UserProfile


class Command(ZulipBaseCommand):
    help = """Add a new user for manual testing of the onboarding process.
If realm is unspecified, will try to use a realm created by add_new_realm,
and will otherwise fall back to the zulip realm."""

    def add_arguments(self, parser: CommandParser) -> None:
        self.add_realm_args(parser)

    def handle(self, **options: Any) -> None:
        realm = self.get_realm(options)
        if realm is None:
            realm = (
                Realm.objects.filter(string_id__startswith='realm').order_by('-string_id').first()
            )
        if realm is None:
            print(
                'Warning: Using default zulip realm, which has an unusual configuration.\n'
                'Try running `manage.py add_new_realm`, and then running this again.'
            )
            valid_realm = Realm.objects.get(string_id='zulip')
            domain = 'zulip.com'
        else:
            valid_realm = realm
            domain = realm.string_id + '.zulip.com'

        name = '{:02}-user'.format(UserProfile.objects.filter(email__contains='user@').count())
        do_create_user(f'{name}@{domain}', 'password', valid_realm, name, acting_user=None)