Пример #1
0
 def test_general_range(self):
     start_date = datetime.datetime(2016, 1, 1)
     end_date = datetime.datetime(2016, 1, 5)
     self.assertEqual(list(date_range(start_date, end_date)), [
         datetime.datetime(2016, 1, 1),
         datetime.datetime(2016, 1, 2),
         datetime.datetime(2016, 1, 3),
         datetime.datetime(2016, 1, 4),
     ])
 def test_general_range(self):
     start_date = datetime.datetime(2016, 1, 1)
     end_date = datetime.datetime(2016, 1, 5)
     self.assertEqual([date for date in date_range(start_date, end_date)], [
         datetime.datetime(2016, 1, 1),
         datetime.datetime(2016, 1, 2),
         datetime.datetime(2016, 1, 3),
         datetime.datetime(2016, 1, 4),
     ])
Пример #3
0
    def get_timeline(self, course_id, username):
        queryset = ModuleEngagement.objects.all().filter(course_id=course_id, username=username) \
            .values('date', 'entity_type', 'event') \
            .annotate(total_count=Sum('count')) \
            .annotate(distinct_entity_count=Count('entity_id', distinct=True)) \
            .order_by('date')

        timelines = []

        for date, engagements in groupby(queryset, lambda x: (x['date'])):
            # Iterate over engagements for this day and create a single day with
            # engagement data.
            day = {
                u'date': date,
            }
            for engagement in engagements:
                engagement_type = EngagementType(engagement['entity_type'],
                                                 engagement['event'])

                if engagement_type.is_counted_by_entity:
                    count_delta = engagement['distinct_entity_count']
                else:
                    count_delta = engagement['total_count']

                day[engagement_type.name] = day.get(engagement_type.name,
                                                    0) + count_delta
            timelines.append(day)

        # Fill in dates that may be missing, since the result store doesn't
        # store empty engagement entries.
        full_timeline = []
        default_timeline_entry = {
            engagement_type: 0
            for engagement_type in EngagementType.ALL_TYPES
        }
        for index, current_date in enumerate(timelines):
            full_timeline.append(current_date)
            try:
                next_date = timelines[index + 1]
            except IndexError:
                continue
            one_day = datetime.timedelta(days=1)
            if next_date['date'] > current_date['date'] + one_day:
                full_timeline += [
                    dict(date=date, **default_timeline_entry)
                    for date in date_range(current_date['date'] +
                                           one_day, next_date['date'])
                ]

        return full_timeline
    def get_timeline(self, course_id, username):
        queryset = ModuleEngagement.objects.all().filter(course_id=course_id, username=username) \
            .values('date', 'entity_type', 'event') \
            .annotate(total_count=Sum('count')) \
            .annotate(distinct_entity_count=Count('entity_id', distinct=True)) \
            .order_by('date')

        timelines = []

        for date, engagements in groupby(queryset, lambda x: (x['date'])):
            # Iterate over engagements for this day and create a single day with
            # engagement data.
            day = {
                u'date': date,
            }
            for engagement in engagements:
                engagement_type = EngagementType(engagement['entity_type'], engagement['event'])

                if engagement_type.is_counted_by_entity:
                    count_delta = engagement['distinct_entity_count']
                else:
                    count_delta = engagement['total_count']

                day[engagement_type.name] = day.get(engagement_type.name, 0) + count_delta
            timelines.append(day)

        # Fill in dates that may be missing, since the result store doesn't
        # store empty engagement entries.
        full_timeline = []
        default_timeline_entry = {engagement_type: 0 for engagement_type in EngagementType.ALL_TYPES}
        for index, current_date in enumerate(timelines):
            full_timeline.append(current_date)
            try:
                next_date = timelines[index + 1]
            except IndexError:
                continue
            one_day = datetime.timedelta(days=1)
            if next_date['date'] > current_date['date'] + one_day:
                full_timeline += [
                    dict(date=date, **default_timeline_entry)
                    for date in date_range(current_date['date'] + one_day, next_date['date'])
                ]

        return full_timeline
Пример #5
0
 def test_empty_range(self):
     same_date = datetime.datetime(2016, 1, 1)
     self.assertEqual(list(date_range(same_date, same_date)), [])
Пример #6
0
 def test_delta_goes_past_end_date(self):
     start_date = datetime.datetime(2016, 1, 1)
     end_date = datetime.datetime(2016, 1, 3)
     time_delta = datetime.timedelta(days=5)
     self.assertEqual(list(date_range(start_date, end_date, time_delta)),
                      [start_date])
Пример #7
0
 def test_range_exclusive(self):
     start_date = datetime.datetime(2016, 1, 1)
     end_date = datetime.datetime(2016, 1, 2)
     self.assertEqual(list(date_range(start_date, end_date)), [start_date])
Пример #8
0
 def test_empty_range(self):
     date = datetime.datetime(2016, 1, 1)
     self.assertEqual([date for date in date_range(date, date)], [])
 def test_delta_goes_past_end_date(self):
     start_date = datetime.datetime(2016, 1, 1)
     end_date = datetime.datetime(2016, 1, 3)
     time_delta = datetime.timedelta(days=5)
     self.assertEqual([date for date in date_range(start_date, end_date, time_delta)], [start_date])
Пример #10
0
 def test_range_exclusive(self):
     start_date = datetime.datetime(2016, 1, 1)
     end_date = datetime.datetime(2016, 1, 2)
     self.assertEqual([date for date in date_range(start_date, end_date)], [start_date])