Adding a better command for assigning roles.

lms/djangoapps/django_comment_client/management/commands/assign_roles_for_course.py

@@ -0,0 +1,27 @@
+"""
+This must be run only after seed_permissions_roles.py!
+
+Creates default roles for all users in the provided course. Just runs through
+Enrollments.
+"""
+from django.core.management.base import BaseCommand, CommandError
+
+from student.models import CourseEnrollment, assign_default_role
+
+class Command(BaseCommand):
+    args = 'course_id'
+    help = 'Add roles for all users in a course'
+
+    def handle(self, *args, **options):
+        if len(args) == 0:
+            raise CommandError("Please provide a course id")
+        if len(args) > 1:
+            raise CommandError("Too many arguments")
+        course_id = args[0]
+
+        print "Updated roles for ",
+        for i, enrollment in enumerate(CourseEnrollment.objects.filter(course_id=course_id), start=1):
+            assign_default_role(None, enrollment)
+            if i % 1000 == 0:
+                print "{0}...".format(i),
+        print