summaryrefslogtreecommitdiff
path: root/archaeological_finds/forms.py
diff options
context:
space:
mode:
Diffstat (limited to 'archaeological_finds/forms.py')
-rw-r--r--archaeological_finds/forms.py20
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 #