mattdrayer/progress-generator-no-zero-entries: Only generate when completions > 0

lms/djangoapps/progress/management/commands/generate_progress_entries.py

@@ -6,7 +6,7 @@ from optparse import make_option
 
 from django.core.management.base import BaseCommand
 from django.conf import settings
-from django.db.models import Q
+from django.db.models import Q, Min
 
 from progress.models import StudentProgress, CourseModuleCompletion
 from student.models import CourseEnrollment
@@ -69,23 +69,45 @@ class Command(BaseCommand):
 
             # For each user...
             for user in users:
+
                 status = 'skipped'
-                completions = CourseModuleCompletion.objects.filter(course_id=course.id, user_id=user.id)\
+                num_completions = CourseModuleCompletion.objects.filter(course_id=course.id, user_id=user.id)\
                     .exclude(cat_list).count()
+
+                if num_completions:
+
+                    start_date = CourseModuleCompletion.objects.filter(course_id=course.id, user_id=user.id)\
+                        .exclude(cat_list).aggregate(Min('created'))['created__min']
+
                     try:
                         existing_record = StudentProgress.objects.get(user=user, course_id=course.id)
-                    if existing_record.completions != completions:
-                        existing_record.completions = completions
-                        existing_record.save()
+
+                        if existing_record.completions != num_completions:
+                            existing_record.completions = num_completions
+                            status = 'updated'
+
+                        if existing_record.created != start_date:
+                            existing_record.created = start_date
                             status = 'updated'
+
+                        if status == 'updated':
+                            existing_record.save()
+
                     except StudentProgress.DoesNotExist:
-                    new_record = StudentProgress.objects.create(
+                        StudentProgress.objects.create(
                             user=user,
                             course_id=course.id,
-                        completions=completions
+                            completions=num_completions,
+                            created=start_date
                         )
                         status = 'created'
-                log_msg = 'Progress entry {} -- Course: {}, User: {}  (completions: {})'.format(status, course.id, user.id
-                                                                                                     , completions)
+
+                log_msg = 'Progress entry {} -- Course: {}, User: {}  (completions: {})'.format(
+                    status,
+                    course.id,
+                    user.id,
+                    num_completions
+                )
+
                 print log_msg
                 log.info(log_msg)