Context record: can change operation (refs #3080)

1 files changed, 15 insertions, 0 deletions

diff --git a/archaeological_context_records/forms.py b/archaeological_context_records/forms.py
index 462300317..225b49bfb 100644
--- a/archaeological_context_records/forms.py
+++ b/archaeological_context_records/forms.py
@@ -43,6 +43,18 @@ from archaeological_operations.forms import OperationSelect, ParcelField,\
     RecordRelationsForm as OpeRecordRelationsForm
 
 
+class OperationFormSelection(forms.Form):
+    form_label = _("Operation")
+    associated_models = {'operation': Operation}
+    currents = {'operation': Operation}
+    operation = forms.IntegerField(
+        label=_(u"Operation"), required=False,
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-operation'),
+            associated_model=Operation),
+        validators=[valid_id(Operation)])
+
+
 class RecordSelect(TableSelect):
     label = forms.CharField(label=_(u"ID"), max_length=100)
     parcel__town = get_town_field()
@@ -149,6 +161,9 @@ class RecordFormGeneral(ManageOldType, forms.Form):
                  'context_record' in kwargs['data']):
             if 'operation' in kwargs['data']:
                 operation = kwargs['data']['operation']
+                # force operation modification
+                kwargs['data'][kwargs['prefix'] + '-operation_id'] = \
+                    operation.id
             if 'context_record' in kwargs['data'] and \
                kwargs['data']['context_record']:
                 operation = kwargs['data']['context_record'].operation