120 lines
4.6 KiB
Python
120 lines
4.6 KiB
Python
from django.shortcuts import render
|
|
from rest_framework import viewsets, filters
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework.pagination import PageNumberPagination
|
|
|
|
from api.logger.mixins import LoggingMixin
|
|
from mixins.filtrado_organizacion import OrganizacionFiltradaMixin, ProcesosPorOrganizacionMixin
|
|
from .models import Task
|
|
from .serializers import TaskSerializer
|
|
from .filters import TaskFilter
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
# Create your views here.
|
|
from core.permissions import (
|
|
IsSameOrganization,
|
|
IsSameOrganizationDeveloper,
|
|
IsSameOrganizationAndAdmin,
|
|
IsSuperUser
|
|
)
|
|
|
|
class TaskPagination(PageNumberPagination):
|
|
page_size = 10
|
|
page_size_query_param = 'page_size'
|
|
max_page_size = 100
|
|
|
|
class TaskViewSet(LoggingMixin,viewsets.ModelViewSet,OrganizacionFiltradaMixin):
|
|
permission_classes = [IsAuthenticated & (IsSameOrganization | IsSameOrganizationAndAdmin | IsSameOrganizationDeveloper | IsSuperUser)]
|
|
queryset = Task.objects.select_related('pedimento', 'servicio').all()
|
|
serializer_class = TaskSerializer
|
|
filter_backends = [DjangoFilterBackend, filters.OrderingFilter]
|
|
filterset_class = TaskFilter
|
|
pagination_class = TaskPagination
|
|
ordering_fields = ['timestamp']
|
|
ordering = ['-timestamp'] # ordenamiento por defecto, más reciente primero
|
|
|
|
my_tags = ['tasks']
|
|
|
|
def get_queryset(self):
|
|
|
|
"""
|
|
Filtra las tareas según la organización del usuario.
|
|
Superusuarios pueden ver todas las tareas.
|
|
"""
|
|
queryset = self.get_queryset_filtrado_por_organizacion() # Tambien filtra por importador
|
|
# if user.is_superuser:
|
|
# return self.queryset
|
|
# # return self.queryset.filter(organizacion_id=user.organizacion.id)
|
|
# else:
|
|
# return self.queryset.filter(organizacion_id=user.organizacion.id)
|
|
return queryset
|
|
|
|
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from celery.result import AsyncResult
|
|
|
|
|
|
class TaskStatusView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get(self, request, task_id):
|
|
"""
|
|
Consulta el estado de una tarea Celery.
|
|
|
|
Estados posibles:
|
|
PENDING — en cola, aún no inició
|
|
STARTED — worker la tomó y está ejecutando
|
|
SUCCESS — terminó correctamente, `result` contiene el resumen
|
|
FAILURE — lanzó una excepción no capturada, `error` describe el problema
|
|
RETRY — el worker la está reintentando
|
|
"""
|
|
try:
|
|
task_result = AsyncResult(task_id)
|
|
state = task_result.state
|
|
|
|
response_data = {
|
|
'task_id': task_id,
|
|
'status': state,
|
|
'ready': task_result.ready(),
|
|
'successful': task_result.successful() if task_result.ready() else None,
|
|
}
|
|
|
|
if state == 'SUCCESS':
|
|
result = task_result.result
|
|
response_data['result'] = result
|
|
|
|
# Resumen legible cuando es auditoría masiva de organización
|
|
if isinstance(result, dict) and 'total_pedimentos' in result:
|
|
total = result.get('total_pedimentos', 0)
|
|
completados = result.get('completados', 0)
|
|
con_pendientes = result.get('con_pendientes', 0)
|
|
con_errores = result.get('con_errores', 0)
|
|
|
|
if con_pendientes == 0 and con_errores == 0:
|
|
mensaje = f'Auditoría completa — {completados}/{total} pedimentos sin pendientes'
|
|
else:
|
|
partes = []
|
|
if con_pendientes:
|
|
partes.append(f'{con_pendientes} con documentos pendientes')
|
|
if con_errores:
|
|
partes.append(f'{con_errores} con error')
|
|
mensaje = f'{completados}/{total} pedimentos completos — {", ".join(partes)}'
|
|
|
|
response_data['mensaje'] = mensaje
|
|
|
|
elif state == 'FAILURE':
|
|
response_data['error'] = str(task_result.info)
|
|
|
|
elif state == 'STARTED':
|
|
response_data['info'] = str(task_result.info) if task_result.info else None
|
|
|
|
return Response(response_data, status=status.HTTP_200_OK)
|
|
|
|
except Exception as e:
|
|
return Response(
|
|
{'error': f'Error al consultar tarea: {str(e)}'},
|
|
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
)
|
|
|