Ticket #6261: query_wherenode.diff

File query_wherenode.diff, 5.0 KB (added by jbronn, 16 years ago)

Patch adding customization options to the Query class.

  • branches/queryset-refactor/django/db/models/sql/query.py

     
    7878
    7979    alias_prefix = 'T'
    8080
    81     def __init__(self, model, connection):
     81    def __init__(self, model, connection, where=WhereNode):
    8282        self.model = model
    8383        self.connection = connection
    8484        self.alias_map = {}     # Maps alias to table name
    … …  
    9191        # SQL-related attributes
    9292        self.select = []
    9393        self.tables = []    # Aliases in the order they are created.
    94         self.where = WhereNode()
     94        self.where = where()
     95        self.where_class = where
    9596        self.group_by = []
    9697        self.having = []
    9798        self.order_by = []
    … …  
    156157        obj.select = self.select[:]
    157158        obj.tables = self.tables[:]
    158159        obj.where = copy.deepcopy(self.where)
     160        obj.where_class = self.where_class
    159161        obj.group_by = self.group_by[:]
    160162        obj.having = self.having[:]
    161163        obj.order_by = self.order_by[:]
    … …  
    192194        obj.clear_limits()
    193195        obj.select_related = False
    194196        if obj.distinct and len(obj.select) > 1:
    195             obj = self.clone(CountQuery, _query=obj, where=WhereNode(),
     197            obj = self.clone(CountQuery, _query=obj, where=self.where_class(),
    196198                    distinct=False)
    197199            obj.select = []
    198200            obj.extra_select = SortedDict()
    … …  
    210212
    211213        return number
    212214
     215    def get_terms(self):
     216        """
     217        Returns all valid Query lookup types.
     218        """
     219        return QUERY_TERMS
     220
    213221    def as_sql(self, with_limits=True):
    214222        """
    215223        Creates the SQL for this query. Returns the SQL string and list of
    … …  
    320328        elif self.where:
    321329            # rhs has an empty where clause. Make it match everything (see
    322330            # above for reasoning).
    323             w = WhereNode()
     331            w = self.where_class()
    324332            alias = self.join((None, self.model._meta.db_table, None, None))
    325333            pk = self.model._meta.pk
    326334            w.add([alias, pk.column, pk, 'isnull', False], AND)
    327335        else:
    328             w = WhereNode()
     336            w = self.where_class()
    329337        self.where.add(w, connector)
    330338
    331339        # Selection columns and extra extensions are those provided by 'rhs'.
    … …  
    705713            raise TypeError("Cannot parse keyword query %r" % arg)
    706714
    707715        # Work out the lookup type and remove it from 'parts', if necessary.
    708         if len(parts) == 1 or parts[-1] not in QUERY_TERMS:
     716        if len(parts) == 1 or parts[-1] not in self.get_terms():
    709717            lookup_type = 'exact'
    710718        else:
    711719            lookup_type = parts.pop()
    … …  
    11101118        for related in cls._meta.get_all_related_many_to_many_objects():
    11111119            if not isinstance(related.field, generic.GenericRelation):
    11121120                for offset in range(0, len(pk_list), GET_ITERATOR_CHUNK_SIZE):
    1113                     where = WhereNode()
     1121                    where = self.where_class()
    11141122                    where.add((None, related.field.m2m_reverse_name(),
    11151123                            related.field, 'in',
    11161124                            pk_list[offset : offset+GET_ITERATOR_CHUNK_SIZE]),
    … …  
    11181126                    self.do_query(related.field.m2m_db_table(), where)
    11191127
    11201128        for f in cls._meta.many_to_many:
    1121             w1 = WhereNode()
     1129            w1 = self.where_class()
    11221130            if isinstance(f, generic.GenericRelation):
    11231131                from django.contrib.contenttypes.models import ContentType
    11241132                field = f.rel.to._meta.get_field(f.content_type_field_name)
    11251133                w1.add((None, field.column, field, 'exact',
    11261134                        ContentType.objects.get_for_model(cls).id), AND)
    11271135            for offset in range(0, len(pk_list), GET_ITERATOR_CHUNK_SIZE):
    1128                 where = WhereNode()
     1136                where = self.where_class()
    11291137                where.add((None, f.m2m_column_name(), f, 'in',
    11301138                        pk_list[offset : offset + GET_ITERATOR_CHUNK_SIZE]),
    11311139                        AND)
    … …  
    11421150        lot of values in pk_list.
    11431151        """
    11441152        for offset in range(0, len(pk_list), GET_ITERATOR_CHUNK_SIZE):
    1145             where = WhereNode()
     1153            where = self.where_class()
    11461154            field = self.model._meta.pk
    11471155            where.add((None, field.column, field, 'in',
    11481156                    pk_list[offset : offset + GET_ITERATOR_CHUNK_SIZE]), AND)
    … …  
    11861194        This is used by the QuerySet.delete_objects() method.
    11871195        """
    11881196        for offset in range(0, len(pk_list), GET_ITERATOR_CHUNK_SIZE):
    1189             where = WhereNode()
     1197            where = self.where_class()
    11901198            f = self.model._meta.pk
    11911199            where.add((None, f.column, f, 'in',
    11921200                    pk_list[offset : offset + GET_ITERATOR_CHUNK_SIZE]),
Back to Top