from django.views.generic import UpdateView, DetailView, ListView 
from django.views.generic.edit import CreateView, DeleteView, UpdateView
from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse

from .models import GTDNote, GTDProject, WiredNote, WiredPost
from .forms import GTDNoteCreateForm, GTDNoteEditForm, GTDProjectCreateForm, GTDProjectUpdateForm, WiredNoteCreateForm, WiredNoteEditForm, WiredPostUpdateForm

class GTDNoteCreateView(CreateView):
    model = GTDNote
    form_class = GTDNoteCreateForm
    template_name = "gtd/note_form.html"

    def get_form_kwargs(self):
       kwargs = super(GTDNoteCreateView, self).get_form_kwargs()
       return kwargs

    def get_success_url(self):
        if 'add_new' in self.request.POST:
            return reverse('gtd:note-create')
        else:
            if self.object.project:
                return reverse('gtd:project-detail', kwargs={"pk": self.object.project.pk})
            else:
                return reverse('gtd:note-create')


class GTDNoteUpdateView(UpdateView):
    model = GTDNote
    form_class = GTDNoteEditForm
    template_name = "gtd/note_form.html"
    
    def get_context_data(self, **kwargs):
        context = super(GTDNoteUpdateView, self).get_context_data(**kwargs)
        context['is_update'] = True
        return context

    def get_success_url(self):
        return reverse('gtd:project-detail', kwargs={"pk": self.object.project.pk})


class GTDNoteListView(ListView):
    model = GTDNote
    template_name = "gtd/note_list.html"

    def get_queryset(self):
        if self.kwargs['status']:
            status_reverse = dict((v, k) for k, v in GTDNote.STATUS)
            status = status_reverse[self.kwargs['status'].title()]
            return GTDNote.objects.filter(status=status)
        return GTDNote.objects.filter(status=1)
    
    def get_context_data(self, **kwargs):
        context = super(GTDNoteListView, self).get_context_data(**kwargs)
        context['note_statuses'] = GTDNote.STATUS
        return context


class GTDProjectCreateView(CreateView):
    model = GTDProject
    form_class = GTDProjectCreateForm
    template_name = "gtd/project_form.html"

    def get_form_kwargs(self):
       kwargs = super(GTDProjectCreateView, self).get_form_kwargs()
       return kwargs

    def get_success_url(self):
        if 'add_new' in self.request.POST:
            return reverse('gtd:project-create')
        else:
            return reverse('gtd:project-detail', kwargs={"pk": self.object.pk})


class GTDProjectUpdateView(UpdateView):
    model = GTDProject
    form_class = GTDProjectUpdateForm
    template_name = "gtd/project_form.html"

    def get_context_data(self, **kwargs):
        context = super(GTDProjectUpdateView, self).get_context_data(**kwargs)
        context['is_update'] = True
        return context


class GTDProjectDetailView(DetailView):
    model = GTDProject 
    template_name = "gtd/project_detail.html"

    def get_context_data(self, **kwargs):
        context = super(GTDProjectDetailView, self).get_context_data(**kwargs)
        return context


class GTDProjectListView(ListView):
    model = GTDProject 
    template_name = "gtd/project_list.html"

    def get_context_data(self, **kwargs):
        context = super(GTDProjectListView, self).get_context_data(**kwargs)
        return context


class GTDNoteTODOView(ListView):
    model = GTDNote
    template_name = "gtd/note_list.html"

    def get_queryset(self):
        return GTDNote.objects.filter(status=1)
    

class WiredNoteCreateView(CreateView):
    model = WiredNote
    form_class = WiredNoteCreateForm

    def get_form_kwargs(self):
       kwargs = super(WiredNoteCreateView, self).get_form_kwargs()
       kwargs.update({'user': self.request.user})
       return kwargs

    def get_success_url(self):
        if 'add_new' in self.request.POST:
            return reverse('gtd:wirednotes-create')
        else:
            if self.object.post:
                return reverse('gtd:post-detail', kwargs={"pk": self.object.post.pk})
            else:
                return reverse('gtd:wirednotes-create')


class WiredNoteUpdateView(UpdateView):
    model = WiredNote
    form_class = WiredNoteEditForm
    
    def get_form_kwargs(self):
       kwargs = super(WiredNoteUpdateView, self).get_form_kwargs()
       return kwargs

    def get_context_data(self, **kwargs):
        context = super(WiredNoteUpdateView, self).get_context_data(**kwargs)
        context['is_update'] = True
        return context

    def get_success_url(self):
        return reverse('gtd:post-detail', kwargs={"pk": self.object.post.pk})


class WiredNoteListView(ListView):
    model = WiredNote

    def get_queryset(self):
        if self.kwargs['status']:
            status_reverse = dict((v, k) for k, v in WiredNote.STATUS)
            status = status_reverse[self.kwargs['status'].title()]
            return WiredNote.objects.filter(status=status)
        return WiredNote.objects.all()
    
    def get_context_data(self, **kwargs):
        context = super(WiredNoteListView, self).get_context_data(**kwargs)
        context['note_statuses'] = WiredNote.STATUS
        return context


class WiredPostListView(ListView):
    model = WiredPost 
    template_name = 'gtd/post_table.html'

    def get_queryset(self):
        return WiredPost.objects.all().order_by("-needs_update")


class WiredPostNewView(ListView):
    template_name = 'gtd/post_list.html'
    
    def get_queryset(self):
        return WiredPost.objects.filter(is_live=0)

    def get_context_data(self, **kwargs):
        context = super(WiredPostNewView, self).get_context_data(**kwargs)
        context['reviews'] = WiredNote.objects.filter(plan__in=[1,2], status__in=[0,1,2,3])
        return context


class WiredPostTodoView(LoginRequiredMixin, ListView):
    template_name = 'posts/post_table.html'
    
    def get_queryset(self):
        qs = WiredPost.objects.filter(user=self.request.user)
        qs = qs.filter(Q(needs_update=True) | Q(is_live=0)).exclude(post_status=1)
        unsorted_results = qs.all()
        return sorted(unsorted_results, key=lambda a: a.days_overdue(), reverse=True)

    def get_context_data(self, **kwargs):
        context = super(WiredPostTodoView, self).get_context_data(**kwargs)
        context['reviews'] = WiredNote.objects.filter(plan__in=[1,2,3]).exclude(status=4)
        return context


class WiredPostNotesView(DetailView):
    model = WiredPost 

    def get_context_data(self, **kwargs):
        context = super(WiredPostNotesView, self).get_context_data(**kwargs)
        context['notes'] = self.get_object().wirednote_set.all().order_by("status")
        return context

class WiredPostUpdateView(UpdateView):
    model = WiredPost
    form_class = WiredPostUpdateForm

    def get_form_kwargs(self):
       kwargs = super(WiredPostUpdateView, self).get_form_kwargs()
       return kwargs

    def get_success_url(self):
        if 'add_new' in self.request.POST:
            return reverse('gtd:posts-create')
        else:
            return reverse('gtd:posts-detail', kwargs={"pk": self.object.pk})