117 lines
4.4 KiB
Python
117 lines
4.4 KiB
Python
from api.reports.models import ReportDocument
|
|
from api.reports.tasks.report_document import generate_report_document
|
|
from django.http import FileResponse
|
|
from rest_framework.decorators import api_view, permission_classes
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def table_summary(request):
|
|
"""
|
|
Solo dispara la tarea asíncrona para generar el reporte CSV. No consulta ni procesa datos.
|
|
"""
|
|
org_id = request.query_params.get('organizacion_id')
|
|
if not org_id:
|
|
return Response({"error": "organizacion_id es requerido"}, status=400)
|
|
# Obtener filtros de query params
|
|
tipo_documento = request.query_params.get('tipo_documento')
|
|
rfc = request.query_params.get('contribuyente__rfc')
|
|
fecha_pago_gte = request.query_params.get('fecha_pago__gte')
|
|
fecha_pago_lte = request.query_params.get('fecha_pago__lte')
|
|
patente = request.query_params.get('patente')
|
|
aduana = request.query_params.get('aduana')
|
|
pedimento = request.query_params.get('pedimento')
|
|
pedimento_app = request.query_params.get('pedimento_app')
|
|
regimen = request.query_params.get('regimen')
|
|
tipo_operacion = request.query_params.get('tipo_operacion')
|
|
filename_param = request.query_params.get('filename')
|
|
if filename_param:
|
|
filename = filename_param
|
|
else:
|
|
filename = None
|
|
# Si no se proporcionan fechas, no las incluyas en los filtros
|
|
if not fecha_pago_gte:
|
|
fecha_pago_gte_str = None
|
|
elif isinstance(fecha_pago_gte, str):
|
|
fecha_pago_gte_str = fecha_pago_gte
|
|
else:
|
|
fecha_pago_gte_str = fecha_pago_gte.isoformat()
|
|
if not fecha_pago_lte:
|
|
fecha_pago_lte_str = None
|
|
elif isinstance(fecha_pago_lte, str):
|
|
fecha_pago_lte_str = fecha_pago_lte
|
|
else:
|
|
fecha_pago_lte_str = fecha_pago_lte.isoformat()
|
|
filtros = {
|
|
"organizacion_id": org_id,
|
|
"tipo_documento": tipo_documento,
|
|
"contribuyente__rfc": rfc,
|
|
"fecha_pago__gte": fecha_pago_gte_str,
|
|
"fecha_pago__lte": fecha_pago_lte_str,
|
|
"patente": patente,
|
|
"aduana": aduana,
|
|
"pedimento": pedimento,
|
|
"pedimento_app": pedimento_app,
|
|
"regimen": regimen,
|
|
"tipo_operacion": tipo_operacion,
|
|
"filename": filename
|
|
}
|
|
report = ReportDocument.objects.create(
|
|
user=request.user,
|
|
filters=filtros,
|
|
status='pending'
|
|
)
|
|
generate_report_document.delay(report.id)
|
|
return Response({
|
|
"report_id": report.id,
|
|
"status": report.status,
|
|
"created_at": report.created_at,
|
|
"download_url": report.file.url if report.file else None
|
|
}, status=202)
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def report_document_status(request, report_id):
|
|
try:
|
|
report = ReportDocument.objects.get(id=report_id, user=request.user)
|
|
data = {
|
|
"report_id": report.id,
|
|
"status": report.status,
|
|
"created_at": report.created_at,
|
|
"finished_at": report.finished_at,
|
|
"error_message": report.error_message,
|
|
"download_url": report.file.url if report.file else None
|
|
}
|
|
return Response(data)
|
|
except ReportDocument.DoesNotExist:
|
|
return Response({"error": "Reporte no encontrado"}, status=404)
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def report_document_list(request):
|
|
reports = ReportDocument.objects.filter(user=request.user).order_by('-created_at')
|
|
data = [
|
|
{
|
|
"report_id": r.id,
|
|
"status": r.status,
|
|
"created_at": r.created_at,
|
|
"finished_at": r.finished_at,
|
|
"error_message": r.error_message,
|
|
"download_url": r.file.url if r.file else None
|
|
}
|
|
for r in reports
|
|
]
|
|
return Response(data)
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def report_document_download(request, report_id):
|
|
try:
|
|
report = ReportDocument.objects.get(id=report_id, user=request.user)
|
|
if not report.file:
|
|
return Response({"error": "El archivo aún no está disponible"}, status=404)
|
|
response = FileResponse(report.file.open('rb'), as_attachment=True, filename=report.file.name)
|
|
return response
|
|
except ReportDocument.DoesNotExist:
|
|
return Response({"error": "Reporte no encontrado"}, status=404) |