mirror of https://github.com/zulip/zulip.git
63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
from __future__ import absolute_import
|
|
|
|
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
|
|
|
|
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 realm.restricted_to_domain and \
|
|
domain.lower() != email.split("@", 1)[-1].lower() 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)
|
|
|