mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
9735025167
Add a function email_allowed_for_realm that checks whether a user with given email is allowed to join a given realm (either because the email has the right domain, or because the realm is open), and use it whenever deciding whether to allow adding a user to a realm. This commit is not intended to change any behavior, except in one case where the Zulip realm's domain was not being converted to lowercase.
62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from confirmation.models import Confirmation
|
|
from zerver.models import UserProfile, PreregistrationUser, \
|
|
get_user_profile_by_email, get_realm, email_allowed_for_realm
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate activation links for users and print them to stdout."
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--domain',
|
|
dest='domain',
|
|
type=str,
|
|
help='The realm in which to generate the invites (use for open realms).')
|
|
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 user to generate an activation link for')
|
|
|
|
def handle(self, *args, **options):
|
|
duplicates = False
|
|
for email in options['emails']:
|
|
try:
|
|
get_user_profile_by_email(email)
|
|
print(email + ": There is already a user registered with that address.")
|
|
duplicates = True
|
|
continue
|
|
except UserProfile.DoesNotExist:
|
|
pass
|
|
|
|
if duplicates:
|
|
return
|
|
|
|
realm = None
|
|
domain = options["domain"]
|
|
if domain:
|
|
realm = get_realm(domain)
|
|
if not realm:
|
|
print("The realm %s doesn't exist yet, please create it first." % (domain,))
|
|
print("Don't forget default streams!")
|
|
exit(1)
|
|
|
|
for email in options['emails']:
|
|
if realm:
|
|
if not email_allowed_for_realm(email, realm) and not options["force"]:
|
|
print("You've asked to add an external user (%s) to a closed realm (%s)." % (
|
|
email, domain))
|
|
print("Are you sure? To do this, pass --force.")
|
|
exit(1)
|
|
else:
|
|
prereg_user = PreregistrationUser(email=email, realm=realm)
|
|
else:
|
|
prereg_user = PreregistrationUser(email=email)
|
|
prereg_user.save()
|
|
print(email + ": " + Confirmation.objects.get_link_for_object(prereg_user))
|
|
|