1 # -*- coding: utf-8 -*-
4 # Copyright (c) 2009-2010, Bolloré telecom
7 # See AUTHORS file for a full list of contributors.
9 # Redistribution and use in source and binary forms, with or without modification,
10 # are permitted provided that the following conditions are met:
12 # 1. Redistributions of source code must retain the above copyright notice,
13 # this list of conditions and the following disclaimer.
15 # 2. Redistributions in binary form must reproduce the above copyright
16 # notice, this list of conditions and the following disclaimer in the
17 # documentation and/or other materials provided with the distribution.
19 # 3. Neither the name of Bolloré telecom nor the names of its contributors
20 # may be used to endorse or promote products derived from this software
21 # without specific prior written permission.
23 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
24 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
26 # DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
27 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
28 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
29 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
30 # ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
32 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35 from copy import deepcopy
38 from django.db import connections
39 from django.db.models.query import QuerySet as BaseQuerySet
40 from django.db.models.query_utils import Q
41 from django.db.models.sql import Query
42 from django.db.models.sql.where import WhereNode as BaseWhereNode, Constraint as BaseConstraint, AND, OR
44 from ldapdb.backends.ldap import compiler
45 from ldapdb.models.fields import CharField
47 class Constraint(BaseConstraint):
49 An object that can be passed to WhereNode.add() and knows how to
50 pre-process itself prior to including in the WhereNode.
53 - we redefine this class, because when self.field is None calls
54 Field().get_db_prep_lookup(), which short-circuits our LDAP-specific code.
56 def process(self, lookup_type, value, connection):
58 Returns a tuple of data suitable for inclusion in a WhereNode
61 # Because of circular imports, we need to import this here.
62 from django.db.models.base import ObjectDoesNotExist
66 params = self.field.get_db_prep_lookup(lookup_type, value,
67 connection=connection, prepared=True)
68 db_type = self.field.db_type()
70 params = CharField().get_db_prep_lookup(lookup_type, value,
71 connection=connection, prepared=True)
73 except ObjectDoesNotExist:
74 raise EmptyShortCircuit
76 return (self.alias, self.col, db_type), params
78 class WhereNode(BaseWhereNode):
79 def add(self, data, connector):
80 if not isinstance(data, (list, tuple)):
81 super(WhereNode, self).add(data, connector)
84 # we replace the native Constraint by our own
85 obj, lookup_type, value = data
86 if hasattr(obj, "process"):
87 obj = Constraint(obj.alias, obj.col, obj.field)
88 super(WhereNode, self).add((obj, lookup_type, value), connector)
90 class QuerySet(BaseQuerySet):
91 def __init__(self, model=None, query=None, using=None):
93 query = Query(model, WhereNode)
94 super(QuerySet, self).__init__(model=model, query=query, using=using)