3ce87583ab
Features coming down the pipe will want to be able to: * Refer to enrollments before they are actually activated (approval step). * See what courses a user used to be enrolled in for when they re-enroll in the same course, or a different run of that course. * Have different "modes" of enrolling in a course, representing things like honor certificate enrollment, auditing (no certs), etc. This change adds an is_active flag and mode (with default being "honor"). The commit is only as large as it is because many parts of the codebase were manipulating enrollments by adding and removing CourseEnrollment objects directly. It was necessary to create classmethods on CourseEnrollment to encapsulate this functionality and then port everything over to using them. The migration to add columns has been tested on a prod replica, and seems to be fine for running on a live system with single digit millions of rows of enrollments.
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
##
|
|
## A script to create some dummy users
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from student.models import CourseEnrollment
|
|
|
|
from student.views import _do_create_account, get_random_post_override
|
|
|
|
|
|
def create(n, course_id):
|
|
"""Create n users, enrolling them in course_id if it's not None"""
|
|
for i in range(n):
|
|
(user, user_profile, _) = _do_create_account(get_random_post_override())
|
|
if course_id is not None:
|
|
CourseEnrollment.enroll(user, course_id)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """Create N new users, with random parameters.
|
|
|
|
Usage: create_random_users.py N [course_id_to_enroll_in].
|
|
|
|
Examples:
|
|
create_random_users.py 1
|
|
create_random_users.py 10 MITx/6.002x/2012_Fall
|
|
create_random_users.py 100 HarvardX/CS50x/2012
|
|
"""
|
|
|
|
def handle(self, *args, **options):
|
|
if len(args) < 1 or len(args) > 2:
|
|
print Command.help
|
|
return
|
|
|
|
n = int(args[0])
|
|
course_id = args[1] if len(args) == 2 else None
|
|
create(n, course_id)
|