#!/usr/bin/env python
# -*- coding: utf-8 -*-

from django.views import View

from django.utils.translation import gettext as _

from django.shortcuts import render, redirect, get_object_or_404
from django.urls import reverse_lazy
from django.contrib import messages
from django.contrib.auth.mixins import UserPassesTestMixin

from lostplaces.common import get_all_subclasses
from lostplaces.views.base_views import IsAuthenticatedMixin
from lostplaces.models.models import Explorer
from lostplaces.models.place import Place, PlaceAsset
from lostplaces.forms import ExplorerChangeForm, ExplorerUserChangeForm

class ExplorerProfileView(IsAuthenticatedMixin, View):
    def get(self, request, explorer_id):
        explorer = get_object_or_404(Explorer, pk=explorer_id)
        place_list = explorer.get_place_list_to_display()
        place_count = place_list.count()

        context={
            'explorer': explorer,
            'place_count': place_count,
            'place_list': place_list,
            'assets': {}
        }

        asset_count = 0
        for subclass in get_all_subclasses(PlaceAsset): # kinda ugly, but PlaceAsset cannot be abstract according to django ORM
            objects = subclass.objects.filter(submitted_by=explorer)
            context['assets'][subclass.__name__.lower()+'s'] = objects
            asset_count += objects.count()
        
        context['asset_count'] = asset_count
                
        return render(
            request=request,
            template_name='explorer/profile.html',
            context=context
        )

class ExplorerProfileUpdateView(IsAuthenticatedMixin, View):
    success_message = ''
    permission_denied_message = ''

    def get(self, request, *args, **kwargs):
        context = {
            'explorer_user_change_form': ExplorerUserChangeForm(instance=request.user),
            'explorer_change_form': ExplorerChangeForm(instance=request.user.explorer)
        }
        if request.user.explorer.profile_image:
            context['explorer_image'] = request.user.explorer.profile_image
        return render(request, 'explorer/profile_update.html', context)

    def post(self, request, *args, **kwargs):
        explorer_user_change_form = ExplorerUserChangeForm(
            request.POST,
            instance=request.user
            )
        explorer_change_form = ExplorerChangeForm(
            request.POST,
            request.FILES,
            instance=request.user.explorer
        )

        if explorer_change_form.is_valid() and explorer_user_change_form.is_valid():
            explorer_user_change_form.save()
            explorer_change_form.save()

            messages.success(
                self.request,
                _('Successfully updated Explorer profile')
            )
        else:
            # Usually the browser should have checked the form before sending.
            messages.error(
                self.request,
                _('Please fill in all required fields.')
            )
        return redirect(reverse_lazy('explorer_profile_update'))
    
class ExplorerDraftsView(IsAuthenticatedMixin, UserPassesTestMixin, View):

    def get_explorer(self):
        return get_object_or_404(Explorer, pk=self.kwargs['explorer_id'])

    def test_func(self):
        if not hasattr(self.request, 'user'):
            return False
        return self.request.user == self.get_explorer().user or self.request.user.is_superuser

    def get(self, request, *args, **kwargs):
        context = {
            'explorer': self.get_explorer(),
            'place_list': self.get_explorer().get_drafts()
        }
        return render(request, 'explorer/drafts.html', context)