from django.shortcuts import resolve_url
from django.views.generic import ListView
from django.views.generic.edit import CreateView, UpdateView

from proposals.models import Proposal
from proposals.forms import UserProposalForm


class ProposalListView(ListView):

    queryset = Proposal.objects.approved()


class ProposalCreationForm(CreateView):

    model = Proposal
    form_class = UserProposalForm

    def get_success_url(self):
        return resolve_url("account:profile")


class ProposalUpdateView(UpdateView):

    form_class = UserProposalForm

    def get_success_url(self):
        return resolve_url("account:profile")

    def get_queryset(self):
        return Proposal.objects.for_user(self.request.user)

    def get_context_data(self, **kwargs):
        context = super(ProposalUpdateView, self).get_context_data(**kwargs)
        context["edit_view"] = True
        return context