39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
##
|
|
## A script to create some dummy users
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from student.models import UserProfile, 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.objects.create(user=user, course_id=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)
|