paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-01 12:33:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

management: Move all_users option to ZulipBaseCommand.

Browse Source
This commit is contained in:
Vishnu Ks
2017-08-25 01:16:13 +05:30
committed by Tim Abbott
parent 9d043e7fb0
commit 2b076ef822
5 changed files with 39 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
zerver/management/commands/add_users_to_streams.py
View File

@@ -15,7 +15,7 @@ class Command(ZulipBaseCommand):
def add_arguments(self, parser):
# type: (CommandParser) -> None
self.add_realm_args(parser, True)
self.add_user_list_args(parser)
self.add_user_list_args(parser, all_users_help="Add all users in realm to these streams.")
parser.add_argument(
'-s', '--streams',
@@ -24,28 +24,12 @@ class Command(ZulipBaseCommand):
required=True,
help='A comma-separated list of stream names.')
parser.add_argument(
'-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):
# type: (**Any) -> None
realm = self.get_realm(options)
user_profiles = self.get_users(options, realm)
if bool(user_profiles) == options["all_users"]:
self.print_help("./manage.py", "add_users_to_streams")
exit(1)
stream_names = set([stream.strip() for stream in options["streams"].split(",")])
# If all_users flag is passed user list should not be passed and vice versa.
if options["all_users"]:
user_profiles = UserProfile.objects.filter(realm=realm)
for stream_name in set(stream_names):
for user_profile in user_profiles:
stream, _ = create_stream_if_needed(realm, stream_name)