Basket: duplicate QA

1 files changed, 30 insertions, 1 deletions

diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py
index f78b4a5f3..e896b310d 100644
--- a/archaeological_finds/forms.py
+++ b/archaeological_finds/forms.py
@@ -84,7 +84,7 @@ __all__ = [
     'FindDeletionForm', 'UpstreamFindFormSelection', 'NewFindBasketForm',
     'SelectFindBasketForm', 'FindBasketAddItemForm',
     'QAFindFormSingle', 'QAFindFormMulti', 'QAFindBasketForm',
-    'QAFindTreatmentForm',
+    'QAFindTreatmentForm', 'QAFindbasketDuplicateForm',
     'N1TreatmentForm', 'OneNTreatmentForm', 'ResultingFindForm',
     'ResultingFindsForm', 'SingleUpstreamFindFormSelection'
 ]
@@ -695,6 +695,35 @@ class QAFindBasketForm(IshtarForm):
             basket.items.add(item)
 
 
+class QAFindbasketDuplicateForm(IshtarForm):
+    label = forms.CharField(label="", max_length=None, required=True)
+
+    def __init__(self, *args, **kwargs):
+        self.user = None
+        if 'user' in kwargs:
+            self.user = kwargs.pop('user')
+            if hasattr(self.user, 'ishtaruser'):
+                self.user = self.user.ishtaruser
+        self.basket = kwargs.pop('items')[0]
+        super(QAFindbasketDuplicateForm, self).__init__(*args, **kwargs)
+        self.fields['label'].initial = self.basket.label + unicode(
+            _(u" - duplicate"))
+
+    def clean(self):
+        label = self.cleaned_data['label'].strip()
+        if not label:
+            raise forms.ValidationError(_(u"A label is required."))
+        if models.FindBasket.objects.filter(user=self.user,
+                                            label=label).count():
+            raise forms.ValidationError(_(u"A basket with this label already "
+                                          u"exists."))
+        return self.cleaned_data
+
+    def save(self):
+        self.basket.duplicate(label=self.cleaned_data['label'],
+                              ishtaruser=self.user)
+
+
 class PreservationForm(CustomForm, ManageOldType):
     form_label = _("Preservation")
     form_admin_name = _(u"Find - 030 - Preservation")