mirror of
https://github.com/zulip/zulip.git
synced 2025-11-05 06:23:38 +00:00
365fe0b3d5
Fixes #2665. Regenerated by tabbott with `lint --fix` after a rebase and change in parameters. Note from tabbott: In a few cases, this converts technical debt in the form of unsorted imports into different technical debt in the form of our largest files having very long, ugly import sequences at the start. I expect this change will increase pressure for us to split those files, which isn't a bad thing. Signed-off-by: Anders Kaseorg <anders@zulip.com>
59 lines
2.4 KiB
Python
59 lines
2.4 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from django.core.management.base import CommandError
|
|
|
|
from confirmation.models import Confirmation, create_confirmation_link
|
|
from zerver.lib.email_validation import email_allowed_for_realm
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
from zerver.models import DomainNotAllowedForRealmError, PreregistrationUser
|
|
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = "Generate activation links for users and print them to stdout."
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument('--force',
|
|
dest='force',
|
|
action="store_true",
|
|
default=False,
|
|
help='Override that the domain is restricted to external users.')
|
|
parser.add_argument('emails', metavar='<email>', type=str, nargs='*',
|
|
help='email of users to generate an activation link for')
|
|
self.add_realm_args(parser, True)
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
duplicates = False
|
|
realm = self.get_realm(options)
|
|
assert realm is not None # Should be ensured by parser
|
|
|
|
if not options['emails']:
|
|
self.print_help("./manage.py", "generate_invite_links")
|
|
raise CommandError
|
|
|
|
for email in options['emails']:
|
|
try:
|
|
self.get_user(email, realm)
|
|
print(email + ": There is already a user registered with that address.")
|
|
duplicates = True
|
|
continue
|
|
except CommandError:
|
|
pass
|
|
|
|
if duplicates:
|
|
return
|
|
|
|
for email in options['emails']:
|
|
try:
|
|
email_allowed_for_realm(email, realm)
|
|
except DomainNotAllowedForRealmError:
|
|
if not options["force"]:
|
|
raise CommandError("You've asked to add an external user '{}' to a "
|
|
"closed realm '{}'.\nAre you sure? To do this, "
|
|
"pass --force.".format(email, realm.string_id))
|
|
|
|
prereg_user = PreregistrationUser(email=email, realm=realm)
|
|
prereg_user.save()
|
|
print(email + ": " + create_confirmation_link(prereg_user, realm.host,
|
|
Confirmation.INVITATION))
|