from django.shortcuts import render, redirect
from .forms import UploadImageForm
from .models import Upload
from django.contrib.contenttypes.models import ContentType


def upload_image_view(request):
    if request.method == 'POST':
        form = UploadImageForm(request.POST, request.FILES)
        if form.is_valid():
            image = form.save()

            upload = Upload.objects.create(
                filename=image.file.name,
                content_type=ContentType.objects.get_for_model(image),
                object_id=image.id,
                user=request.user if request.user.is_authenticated else None
            )

            return redirect('upload_success', slug=upload.slug)

    else:
        form = UploadImageForm()

    return render(request, 'upload/upload_form.html', {'form': form})