import secrets

from django.views import View
from django.views.generic.list import ListView
from django.shortcuts import render, redirect
from django.urls import reverse

from .forms import ImageUploadForm
from .models import (
    Image,
    Visitor
)

class VisitorSessionMixin(View):
    def get_visitor(self):
        if self.request.session:
            if 'visitor_session' in self.request.session:
                if Visitor.objects.get(session_id=self.request.session['visitor_session']).exists():
                    return Visitor.objects.get(session_id=self.request.session['visitor_session'])
            else:
                visitor = Visitor.objects.create()
                self.request.session['visitor_session'] = visitor.session_id
                return visitor
        else:   
            return None


class ImageUploadView(VisitorSessionMixin, View):
    def get(self, request):
        form = ImageUploadForm()
        return render(
            request,
            'upload_image/upload.html',
            {
                'form': form
            }
        )

    def post(self, request):
        form = ImageUploadForm(
            request.POST,
            request.FILES
        )

        if form.is_valid():
            image = form.save(commit=False)

            if image.private:
                image.access_code = secrets.token_hex(32)

            image.uploaded_by = self.get_visitor()
            print(image.uploaded_by)
            image.save()
            return redirect(
                reverse('home')
            )
        else:
            form = ImageUploadForm()
            return render(
                request,
                'upload_image/upload.html',
                {
                    'form': form
                }
            )

class PublicImageListView(ListView):
    model = Image
    paginate_by = 20
    context_object_name = 'images'
    template_name = 'list_images/image_list.html'

    def get_queryset(self):
        return Image.objects.all().filter(private=False)