# -*- coding: utf-8 -*-
"""Akvo RSR is covered by the GNU Affero General Public License.
See more details in the license.txt file located at the root folder of the Akvo RSR module.
For additional details on the GNU license please see < http://www.gnu.org/licenses/agpl.html >.
"""
from akvo.rest.authentication import TastyTokenAuthentication, JWTAuthentication
from akvo.rsr.models import Indicator, Result, Project
from django.shortcuts import get_object_or_404
from rest_framework.authentication import SessionAuthentication
from rest_framework.decorators import api_view, authentication_classes
from rest_framework.response import Response
from ..serializers import (ResultSerializer, ResultsFrameworkSerializer, ResultSerializerV2,
ResultsFrameworkLiteSerializer, ResultFrameworkNotSoLiteSerializer)
from ..viewsets import PublicProjectViewSet
[docs]class ResultsViewSet(PublicProjectViewSet):
"""Results resource."""
queryset = Result.objects.all().select_related('project', 'parent_result')
[docs] def get_serializer_class(self):
if self.request.version == 'v2':
return ResultSerializerV2
return ResultSerializer
[docs]class ResultsFrameworkViewSet(PublicProjectViewSet):
"""Results framework resource."""
queryset = Result.objects.select_related('project').prefetch_related(
'indicators',
'indicators__dimension_names',
'indicators__periods',
'indicators__periods__data',
'indicators__periods__data__comments',
'indicators__periods__disaggregation_targets',
)
serializer_class = ResultsFrameworkSerializer
[docs]class ResultsFrameworkLiteViewSet(PublicProjectViewSet):
"""Results framework lite resource."""
queryset = Result.objects.select_related('project').prefetch_related(
'indicators',
'indicators__custom_values',
'indicators__custom_values__custom_field',
'indicators__dimension_names',
'indicators__periods',
'indicators__periods__disaggregation_targets',
)
serializer_class = ResultsFrameworkLiteSerializer
[docs]@api_view(['GET'])
@authentication_classes([SessionAuthentication, TastyTokenAuthentication, JWTAuthentication])
def project_results_framework(request, project_pk):
project = get_object_or_404(Project, pk=project_pk)
queryset = Result.objects.filter(project=project).select_related('project').prefetch_related(
'indicators',
'indicators__dimension_names',
'indicators__dimension_names__dimension_values',
'indicators__disaggregation_targets',
'indicators__labels',
'indicators__periods',
'indicators__periods__data',
'indicators__periods__data__comments',
'indicators__periods__data__disaggregations',
'indicators__periods__data__disaggregations__dimension_value',
'indicators__periods__data__disaggregations__dimension_value__name',
'indicators__periods__data__indicatorperioddatafile_set',
'indicators__periods__data__indicatorperioddataphoto_set',
'indicators__periods__data__user',
'indicators__periods__disaggregation_targets',
'indicators__periods__disaggregation_targets__dimension_value',
'indicators__periods__disaggregation_targets__dimension_value__name',
'indicators__periods__disaggregations',
'indicators__periods__disaggregations__dimension_value',
'indicators__periods__disaggregations__dimension_value__name',
)
serializer = ResultFrameworkNotSoLiteSerializer(queryset, many=True, context={'request': request})
# FIXME: It may be better to not serialize all the indicators at all, but
# this seems easier to implement than hacking around in DRF, with nested serializers
project_has_assignments = Indicator.objects.filter(result__project=project, enumerators__gt=0).exists()
view = 'm&e' if request.user.has_perm('rsr.do_me_manager_actions', project) else 'enumerator'
web_token = request.GET.get('rt')
show_filtered_view = view == 'enumerator' or (web_token is not None and web_token != 'preview')
if not request.user.is_anonymous and project_has_assignments and show_filtered_view:
user_assignments = Indicator.objects.filter(result__project=project, enumerators__in=[request.user])\
.values_list('pk', flat=True)
_filter_indicators(serializer.data, set(user_assignments))
data = {
'results': serializer.data,
'title': project.title,
'targets_at': project.get_root().targets_at,
'view': view,
}
return Response(data)
def _filter_indicators(results, indicator_ids):
for result in results:
result['indicators'] = [
indicator for indicator in result['indicators'] if indicator['id'] in indicator_ids
]
return results