From 93d9541ef82deb918f721cb74377e630d271de7d Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Tue, 16 Apr 2013 21:59:05 +0200 Subject: Manage OR condition in searches --- archaeological_operations/views.py | 6 ++++-- ishtar_common/views.py | 12 +++++++++++- 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/archaeological_operations/views.py b/archaeological_operations/views.py index 1ce4c4622..e5bf16ce8 100644 --- a/archaeological_operations/views.py +++ b/archaeological_operations/views.py @@ -103,8 +103,10 @@ get_operation = get_item(models.Operation, 'get_operation', 'operation', 'start_after':'start_date__gte', 'end_before':'excavation_end_date__lte', 'end_after':'excavation_end_date__gte', - 'parcel_0':'parcels__section', - 'parcel_1':'parcels__parcel_number', + 'parcel_0':('parcels__section', + 'associated_file__parcels__section'), + 'parcel_1':('parcels__parcel_number', + 'associated_file__parcels__parcel_number'), }, ) show_operation = show_item(models.Operation, 'operation') diff --git a/ishtar_common/views.py b/ishtar_common/views.py index 42e826103..346edc0d7 100644 --- a/ishtar_common/views.py +++ b/ishtar_common/views.py @@ -229,7 +229,17 @@ def get_item(model, func_name, default_name, extra_request_keys=[], or_reqs = [] for req in dct.copy(): for k_hr in HIERARCHIC_FIELDS: - if req.endswith(k_hr + '__pk'): + if type(req) in (list, tuple): + val = dct.pop(req) + q = None + for idx, r in enumerate(req): + if not idx: + q = Q(**{r:val}) + else: + q = q | Q(**{r:val}) + or_reqs.append(q) + break + elif req.endswith(k_hr + '__pk'): val = dct.pop(req) reqs = Q(**{req:val}) req = req[:-2] + '__' -- cgit v1.2.3 From d989ccd4646601e6ae482dbbc2b4edc0db46481c Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Tue, 16 Apr 2013 23:44:01 +0200 Subject: Fix parcel search for archaeological files (closes #1215) --- archaeological_files/views.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/archaeological_files/views.py b/archaeological_files/views.py index 2b5003911..1919d8b62 100644 --- a/archaeological_files/views.py +++ b/archaeological_files/views.py @@ -61,8 +61,10 @@ def autocomplete_file(request): return HttpResponse(data, mimetype='text/plain') get_file = get_item(models.File, 'get_file', 'file', - extra_request_keys={'parcel_0':'operations__parcels__section', - 'parcel_1':'operations__parcels__parcel_number', + extra_request_keys={'parcel_0':('parcels__section', + 'operations__parcels__section'), + 'parcel_1':('parcels__parcel_number', + 'operations__parcels__parcel_number'), }, ) show_file = show_item(models.File, 'file') -- cgit v1.2.3