Generic QA form - QA find basket form

1 files changed, 25 insertions, 1 deletions

diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py
index e50172f1c..eb27c22cc 100644
--- a/archaeological_finds/forms.py
+++ b/archaeological_finds/forms.py
@@ -79,7 +79,7 @@ __all__ = [
     'check_treatment', 'ResultFindForm', 'ResultFindFormSet',
     'FindDeletionForm', 'UpstreamFindFormSelection', 'NewFindBasketForm',
     'SelectFindBasketForm', 'DeleteFindBasketForm', 'FindBasketAddItemForm',
-    'QAFindFormSingle', 'QAFindFormMulti'
+    'QAFindFormSingle', 'QAFindFormMulti', 'QAFindBasketForm'
 ]
 
 logger = logging.getLogger(__name__)
@@ -442,6 +442,30 @@ class QAFindFormSingle(QAFindFormMulti):
             self.fields.pop('qa_seal_number')
 
 
+class QAFindBasketForm(IshtarForm):
+    create_or_update = forms.ChoiceField(
+        choices=(('create', _(u"Create")),
+                 ('update', _(u"Update"))), initial='create')
+    label = forms.CharField(label="", max_length=None, required=False)
+    basket = forms.ChoiceField(label=_(u"Basket"), required=True, choices=[])
+
+    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.items = kwargs.pop('items')
+
+        super(QAFindBasketForm, self).__init__(*args, **kwargs)
+        if not self.user:
+            return
+        self.fields['basket'].choices = [('', '--')] + [
+            (b.pk, unicode(b))
+            for b in models.FindBasket.objects.filter(user=self.user)
+        ]
+
+
 class PreservationForm(CustomForm, ManageOldType):
     form_label = _("Preservation")
     form_admin_name = _(u"Find - 030 - Preservation")