management: add a script to add users in a domain to streams.

(imported from commit e234fe4d283882a5cb2e7a57bcaf8939d973b128)

zephyr/management/commands/add_users_to_streams.py

@@ -0,0 +1,55 @@
+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)