views.py

# -*- coding: utf-8 -*-
from rest_framework import viewsets
from rest_framework.response import Response
from braces.views import LoginRequiredMixin
from core.models import Course, CourseStudent, Class
from django.core.exceptions import ObjectDoesNotExist
from reports.serializer import UserCourseStatsSerializer, CourseStats, LessonUserStats
from rest_framework.pagination import PageNumberPagination
from django.db.models import Q


class CustomPagination(PageNumberPagination):
    page_size = 40


class UserCourseStats(LoginRequiredMixin, viewsets.ReadOnlyModelViewSet):
    model = CourseStudent
    queryset = CourseStudent.objects.all()
    serializer_class = UserCourseStatsSerializer
    filter_fields = ('course',)
    pagination_class = CustomPagination

    def get_queryset(self):
        queryset = super(UserCourseStats, self).get_queryset()
        user = self.request.user
        course_id = self.request.query_params.get('course')
        role = None
        try:
            role = self.request.user.teaching_courses.get(course__id=course_id).role
        except ObjectDoesNotExist:
            pass

        search = self.request.query_params.get('s')
        if search:
            queryset = queryset.filter(Q(user__first_name__icontains=search) |
                                       Q(user__last_name__icontains=search) |
                                       Q(user__username__icontains=search) |
                                       Q(user__email__icontains=search))

        if (role and role == 'coordinator') or self.request.user.is_staff or self.request.user.is_superuser:
            pass
        else:
            # if user is not coordinator or admin, only show his classes
            classes = Class.objects.filter(course=course_id)
            classes = classes.filter(assistant=user)
            queryset = queryset.filter(user__classes__in=classes)

        # TODO: Fix ordering
        ordering = self.request.query_params.get('ordering')
        if ordering and ordering == 'course_progress':
            pass
            #         id_progress = {}
            #         for item in queryset:
            #             id_progress[item.id] = item.percent_progress()
            #         import operator
            #         id_progress = sorted(id_progress.items(), key=operator.itemgetter(1))
            #         queryset = CourseStudent.objects.filter(id__in=id_progress.keys())
        else:
            queryset = queryset.order_by('user__first_name')

        reverse = self.request.query_params.get('reverse')
        if reverse and reverse == 'true':
            queryset = queryset.reverse()

        return queryset


class UserCourseLessonsStats(LoginRequiredMixin, viewsets.ReadOnlyModelViewSet):
    model = CourseStudent
    queryset = CourseStudent.objects.all()
    serializer_class = LessonUserStats
    filter_fields = ('course', 'user',)
    lookup_field = 'course'


class CourseStatsByLessonViewSet(LoginRequiredMixin, viewsets.ReadOnlyModelViewSet):
    model = Course
    queryset = Course.objects.all()
    serializer_class = CourseStats
    pagination_class = CustomPagination

    def retrieve(self, request, *args, **kwargs):
        self.object = self.get_object()

        course_id = self.kwargs.get('pk')
        role = None
        try:
            role = self.request.user.teaching_courses.get(course__id=course_id).role
        except ObjectDoesNotExist:
            pass

        classes_id = self.request.query_params.getlist('classes')
        # class passed as get paremeter
        classes = Class.objects.filter(course=course_id)
        if classes_id:
            classes = classes.filter(id__in=classes_id)
        # if user is not coordinator or admin, only show his classes
        if not (role and (role == 'coordinator') and self.request.user.is_staff and self.request.user.is_superuser):
            classes = classes.filter(assistant=self.request.user)

        self.object.classes = classes
        serializer = self.get_serializer(self.object)
        return Response(serializer.data)