import os
from pathlib import Path

from django.shortcuts import get_object_or_404
from django_drf_filepond.models import TemporaryUpload
from rest_framework import status
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from rest_framework.views import APIView

from .celery_tasks import import_dataset, upload_to_store
from .pipeline.catalog import Options
from projects.models import Project
from projects.permissions import IsProjectAdmin


class DatasetCatalog(APIView):
    permission_classes = [IsAuthenticated & IsProjectAdmin]

    def get(self, request, *args, **kwargs):
        project_id = kwargs["project_id"]
        project = get_object_or_404(Project, pk=project_id)
        options = Options.filter_by_task(project.project_type)
        return Response(data=options, status=status.HTTP_200_OK)


class DatasetImportAPI(APIView):
    permission_classes = [IsAuthenticated & IsProjectAdmin]

    def post(self, request, *args, **kwargs):
        project_id = self.kwargs["project_id"]
        upload_ids = request.data.pop("uploadIds")
        file_format = request.data.pop("format")

        # Rename file
        temporary_uploads = TemporaryUpload.objects.filter(upload_id__in=upload_ids)
        for tu in temporary_uploads:
            p = Path(tu.get_file_path())
            new_path = Path(p.parent, tu.upload_name)
            p.rename(new_path)
            tu.file.name = os.path.join(new_path.parent.stem, new_path.name)
        TemporaryUpload.objects.bulk_update(temporary_uploads, fields=["file"])

        file_paths = [tu.get_file_path() for tu in temporary_uploads]
        save_names = {tu.get_file_path(): tu.file.name for tu in temporary_uploads}

        task = import_dataset.delay(
            user_id=request.user.id,
            project_id=project_id,
            filenames=file_paths,
            file_format=file_format,
            save_names=save_names,
            **request.data,
        )
        upload_task = upload_to_store.delay(upload_ids)
        return Response({"task_id": task.task_id, "uploadTaskId": upload_task.task_id})