Commit b23464c7 by Tyler Hallada

Add no-videos option to fake data, use for Travis

parent e6235e51
......@@ -70,4 +70,4 @@ demo: clean requirements loaddata
travis: clean test.requirements migrate
python manage.py set_api_key edx edx
python manage.py loaddata problem_response_answer_distribution --database=analytics
python manage.py generate_fake_course_data --num-weeks=1
python manage.py generate_fake_course_data --num-weeks=1 --no-videos
......@@ -58,6 +58,13 @@ class Command(BaseCommand):
default='ed_xavier',
help='Username for which to generate fake data',
)
parser.add_argument(
'--no-videos',
action='store_false',
dest='videos',
default=True,
help='Disables pulling video ids from the LMS server to generate fake video data and instead uses fake ids.'
)
def generate_daily_data(self, course_id, start_date, end_date):
# Use the preset ratios below to generate data in the specified demographics
......@@ -308,18 +315,26 @@ class Command(BaseCommand):
logger.info("Done!")
def fake_video_ids_fallback(self):
return [
{
'video_id': '0fac49ba',
'video_module_id': 'i4x-edX-DemoX-video-5c90cffecd9b48b188cbfea176bf7fe9'
}
]
def handle(self, *args, **options):
course_id = options['course_id']
username = options['username']
video_ids = self.fetch_videos_from_course_blocks(course_id)
if not video_ids:
logger.warning("Falling back to fake video id due to Course Blocks API failure...")
video_ids = [
{
'video_id': '0fac49ba',
'video_module_id': 'i4x-edX-DemoX-video-5c90cffecd9b48b188cbfea176bf7fe9'
}
]
videos = options['videos']
if videos:
video_ids = self.fetch_videos_from_course_blocks(course_id)
if not video_ids:
logger.warning("Falling back to fake video id due to Course Blocks API failure...")
video_ids = self.fake_video_ids_fallback()
else:
logger.info("Option to generate videos with ids pulled from the LMS is disabled, using fake video ids...")
video_ids = self.fake_video_ids_fallback()
start_date = timezone.now() - datetime.timedelta(weeks=10)
num_weeks = options['num_weeks']
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment