|
4 | 4 | import logging |
5 | 5 | import math |
6 | 6 | import random |
7 | | -from optparse import make_option |
8 | 7 | from tqdm import tqdm |
9 | 8 |
|
10 | 9 | from django.core.management.base import BaseCommand |
@@ -32,14 +31,29 @@ def get_count(start): |
32 | 31 |
|
33 | 32 | class Command(BaseCommand): |
34 | 33 | help = 'Generate fake data' |
35 | | - option_list = BaseCommand.option_list + ( |
36 | | - make_option('-n', '--num-weeks', action='store', type="int", dest='num_weeks', |
37 | | - help='Number of weeks worth of data to generate.'), |
38 | | - make_option('-c', '--course_id', action='store', type='string', dest='course_id', |
39 | | - default='edX/DemoX/Demo_Course', help='Course ID for which to generate fake data'), |
40 | | - make_option('-u', '--username', action='store', type='string', dest='username', |
41 | | - default='ed_xavier', help='Username for which to generate fake data'), |
42 | | - ) |
| 34 | + |
| 35 | + def add_arguments(self, parser): |
| 36 | + parser.add_argument( |
| 37 | + '--num-weeks', |
| 38 | + action='store', |
| 39 | + type=int, |
| 40 | + dest='num_weeks', |
| 41 | + help='Number of weeks worth of data to generate.', |
| 42 | + ) |
| 43 | + parser.add_argument( |
| 44 | + '--course_id', |
| 45 | + action='store', |
| 46 | + dest='course_id', |
| 47 | + default='edX/DemoX/Demo_Course', |
| 48 | + help='Course ID for which to generate fake data', |
| 49 | + ) |
| 50 | + parser.add_argument( |
| 51 | + '--username', |
| 52 | + action='store', |
| 53 | + dest='username', |
| 54 | + default='ed_xavier', |
| 55 | + help='Username for which to generate fake data', |
| 56 | + ) |
43 | 57 |
|
44 | 58 | def generate_daily_data(self, course_id, start_date, end_date): |
45 | 59 | # Use the preset ratios below to generate data in the specified demographics |
|
0 commit comments