1 files changed, 28 insertions, 1 deletions
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 46594a2b0..e40a3acb9 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -18,16 +18,19 @@
 # See the file COPYING for details.
 
 from django.core.urlresolvers import reverse
+from django.http import HttpResponseRedirect
 from django.shortcuts import redirect
 from django.utils.translation import ugettext_lazy as _
+from django.views.generic.edit import CreateView, DeleteView
 
 from ishtar_common.forms import FinalForm
 from ishtar_common.forms_common import SourceForm, AuthorFormset, \
     SourceDeletionForm
+from ishtar_common.models import IshtarUser
 from archaeological_context_records.forms import RecordFormSelection
 
 from ishtar_common.views import get_item, show_item, revert_item, \
-    get_autocomplete_generic
+    get_autocomplete_generic, IshtarMixin, LoginRequiredMixin
 from ishtar_common.wizards import SearchWizard
 
 from wizards import *
@@ -156,6 +159,30 @@ autocomplete_preservationtype = get_autocomplete_generic(
     models.PreservationType)
 autocomplete_integritytype = get_autocomplete_generic(models.IntegrityType)
 
+
+class NewFindBasketView(IshtarMixin, LoginRequiredMixin, CreateView):
+    template_name = 'ishtar/form.html'
+    model = models.FindBasket
+    form_class = NewFindBasketForm
+    page_name = _(u"New basket")
+
+    def get_form_kwargs(self):
+        kwargs = super(NewFindBasketView, self).get_form_kwargs()
+        kwargs['user'] = IshtarUser.objects.get(pk=self.request.user.pk)
+        print(kwargs)
+        return kwargs
+
+    def get_success_url(self):
+        return reverse('new_findbasket')
+
+    def form_valid(self, form):
+        self.object = form.save()
+        return HttpResponseRedirect(self.get_success_url())
+
+
+class DeleteFindBasketView(IshtarMixin, LoginRequiredMixin, DeleteView):
+    pass
+
 """
 treatment_creation_wizard = TreatmentWizard.as_view([
     ('basetreatment-treatment_creation', BaseTreatmentForm),