# -*- 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 django.core.exceptions import ValidationError
from django.db import transaction
from django.http import HttpResponseForbidden, HttpResponseNotFound, HttpResponseBadRequest
from rest_framework import status as http_status
from rest_framework.authentication import SessionAuthentication
from rest_framework.decorators import api_view, permission_classes, authentication_classes
from rest_framework.exceptions import ValidationError as RestValidationError
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from akvo.rest.authentication import TastyTokenAuthentication
from akvo.rsr.models import Indicator, Project, Result
[docs]@api_view(['POST'])
@permission_classes((IsAuthenticated,))
@transaction.atomic
def project_editor_reorder_items(request, project_pk=None):
"""API call to reorder results or indicators"""
project = Project.objects.get(pk=project_pk)
if project.has_imported_results():
return Response(data={'errors': ['Reordering not allowed']}, status=http_status.HTTP_400_BAD_REQUEST)
errors, item_list, item_selected, swap_id = [], [], None, -1
item_type = request.POST.get('item_type', False)
item_id = request.POST.get('item_id', False)
item_direction = request.POST.get('item_direction', False)
if item_type == 'result':
item_selected = Result.objects.get(id=item_id)
item_list = Result.objects.filter(project_id=project_pk)
elif item_type == 'indicator':
item_selected = Indicator.objects.get(id=item_id)
item_list = Indicator.objects.filter(result_id=item_selected.result_id)
else:
errors += ['Invalid item type']
if not errors:
# assign order if it doesn't already exist or is messed up
n_items = item_list.count()
if n_items > 0 and set(item_list.values_list('order', flat=True)) != set(range(n_items)):
for i, item in enumerate(item_list):
item.order = i
item.save()
if item_type == 'result':
item_original_order = Result.objects.get(id=item_id).order
else:
item_original_order = Indicator.objects.get(id=item_id).order
if item_direction == 'up' and not item_original_order < 1:
item_swap = item_list.get(order=item_original_order - 1)
item_swap.order = item_original_order
item_swap.save()
swap_id = item_swap.id
if item_selected:
item_selected.order = item_original_order - 1
item_selected.save()
elif item_direction == 'down' and not item_original_order >= len(item_list) - 1:
item_swap = item_list.get(order=item_original_order + 1)
item_swap.order = item_original_order
item_swap.save()
swap_id = item_swap.id
if item_selected:
item_selected.order = item_original_order + 1
item_selected.save()
else:
errors += ['Unable to reorder the selected item, it may already be at top/bottom of '
'list.']
return Response(
{
'errors': errors,
'swap_id': swap_id,
}
)
[docs]@api_view(['POST'])
@permission_classes((IsAuthenticated, ))
def project_editor_import_results(request, project_pk=None):
project = Project.objects.get(pk=project_pk)
user = request.user
if not user.can_import_results(project):
return HttpResponseForbidden()
status_code, message = project.import_results()
if status_code == 1:
data = {'project_id': project_pk, 'import_success': True}
status = http_status.HTTP_201_CREATED
else:
data = {'project_id': project_pk, 'import_success': False, 'message': message}
status = http_status.HTTP_400_BAD_REQUEST
return Response(data=data, status=status)
[docs]@api_view(['POST'])
@permission_classes((IsAuthenticated, ))
def project_editor_copy_results(request, project_pk=None, source_pk=None):
project = Project.objects.get(pk=project_pk)
source_project = Project.objects.get(pk=source_pk)
user = request.user
if not user.can_import_results(project):
return HttpResponseForbidden()
if not user.has_perm('rsr.change_project', source_project):
return HttpResponseForbidden()
data = {'project_id': project_pk, 'copy_success': True}
try:
project.copy_results(source_project)
status = http_status.HTTP_201_CREATED
except RuntimeError as e:
data['copy_success'] = False
data['message'] = str(e)
status = http_status.HTTP_400_BAD_REQUEST
return Response(data=data, status=status)
[docs]@api_view(['POST'])
@permission_classes((IsAuthenticated, ))
@authentication_classes((SessionAuthentication, TastyTokenAuthentication))
def project_editor_import_result(request, project_pk, parent_result_id):
try:
project = Project.objects.get(pk=project_pk)
except Project.DoesNotExist:
return HttpResponseNotFound()
except Project.MultipleObjectsReturned:
return HttpResponseBadRequest()
user = request.user
if not user.can_import_results(project):
return HttpResponseForbidden()
try:
result = project.import_result(parent_result_id)
except (Project.DoesNotExist, Project.MultipleObjectsReturned, Result.DoesNotExist,
Result.MultipleObjectsReturned, ValidationError) as e:
raise RestValidationError(str(e))
data = {'result_id': result.pk, 'import_success': True}
return Response(data=data, status=http_status.HTTP_201_CREATED)
[docs]@api_view(['POST'])
@permission_classes((IsAuthenticated, ))
@authentication_classes((SessionAuthentication, TastyTokenAuthentication))
def project_editor_import_indicator(request, project_pk, parent_indicator_id):
try:
project = Project.objects.get(pk=project_pk)
except Project.DoesNotExist:
return HttpResponseNotFound()
except Project.MultipleObjectsReturned:
return HttpResponseBadRequest()
user = request.user
if not user.can_import_results(project):
return HttpResponseForbidden()
try:
indicator = project.import_indicator(parent_indicator_id)
except (Project.DoesNotExist, Project.MultipleObjectsReturned, Indicator.DoesNotExist,
Indicator.MultipleObjectsReturned, ValidationError) as e:
raise RestValidationError(str(e))
data = {'indicator_id': indicator.pk, 'import_success': True}
return Response(data=data, status=http_status.HTTP_201_CREATED)