Improve file and admin act (file) searches (refs #1533)

1 files changed, 34 insertions, 3 deletions

diff --git a/archaeological_files/forms.py b/archaeological_files/forms.py
index bf05c996d..b97ca9f15 100644
--- a/archaeological_files/forms.py
+++ b/archaeological_files/forms.py
@@ -46,13 +46,40 @@ from ishtar_common import widgets
 
 class FileSelect(TableSelect):
     towns = get_town_field()
-    in_charge = get_person_field(label=_(u"Person in charge"),
-                                 person_types=['sra_agent'])
+    name = forms.CharField(label=_(u"File name"), max_length=200)
     file_type = forms.ChoiceField(label=_("File type"), choices=[])
     saisine_type = forms.ChoiceField(label=_("Saisine type"), choices=[])
-    year = forms.IntegerField(label=_("Year"))
     parcel = ParcelField(label=_("Parcel (section/number)"))
     end_date = forms.NullBooleanField(label=_(u"Is active?"), initial=True)
+    general_contractor = forms.IntegerField(
+            label=_(u"General contractor"),
+            widget=widgets.JQueryAutoComplete(
+                reverse_lazy('autocomplete-person',
+                args=[PersonType.objects.get(txt_idx='general_contractor').pk]),
+                associated_model=Person),
+            validators=[valid_id(Person)])
+    general_contractor__attached_to = forms.IntegerField(
+                label=_(u"Organization of general contractor"),
+            widget=widgets.JQueryAutoComplete(
+                reverse_lazy('autocomplete-organization',
+                args=[OrganizationType.objects.get(
+                                        txt_idx='general_contractor').pk]),
+                associated_model=Organization),
+            validators=[valid_id(Organization)])
+    numeric_reference = forms.IntegerField(
+                                           label=_("Numeric reference"))
+    year = forms.IntegerField(label=_("Year"))
+    internal_reference = forms.CharField(max_length=200,
+                                         label=_("Other reference"))
+    in_charge = forms.IntegerField(
+            label=_(u"In charge"),
+            widget=widgets.JQueryAutoComplete(
+                reverse_lazy('autocomplete-person',
+                args=[PersonType.objects.get(txt_idx='sra_agent').pk]),
+                associated_model=Person),
+            validators=[valid_id(Person)])
+    permit_reference = forms.CharField(max_length=200,
+                                       label=_("Permit reference"))
 
     def __init__(self, *args, **kwargs):
         super(FileSelect, self).__init__(*args, **kwargs)
@@ -226,6 +253,8 @@ class AdministrativeActFileSelect(TableSelect):
                                             widget=widgets.JQueryDate)
     signature_date_before = forms.DateField(label=_(u"Signature date before"),
                                             widget=widgets.JQueryDate)
+    associated_file__name = forms.CharField(label=_(u"File name"),
+                                            max_length=200)
     associated_file__general_contractor = forms.IntegerField(
             label=_(u"General contractor"),
             widget=widgets.JQueryAutoComplete(
@@ -253,6 +282,8 @@ class AdministrativeActFileSelect(TableSelect):
                 args=[PersonType.objects.get(txt_idx='sra_agent').pk]),
                 associated_model=Person),
             validators=[valid_id(Person)])
+    associated_file__permit_reference = forms.CharField(max_length=200,
+                                           label=_("Permit reference"))
 
     def __init__(self, *args, **kwargs):
         super(AdministrativeActFileSelect, self).__init__(*args, **kwargs)