1 files changed, 30 insertions, 0 deletions

diff --git a/proposals/views.py b/proposals/views.py
new file mode 100644
index 0000000..aacba55
--- /dev/null
+++ b/proposals/views.py
@@ -0,0 +1,30 @@
+from django.shortcuts import resolve_url
+from django.views.generic.edit import CreateView, UpdateView
+
+from proposals.models import Proposal
+from proposals.forms import UserProposalForm
+
+
+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