diff --git a/lms/djangoapps/django_comment_client/management/commands/create_roles_for_existing.py b/lms/djangoapps/django_comment_client/management/commands/create_roles_for_existing.py new file mode 100644 index 0000000000..148c204acf --- /dev/null +++ b/lms/djangoapps/django_comment_client/management/commands/create_roles_for_existing.py @@ -0,0 +1,26 @@ +""" +This must be run only after seed_permissions_roles.py! + +Creates default roles for all users currently in the database. Just runs through +Enrollments. +""" +from django.core.management.base import BaseCommand, CommandError + +from student.models import CourseEnrollment +from django_comment_client.permissions import assign_default_role + + +class Command(BaseCommand): + args = 'course_id' + help = 'Seed default permisssions and roles' + + def handle(self, *args, **options): + if len(args) != 0: + raise CommandError("This Command takes no arguments") + + print "Updated roles for ", + for i, enrollment in enumerate(CourseEnrollment.objects.all(), start=1): + assign_default_role(None, enrollment) + if i % 1000 == 0: + print "{0}...".format(i), + print \ No newline at end of file