Delete basket

1 files changed, 20 insertions, 0 deletions

diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py
index 2b3586979..3c06af375 100644
--- a/archaeological_finds/forms.py
+++ b/archaeological_finds/forms.py
@@ -411,6 +411,26 @@ class NewFindBasketForm(forms.ModelForm):
         self.instance.user = self.user
         return super(NewFindBasketForm, self).save(commit)
 
+
+class DeleteFindBasketForm(forms.Form):
+    basket = forms.ChoiceField(label=_(u"Basket"), required=True, choices=[])
+
+    def __init__(self, *args, **kwargs):
+        self.user = kwargs.pop('user')
+        super(DeleteFindBasketForm, self).__init__(*args, **kwargs)
+        self.fields['basket'].choices = [('', '--')] + [
+            (b.pk, unicode(b))
+            for b in models.FindBasket.objects.filter(user=self.user)]
+
+    def save(self):
+        try:
+            models.FindBasket.objects.get(pk=self.cleaned_data['basket'],
+                                          user=self.user).delete()
+        except models.FindBasket.DoesNotExist:
+            # something strange... TODO: log it
+            pass
+        return
+
 """
 ####################################
 # Source management for treatments #