Allow to attach shared basket to treatment files

1 files changed, 23 insertions, 2 deletions

diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py
index 150acfa05..284b9a76c 100644
--- a/archaeological_finds/forms.py
+++ b/archaeological_finds/forms.py
@@ -83,7 +83,7 @@ __all__ = [
     'check_value', 'check_type_field', 'check_type_not_field',
     'check_treatment', 'ResultFindForm', 'ResultFindFormSet',
     'FindDeletionForm', 'UpstreamFindFormSelection', 'NewFindBasketForm',
-    'SelectFindBasketForm', 'FindBasketAddItemForm',
+    'SelectFindBasketForm', 'SelectFindBasketWriteForm', 'FindBasketAddItemForm',
     'QAFindFormSingle', 'QAFindFormMulti', 'QAFindBasketForm',
     'QAFindTreatmentForm', 'QAFindbasketDuplicateForm',
     'N1TreatmentForm', 'OneNTreatmentForm', 'ResultingFindForm',
@@ -1646,7 +1646,28 @@ class SelectFindBasketForm(IshtarForm):
             self.user = kwargs.pop('user')
             if hasattr(self.user, 'ishtaruser'):
                 self.user = self.user.ishtaruser
-        super(SelectFindBasketForm, self).__init__(*args, **kwargs)
+        super(SelectFindBasketWriteForm, self).__init__(*args, **kwargs)
+
+
+class SelectFindBasketWriteForm(IshtarForm):
+    form_label = _(u"Basket")
+    associated_models = {'basket': models.FindBasket}
+    need_user_for_initialization = True
+
+    basket = forms.IntegerField(
+        label=_(u"Basket"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-findbasket-write'),
+            associated_model=models.FindBasket),
+        validators=[valid_id(models.FindBasket)])
+
+    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
+        super(SelectFindBasketWriteForm, self).__init__(*args, **kwargs)
 
 
 class FindBasketAddItemForm(forms.Form):