mirror of
https://github.com/zulip/zulip.git
synced 2025-11-06 15:03:34 +00:00
8e7ce7cc79
This is a preparatory commit for using isort for sorting all of our imports, merging changes to files where we can easily review the changes as something we're happy with. These are also files with relatively little active development, which means we don't expect much merge conflict risk from these changes.
59 lines
2.3 KiB
Python
59 lines
2.3 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.management import ZulipBaseCommand
|
|
from zerver.models import DomainNotAllowedForRealmError, PreregistrationUser, \
|
|
email_allowed_for_realm
|
|
|
|
|
|
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))
|