mirror of https://github.com/zulip/zulip.git
56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
|
from optparse import make_option
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
|
||
|
|
from zephyr.models import create_stream_if_needed, do_add_subscription, Realm, \
|
||
|
|
User, UserProfile
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = """Add some or all users in a realm to a set of streams."""
|
||
|
|
|
||
|
|
option_list = BaseCommand.option_list + (
|
||
|
|
make_option('-d', '--domain',
|
||
|
|
dest='domain',
|
||
|
|
type='str',
|
||
|
|
help='The name of the realm in which you are adding people to streams.'),
|
||
|
|
make_option('-s', '--streams',
|
||
|
|
dest='streams',
|
||
|
|
type='str',
|
||
|
|
help='A comma-separated list of stream names.'),
|
||
|
|
make_option('-u', '--users',
|
||
|
|
dest='users',
|
||
|
|
type='str',
|
||
|
|
help='A comma-separated list of email addresses.'),
|
||
|
|
make_option('-a', '--all-users',
|
||
|
|
dest='all_users',
|
||
|
|
action="store_true",
|
||
|
|
default=False,
|
||
|
|
help='Add all users in this realm to these streams.'),
|
||
|
|
)
|
||
|
|
|
||
|
|
def handle(self, **options):
|
||
|
|
if options["domain"] is None or options["streams"] is None or \
|
||
|
|
(options["users"] is None and options["all_users"] is None):
|
||
|
|
self.print_help("python manage.py", "add_users_to_streams")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
stream_names = set([stream.strip() for stream in options["streams"].split(",")])
|
||
|
|
realm = Realm.objects.get(domain=options["domain"])
|
||
|
|
|
||
|
|
if options["all_users"]:
|
||
|
|
user_profiles = UserProfile.objects.filter(realm=realm)
|
||
|
|
else:
|
||
|
|
emails = set([email.strip() for email in options["users"].split(",")])
|
||
|
|
user_profiles = []
|
||
|
|
for email in emails:
|
||
|
|
user_profiles.append(UserProfile.objects.get(
|
||
|
|
user=User.objects.get(email__iexact=email)))
|
||
|
|
|
||
|
|
for stream_name in set(stream_names):
|
||
|
|
for user_profile in user_profiles:
|
||
|
|
stream = create_stream_if_needed(user_profile.realm, stream_name)
|
||
|
|
did_subscribe = do_add_subscription(user_profile, stream)
|
||
|
|
print "%s %s to %s" % (
|
||
|
|
"Subscribed" if did_subscribe else "Already subscribed",
|
||
|
|
user_profile.user.email, stream_name)
|