From 9b6160b263f737271a0f1aad44b02bca0a70b878 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@proxience.com>
Date: Mon, 14 Dec 2015 21:44:48 +0100
Subject: Force filtering on "own" items on selection for modification

---
 archaeological_operations/urls.py    | 6 ++++++
 archaeological_operations/wizards.py | 2 ++
 2 files changed, 8 insertions(+)

(limited to 'archaeological_operations')

diff --git a/archaeological_operations/urls.py b/archaeological_operations/urls.py
index a6c4bd7b3..68ac0bad0 100644
--- a/archaeological_operations/urls.py
+++ b/archaeological_operations/urls.py
@@ -89,10 +89,16 @@ urlpatterns += patterns(
     'archaeological_operations.views',
     url(r'autocomplete-operation/$', 'autocomplete_operation',
         name='autocomplete-operation'),
+    url(r'get-operation/own/(?P<type>.+)?$',
+        'get_operation', name='get-own-operation',
+        kwargs={'force_own': True}),
     url(r'get-operation/(?P<type>.+)?$', 'get_operation',
         name='get-operation'),
     url(r'get-operation-full/(?P<type>.+)?$', 'get_operation',
         name='get-operation-full', kwargs={'full': True}),
+    url(r'get-operation-full/own/(?P<type>.+)?$',
+        'get_operation', name='get-own-operation-full',
+        kwargs={'full': True, 'force_own': True}),
     url(r'get-available-operation-code/(?P<year>.+)?$',
         'get_available_operation_code', name='get_available_operation_code'),
     url(r'revert-operation/(?P<pk>.+)/(?P<date>.+)$',
diff --git a/archaeological_operations/wizards.py b/archaeological_operations/wizards.py
index 6f4a6583a..78a5c3350 100644
--- a/archaeological_operations/wizards.py
+++ b/archaeological_operations/wizards.py
@@ -270,6 +270,7 @@ class OperationWizard(Wizard):
 
 class OperationModificationWizard(OperationWizard):
     modification = True
+    filter_owns = {'selec-operation_modification': ['pk']}
 
 
 class OperationClosingWizard(ClosingWizard):
@@ -282,6 +283,7 @@ class OperationClosingWizard(ClosingWizard):
 class OperationDeletionWizard(DeletionWizard):
     model = models.Operation
     fields = OperationClosingWizard.fields
+    filter_owns = {'selec-operation_deletion': ['pk']}
 
 
 class OperationSourceWizard(SourceWizard):
-- 
cgit v1.2.3