Continued models/__init__.py breakup
git-svn-id: http://code.djangoproject.com/svn/django/branches/magic-removal@1694 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Robert Wittams
parent
9394d004db
commit
e5b3fd474b
|
|
@ -9,6 +9,7 @@ import copy, datetime, os, re, sys, types
|
||||||
|
|
||||||
from django.db.models.loading import get_installed_models, get_installed_model_modules
|
from django.db.models.loading import get_installed_models, get_installed_model_modules
|
||||||
from django.db.models.manipulators import ManipulatorDescriptor, ModelAddManipulator, ModelChangeManipulator
|
from django.db.models.manipulators import ManipulatorDescriptor, ModelAddManipulator, ModelChangeManipulator
|
||||||
|
from django.db.models.query import Q, parse_lookup, get_where_clause, get_cached_row, fill_table_cache, throw_bad_kwarg_error
|
||||||
|
|
||||||
# Admin stages.
|
# Admin stages.
|
||||||
ADD, CHANGE, BOTH = 1, 2, 3
|
ADD, CHANGE, BOTH = 1, 2, 3
|
||||||
|
|
@ -28,7 +29,7 @@ MODEL_FUNCTIONS_PREFIX = '_module_'
|
||||||
# converted to manipulator methods.
|
# converted to manipulator methods.
|
||||||
MANIPULATOR_FUNCTIONS_PREFIX = '_manipulator_'
|
MANIPULATOR_FUNCTIONS_PREFIX = '_manipulator_'
|
||||||
|
|
||||||
LOOKUP_SEPARATOR = '__'
|
|
||||||
|
|
||||||
####################
|
####################
|
||||||
# HELPER FUNCTIONS #
|
# HELPER FUNCTIONS #
|
||||||
|
|
@ -247,80 +248,6 @@ class RelatedObject(object):
|
||||||
rel_obj_name = '%s_%s' % (self.opts.app_label, rel_obj_name)
|
rel_obj_name = '%s_%s' % (self.opts.app_label, rel_obj_name)
|
||||||
return rel_obj_name
|
return rel_obj_name
|
||||||
|
|
||||||
class QBase:
|
|
||||||
"Base class for QAnd and QOr"
|
|
||||||
def __init__(self, *args):
|
|
||||||
self.args = args
|
|
||||||
|
|
||||||
def __repr__(self):
|
|
||||||
return '(%s)' % self.operator.join([repr(el) for el in self.args])
|
|
||||||
|
|
||||||
def get_sql(self, opts, table_count):
|
|
||||||
tables, join_where, where, params = [], [], [], []
|
|
||||||
for val in self.args:
|
|
||||||
tables2, join_where2, where2, params2, table_count = val.get_sql(opts, table_count)
|
|
||||||
tables.extend(tables2)
|
|
||||||
join_where.extend(join_where2)
|
|
||||||
where.extend(where2)
|
|
||||||
params.extend(params2)
|
|
||||||
return tables, join_where, ['(%s)' % self.operator.join(where)], params, table_count
|
|
||||||
|
|
||||||
class QAnd(QBase):
|
|
||||||
"Encapsulates a combined query that uses 'AND'."
|
|
||||||
operator = ' AND '
|
|
||||||
def __or__(self, other):
|
|
||||||
if isinstance(other, (QAnd, QOr, Q)):
|
|
||||||
return QOr(self, other)
|
|
||||||
else:
|
|
||||||
raise TypeError, other
|
|
||||||
|
|
||||||
def __and__(self, other):
|
|
||||||
if isinstance(other, QAnd):
|
|
||||||
return QAnd(*(self.args+other.args))
|
|
||||||
elif isinstance(other, (Q, QOr)):
|
|
||||||
return QAnd(*(self.args+(other,)))
|
|
||||||
else:
|
|
||||||
raise TypeError, other
|
|
||||||
|
|
||||||
class QOr(QBase):
|
|
||||||
"Encapsulates a combined query that uses 'OR'."
|
|
||||||
operator = ' OR '
|
|
||||||
def __and__(self, other):
|
|
||||||
if isinstance(other, (QAnd, QOr, Q)):
|
|
||||||
return QAnd(self, other)
|
|
||||||
else:
|
|
||||||
raise TypeError, other
|
|
||||||
|
|
||||||
def __or__(self, other):
|
|
||||||
if isinstance(other, QOr):
|
|
||||||
return QOr(*(self.args+other.args))
|
|
||||||
elif isinstance(other, (Q, QAnd)):
|
|
||||||
return QOr(*(self.args+(other,)))
|
|
||||||
else:
|
|
||||||
raise TypeError, other
|
|
||||||
|
|
||||||
class Q:
|
|
||||||
"Encapsulates queries for the 'complex' parameter to Django API functions."
|
|
||||||
def __init__(self, **kwargs):
|
|
||||||
self.kwargs = kwargs
|
|
||||||
|
|
||||||
def __repr__(self):
|
|
||||||
return 'Q%r' % self.kwargs
|
|
||||||
|
|
||||||
def __and__(self, other):
|
|
||||||
if isinstance(other, (Q, QAnd, QOr)):
|
|
||||||
return QAnd(self, other)
|
|
||||||
else:
|
|
||||||
raise TypeError, other
|
|
||||||
|
|
||||||
def __or__(self, other):
|
|
||||||
if isinstance(other, (Q, QAnd, QOr)):
|
|
||||||
return QOr(self, other)
|
|
||||||
else:
|
|
||||||
raise TypeError, other
|
|
||||||
|
|
||||||
def get_sql(self, opts, table_count):
|
|
||||||
return _parse_lookup(self.kwargs.items(), opts, table_count)
|
|
||||||
|
|
||||||
class Options:
|
class Options:
|
||||||
def __init__(self, module_name='', verbose_name='', verbose_name_plural='', db_table='',
|
def __init__(self, module_name='', verbose_name='', verbose_name_plural='', db_table='',
|
||||||
|
|
@ -554,7 +481,7 @@ class Manager(object):
|
||||||
params = kwargs.get('params') and kwargs['params'][:] or []
|
params = kwargs.get('params') and kwargs['params'][:] or []
|
||||||
|
|
||||||
# Convert the kwargs into SQL.
|
# Convert the kwargs into SQL.
|
||||||
tables2, join_where2, where2, params2, _ = _parse_lookup(kwargs.items(), opts)
|
tables2, join_where2, where2, params2, _ = parse_lookup(kwargs.items(), opts)
|
||||||
tables.extend(tables2)
|
tables.extend(tables2)
|
||||||
where.extend(join_where2 + where2)
|
where.extend(join_where2 + where2)
|
||||||
params.extend(params2)
|
params.extend(params2)
|
||||||
|
|
@ -564,7 +491,7 @@ class Manager(object):
|
||||||
|
|
||||||
# Add additional tables and WHERE clauses based on select_related.
|
# Add additional tables and WHERE clauses based on select_related.
|
||||||
if kwargs.get('select_related') is True:
|
if kwargs.get('select_related') is True:
|
||||||
_fill_table_cache(opts, select, tables, where, opts.db_table, [opts.db_table])
|
fill_table_cache(opts, select, tables, where, opts.db_table, [opts.db_table])
|
||||||
|
|
||||||
# Add any additional SELECTs passed in via kwargs.
|
# Add any additional SELECTs passed in via kwargs.
|
||||||
if kwargs.get('select'):
|
if kwargs.get('select'):
|
||||||
|
|
@ -620,7 +547,7 @@ class Manager(object):
|
||||||
raise StopIteration
|
raise StopIteration
|
||||||
for row in rows:
|
for row in rows:
|
||||||
if fill_cache:
|
if fill_cache:
|
||||||
obj, index_end = _get_cached_row(self.klass, row, 0)
|
obj, index_end = get_cached_row(self.klass, row, 0)
|
||||||
else:
|
else:
|
||||||
obj = self.klass(*row[:index_end])
|
obj = self.klass(*row[:index_end])
|
||||||
for i, k in enumerate(kwargs['select']):
|
for i, k in enumerate(kwargs['select']):
|
||||||
|
|
@ -1275,184 +1202,4 @@ def method_get_order(ordered_obj, self):
|
||||||
def get_absolute_url(opts, func, self):
|
def get_absolute_url(opts, func, self):
|
||||||
return settings.ABSOLUTE_URL_OVERRIDES.get('%s.%s' % (opts.app_label, opts.module_name), func)(self)
|
return settings.ABSOLUTE_URL_OVERRIDES.get('%s.%s' % (opts.app_label, opts.module_name), func)(self)
|
||||||
|
|
||||||
def _get_where_clause(lookup_type, table_prefix, field_name, value):
|
|
||||||
if table_prefix.endswith('.'):
|
|
||||||
table_prefix = backend.quote_name(table_prefix[:-1])+'.'
|
|
||||||
field_name = backend.quote_name(field_name)
|
|
||||||
try:
|
|
||||||
return '%s%s %s' % (table_prefix, field_name, (backend.OPERATOR_MAPPING[lookup_type] % '%s'))
|
|
||||||
except KeyError:
|
|
||||||
pass
|
|
||||||
if lookup_type == 'in':
|
|
||||||
return '%s%s IN (%s)' % (table_prefix, field_name, ','.join(['%s' for v in value]))
|
|
||||||
elif lookup_type in ('range', 'year'):
|
|
||||||
return '%s%s BETWEEN %%s AND %%s' % (table_prefix, field_name)
|
|
||||||
elif lookup_type in ('month', 'day'):
|
|
||||||
return "%s = %%s" % backend.get_date_extract_sql(lookup_type, table_prefix + field_name)
|
|
||||||
elif lookup_type == 'isnull':
|
|
||||||
return "%s%s IS %sNULL" % (table_prefix, field_name, (not value and 'NOT ' or ''))
|
|
||||||
raise TypeError, "Got invalid lookup_type: %s" % repr(lookup_type)
|
|
||||||
|
|
||||||
def _get_cached_row(klass, row, index_start):
|
|
||||||
"Helper function that recursively returns an object with cache filled"
|
|
||||||
index_end = index_start + len(klass._meta.fields)
|
|
||||||
obj = klass(*row[index_start:index_end])
|
|
||||||
for f in klass._meta.fields:
|
|
||||||
if f.rel and not f.null:
|
|
||||||
rel_obj, index_end = _get_cached_row(f.rel.to, row, index_end)
|
|
||||||
setattr(obj, f.get_cache_name(), rel_obj)
|
|
||||||
return obj, index_end
|
|
||||||
|
|
||||||
def _fill_table_cache(opts, select, tables, where, old_prefix, cache_tables_seen):
|
|
||||||
"""
|
|
||||||
Helper function that recursively populates the select, tables and where (in
|
|
||||||
place) for fill-cache queries.
|
|
||||||
"""
|
|
||||||
for f in opts.fields:
|
|
||||||
if f.rel and not f.null:
|
|
||||||
db_table = f.rel.to._meta.db_table
|
|
||||||
if db_table not in cache_tables_seen:
|
|
||||||
tables.append(backend.quote_name(db_table))
|
|
||||||
else: # The table was already seen, so give it a table alias.
|
|
||||||
new_prefix = '%s%s' % (db_table, len(cache_tables_seen))
|
|
||||||
tables.append('%s %s' % (backend.quote_name(db_table), backend.quote_name(new_prefix)))
|
|
||||||
db_table = new_prefix
|
|
||||||
cache_tables_seen.append(db_table)
|
|
||||||
where.append('%s.%s = %s.%s' % \
|
|
||||||
(backend.quote_name(old_prefix), backend.quote_name(f.column),
|
|
||||||
backend.quote_name(db_table), backend.quote_name(f.rel.get_related_field().column)))
|
|
||||||
select.extend(['%s.%s' % (backend.quote_name(db_table), backend.quote_name(f2.column)) for f2 in f.rel.to._meta.fields])
|
|
||||||
_fill_table_cache(f.rel.to._meta, select, tables, where, db_table, cache_tables_seen)
|
|
||||||
|
|
||||||
def _throw_bad_kwarg_error(kwarg):
|
|
||||||
# Helper function to remove redundancy.
|
|
||||||
raise TypeError, "got unexpected keyword argument '%s'" % kwarg
|
|
||||||
|
|
||||||
def _parse_lookup(kwarg_items, opts, table_count=0):
|
|
||||||
# Helper function that handles converting API kwargs (e.g.
|
|
||||||
# "name__exact": "tom") to SQL.
|
|
||||||
|
|
||||||
# Note that there is a distinction between where and join_where. The latter
|
|
||||||
# is specifically a list of where clauses to use for JOINs. This
|
|
||||||
# distinction is necessary because of support for "_or".
|
|
||||||
|
|
||||||
# table_count is used to ensure table aliases are unique.
|
|
||||||
tables, join_where, where, params = [], [], [], []
|
|
||||||
for kwarg, kwarg_value in kwarg_items:
|
|
||||||
if kwarg in ('order_by', 'limit', 'offset', 'select_related', 'distinct', 'select', 'tables', 'where', 'params'):
|
|
||||||
continue
|
|
||||||
if kwarg_value is None:
|
|
||||||
continue
|
|
||||||
if kwarg == 'complex':
|
|
||||||
tables2, join_where2, where2, params2, table_count = kwarg_value.get_sql(opts, table_count)
|
|
||||||
tables.extend(tables2)
|
|
||||||
join_where.extend(join_where2)
|
|
||||||
where.extend(where2)
|
|
||||||
params.extend(params2)
|
|
||||||
continue
|
|
||||||
if kwarg == '_or':
|
|
||||||
for val in kwarg_value:
|
|
||||||
tables2, join_where2, where2, params2, table_count = _parse_lookup(val, opts, table_count)
|
|
||||||
tables.extend(tables2)
|
|
||||||
join_where.extend(join_where2)
|
|
||||||
where.append('(%s)' % ' OR '.join(where2))
|
|
||||||
params.extend(params2)
|
|
||||||
continue
|
|
||||||
lookup_list = kwarg.split(LOOKUP_SEPARATOR)
|
|
||||||
# pk="value" is shorthand for (primary key)__exact="value"
|
|
||||||
if lookup_list[-1] == 'pk':
|
|
||||||
if opts.pk.rel:
|
|
||||||
lookup_list = lookup_list[:-1] + [opts.pk.name, opts.pk.rel.field_name, 'exact']
|
|
||||||
else:
|
|
||||||
lookup_list = lookup_list[:-1] + [opts.pk.name, 'exact']
|
|
||||||
if len(lookup_list) == 1:
|
|
||||||
_throw_bad_kwarg_error(kwarg)
|
|
||||||
lookup_type = lookup_list.pop()
|
|
||||||
current_opts = opts # We'll be overwriting this, so keep a reference to the original opts.
|
|
||||||
current_table_alias = current_opts.db_table
|
|
||||||
param_required = False
|
|
||||||
while lookup_list or param_required:
|
|
||||||
table_count += 1
|
|
||||||
try:
|
|
||||||
# "current" is a piece of the lookup list. For example, in
|
|
||||||
# choices.get_list(poll__sites__id__exact=5), lookup_list is
|
|
||||||
# ["polls", "sites", "id"], and the first current is "polls".
|
|
||||||
try:
|
|
||||||
current = lookup_list.pop(0)
|
|
||||||
except IndexError:
|
|
||||||
# If we're here, lookup_list is empty but param_required
|
|
||||||
# is set to True, which means the kwarg was bad.
|
|
||||||
# Example: choices.get_list(poll__exact='foo')
|
|
||||||
_throw_bad_kwarg_error(kwarg)
|
|
||||||
# Try many-to-many relationships first...
|
|
||||||
for f in current_opts.many_to_many:
|
|
||||||
if f.name == current:
|
|
||||||
rel_table_alias = backend.quote_name('t%s' % table_count)
|
|
||||||
table_count += 1
|
|
||||||
tables.append('%s %s' % \
|
|
||||||
(backend.quote_name(f.get_m2m_db_table(current_opts)), rel_table_alias))
|
|
||||||
join_where.append('%s.%s = %s.%s' % \
|
|
||||||
(backend.quote_name(current_table_alias),
|
|
||||||
backend.quote_name(current_opts.pk.column),
|
|
||||||
rel_table_alias,
|
|
||||||
backend.quote_name(current_opts.object_name.lower() + '_id')))
|
|
||||||
# Optimization: In the case of primary-key lookups, we
|
|
||||||
# don't have to do an extra join.
|
|
||||||
if lookup_list and lookup_list[0] == f.rel.to._meta.pk.name and lookup_type == 'exact':
|
|
||||||
where.append(_get_where_clause(lookup_type, rel_table_alias+'.',
|
|
||||||
f.rel.to._meta.object_name.lower()+'_id', kwarg_value))
|
|
||||||
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
|
||||||
lookup_list.pop()
|
|
||||||
param_required = False
|
|
||||||
else:
|
|
||||||
new_table_alias = 't%s' % table_count
|
|
||||||
tables.append('%s %s' % (backend.quote_name(f.rel.to._meta.db_table),
|
|
||||||
backend.quote_name(new_table_alias)))
|
|
||||||
join_where.append('%s.%s = %s.%s' % \
|
|
||||||
(backend.quote_name(rel_table_alias),
|
|
||||||
backend.quote_name(f.rel.to._meta.object_name.lower() + '_id'),
|
|
||||||
backend.quote_name(new_table_alias),
|
|
||||||
backend.quote_name(f.rel.to._meta.pk.column)))
|
|
||||||
current_table_alias = new_table_alias
|
|
||||||
param_required = True
|
|
||||||
current_opts = f.rel.to._meta
|
|
||||||
raise StopIteration
|
|
||||||
for f in current_opts.fields:
|
|
||||||
# Try many-to-one relationships...
|
|
||||||
if f.rel and f.name == current:
|
|
||||||
# Optimization: In the case of primary-key lookups, we
|
|
||||||
# don't have to do an extra join.
|
|
||||||
if lookup_list and lookup_list[0] == f.rel.to._meta.pk.name and lookup_type == 'exact':
|
|
||||||
where.append(_get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value))
|
|
||||||
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
|
||||||
lookup_list.pop()
|
|
||||||
param_required = False
|
|
||||||
# 'isnull' lookups in many-to-one relationships are a special case,
|
|
||||||
# because we don't want to do a join. We just want to find out
|
|
||||||
# whether the foreign key field is NULL.
|
|
||||||
elif lookup_type == 'isnull' and not lookup_list:
|
|
||||||
where.append(_get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value))
|
|
||||||
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
|
||||||
else:
|
|
||||||
new_table_alias = 't%s' % table_count
|
|
||||||
tables.append('%s %s' % \
|
|
||||||
(backend.quote_name(f.rel.to._meta.db_table), backend.quote_name(new_table_alias)))
|
|
||||||
join_where.append('%s.%s = %s.%s' % \
|
|
||||||
(backend.quote_name(current_table_alias), backend.quote_name(f.column),
|
|
||||||
backend.quote_name(new_table_alias), backend.quote_name(f.rel.to._meta.pk.column)))
|
|
||||||
current_table_alias = new_table_alias
|
|
||||||
param_required = True
|
|
||||||
current_opts = f.rel.to._meta
|
|
||||||
raise StopIteration
|
|
||||||
# Try direct field-name lookups...
|
|
||||||
if f.name == current:
|
|
||||||
where.append(_get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value))
|
|
||||||
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
|
||||||
param_required = False
|
|
||||||
raise StopIteration
|
|
||||||
# If we haven't hit StopIteration at this point, "current" must be
|
|
||||||
# an invalid lookup, so raise an exception.
|
|
||||||
_throw_bad_kwarg_error(kwarg)
|
|
||||||
except StopIteration:
|
|
||||||
continue
|
|
||||||
return tables, join_where, where, params, table_count
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,261 @@
|
||||||
|
from django.db import backend, connection
|
||||||
|
|
||||||
|
LOOKUP_SEPARATOR = '__'
|
||||||
|
|
||||||
|
class QBase:
|
||||||
|
"Base class for QAnd and QOr"
|
||||||
|
def __init__(self, *args):
|
||||||
|
self.args = args
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return '(%s)' % self.operator.join([repr(el) for el in self.args])
|
||||||
|
|
||||||
|
def get_sql(self, opts, table_count):
|
||||||
|
tables, join_where, where, params = [], [], [], []
|
||||||
|
for val in self.args:
|
||||||
|
tables2, join_where2, where2, params2, table_count = val.get_sql(opts, table_count)
|
||||||
|
tables.extend(tables2)
|
||||||
|
join_where.extend(join_where2)
|
||||||
|
where.extend(where2)
|
||||||
|
params.extend(params2)
|
||||||
|
return tables, join_where, ['(%s)' % self.operator.join(where)], params, table_count
|
||||||
|
|
||||||
|
class QAnd(QBase):
|
||||||
|
"Encapsulates a combined query that uses 'AND'."
|
||||||
|
operator = ' AND '
|
||||||
|
def __or__(self, other):
|
||||||
|
if isinstance(other, (QAnd, QOr, Q)):
|
||||||
|
return QOr(self, other)
|
||||||
|
else:
|
||||||
|
raise TypeError, other
|
||||||
|
|
||||||
|
def __and__(self, other):
|
||||||
|
if isinstance(other, QAnd):
|
||||||
|
return QAnd(*(self.args+other.args))
|
||||||
|
elif isinstance(other, (Q, QOr)):
|
||||||
|
return QAnd(*(self.args+(other,)))
|
||||||
|
else:
|
||||||
|
raise TypeError, other
|
||||||
|
|
||||||
|
class QOr(QBase):
|
||||||
|
"Encapsulates a combined query that uses 'OR'."
|
||||||
|
operator = ' OR '
|
||||||
|
def __and__(self, other):
|
||||||
|
if isinstance(other, (QAnd, QOr, Q)):
|
||||||
|
return QAnd(self, other)
|
||||||
|
else:
|
||||||
|
raise TypeError, other
|
||||||
|
|
||||||
|
def __or__(self, other):
|
||||||
|
if isinstance(other, QOr):
|
||||||
|
return QOr(*(self.args+other.args))
|
||||||
|
elif isinstance(other, (Q, QAnd)):
|
||||||
|
return QOr(*(self.args+(other,)))
|
||||||
|
else:
|
||||||
|
raise TypeError, other
|
||||||
|
|
||||||
|
class Q:
|
||||||
|
"Encapsulates queries for the 'complex' parameter to Django API functions."
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
self.kwargs = kwargs
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return 'Q%r' % self.kwargs
|
||||||
|
|
||||||
|
def __and__(self, other):
|
||||||
|
if isinstance(other, (Q, QAnd, QOr)):
|
||||||
|
return QAnd(self, other)
|
||||||
|
else:
|
||||||
|
raise TypeError, other
|
||||||
|
|
||||||
|
def __or__(self, other):
|
||||||
|
if isinstance(other, (Q, QAnd, QOr)):
|
||||||
|
return QOr(self, other)
|
||||||
|
else:
|
||||||
|
raise TypeError, other
|
||||||
|
|
||||||
|
def get_sql(self, opts, table_count):
|
||||||
|
return parse_lookup(self.kwargs.items(), opts, table_count)
|
||||||
|
|
||||||
|
|
||||||
|
def get_where_clause(lookup_type, table_prefix, field_name, value):
|
||||||
|
if table_prefix.endswith('.'):
|
||||||
|
table_prefix = backend.quote_name(table_prefix[:-1])+'.'
|
||||||
|
field_name = backend.quote_name(field_name)
|
||||||
|
try:
|
||||||
|
return '%s%s %s' % (table_prefix, field_name, (backend.OPERATOR_MAPPING[lookup_type] % '%s'))
|
||||||
|
except KeyError:
|
||||||
|
pass
|
||||||
|
if lookup_type == 'in':
|
||||||
|
return '%s%s IN (%s)' % (table_prefix, field_name, ','.join(['%s' for v in value]))
|
||||||
|
elif lookup_type in ('range', 'year'):
|
||||||
|
return '%s%s BETWEEN %%s AND %%s' % (table_prefix, field_name)
|
||||||
|
elif lookup_type in ('month', 'day'):
|
||||||
|
return "%s = %%s" % backend.get_date_extract_sql(lookup_type, table_prefix + field_name)
|
||||||
|
elif lookup_type == 'isnull':
|
||||||
|
return "%s%s IS %sNULL" % (table_prefix, field_name, (not value and 'NOT ' or ''))
|
||||||
|
raise TypeError, "Got invalid lookup_type: %s" % repr(lookup_type)
|
||||||
|
|
||||||
|
def get_cached_row(klass, row, index_start):
|
||||||
|
"Helper function that recursively returns an object with cache filled"
|
||||||
|
index_end = index_start + len(klass._meta.fields)
|
||||||
|
obj = klass(*row[index_start:index_end])
|
||||||
|
for f in klass._meta.fields:
|
||||||
|
if f.rel and not f.null:
|
||||||
|
rel_obj, index_end = get_cached_row(f.rel.to, row, index_end)
|
||||||
|
setattr(obj, f.get_cache_name(), rel_obj)
|
||||||
|
return obj, index_end
|
||||||
|
|
||||||
|
def fill_table_cache(opts, select, tables, where, old_prefix, cache_tables_seen):
|
||||||
|
"""
|
||||||
|
Helper function that recursively populates the select, tables and where (in
|
||||||
|
place) for fill-cache queries.
|
||||||
|
"""
|
||||||
|
for f in opts.fields:
|
||||||
|
if f.rel and not f.null:
|
||||||
|
db_table = f.rel.to._meta.db_table
|
||||||
|
if db_table not in cache_tables_seen:
|
||||||
|
tables.append(backend.quote_name(db_table))
|
||||||
|
else: # The table was already seen, so give it a table alias.
|
||||||
|
new_prefix = '%s%s' % (db_table, len(cache_tables_seen))
|
||||||
|
tables.append('%s %s' % (backend.quote_name(db_table), backend.quote_name(new_prefix)))
|
||||||
|
db_table = new_prefix
|
||||||
|
cache_tables_seen.append(db_table)
|
||||||
|
where.append('%s.%s = %s.%s' % \
|
||||||
|
(backend.quote_name(old_prefix), backend.quote_name(f.column),
|
||||||
|
backend.quote_name(db_table), backend.quote_name(f.rel.get_related_field().column)))
|
||||||
|
select.extend(['%s.%s' % (backend.quote_name(db_table), backend.quote_name(f2.column)) for f2 in f.rel.to._meta.fields])
|
||||||
|
fill_table_cache(f.rel.to._meta, select, tables, where, db_table, cache_tables_seen)
|
||||||
|
|
||||||
|
def throw_bad_kwarg_error(kwarg):
|
||||||
|
# Helper function to remove redundancy.
|
||||||
|
raise TypeError, "got unexpected keyword argument '%s'" % kwarg
|
||||||
|
|
||||||
|
def parse_lookup(kwarg_items, opts, table_count=0):
|
||||||
|
# Helper function that handles converting API kwargs (e.g.
|
||||||
|
# "name__exact": "tom") to SQL.
|
||||||
|
|
||||||
|
# Note that there is a distinction between where and join_where. The latter
|
||||||
|
# is specifically a list of where clauses to use for JOINs. This
|
||||||
|
# distinction is necessary because of support for "_or".
|
||||||
|
|
||||||
|
# table_count is used to ensure table aliases are unique.
|
||||||
|
tables, join_where, where, params = [], [], [], []
|
||||||
|
for kwarg, kwarg_value in kwarg_items:
|
||||||
|
if kwarg in ('order_by', 'limit', 'offset', 'select_related', 'distinct', 'select', 'tables', 'where', 'params'):
|
||||||
|
continue
|
||||||
|
if kwarg_value is None:
|
||||||
|
continue
|
||||||
|
if kwarg == 'complex':
|
||||||
|
tables2, join_where2, where2, params2, table_count = kwarg_value.get_sql(opts, table_count)
|
||||||
|
tables.extend(tables2)
|
||||||
|
join_where.extend(join_where2)
|
||||||
|
where.extend(where2)
|
||||||
|
params.extend(params2)
|
||||||
|
continue
|
||||||
|
if kwarg == '_or':
|
||||||
|
for val in kwarg_value:
|
||||||
|
tables2, join_where2, where2, params2, table_count = parse_lookup(val, opts, table_count)
|
||||||
|
tables.extend(tables2)
|
||||||
|
join_where.extend(join_where2)
|
||||||
|
where.append('(%s)' % ' OR '.join(where2))
|
||||||
|
params.extend(params2)
|
||||||
|
continue
|
||||||
|
lookup_list = kwarg.split(LOOKUP_SEPARATOR)
|
||||||
|
# pk="value" is shorthand for (primary key)__exact="value"
|
||||||
|
if lookup_list[-1] == 'pk':
|
||||||
|
if opts.pk.rel:
|
||||||
|
lookup_list = lookup_list[:-1] + [opts.pk.name, opts.pk.rel.field_name, 'exact']
|
||||||
|
else:
|
||||||
|
lookup_list = lookup_list[:-1] + [opts.pk.name, 'exact']
|
||||||
|
if len(lookup_list) == 1:
|
||||||
|
_throw_bad_kwarg_error(kwarg)
|
||||||
|
lookup_type = lookup_list.pop()
|
||||||
|
current_opts = opts # We'll be overwriting this, so keep a reference to the original opts.
|
||||||
|
current_table_alias = current_opts.db_table
|
||||||
|
param_required = False
|
||||||
|
while lookup_list or param_required:
|
||||||
|
table_count += 1
|
||||||
|
try:
|
||||||
|
# "current" is a piece of the lookup list. For example, in
|
||||||
|
# choices.get_list(poll__sites__id__exact=5), lookup_list is
|
||||||
|
# ["polls", "sites", "id"], and the first current is "polls".
|
||||||
|
try:
|
||||||
|
current = lookup_list.pop(0)
|
||||||
|
except IndexError:
|
||||||
|
# If we're here, lookup_list is empty but param_required
|
||||||
|
# is set to True, which means the kwarg was bad.
|
||||||
|
# Example: choices.get_list(poll__exact='foo')
|
||||||
|
throw_bad_kwarg_error(kwarg)
|
||||||
|
# Try many-to-many relationships first...
|
||||||
|
for f in current_opts.many_to_many:
|
||||||
|
if f.name == current:
|
||||||
|
rel_table_alias = backend.quote_name('t%s' % table_count)
|
||||||
|
table_count += 1
|
||||||
|
tables.append('%s %s' % \
|
||||||
|
(backend.quote_name(f.get_m2m_db_table(current_opts)), rel_table_alias))
|
||||||
|
join_where.append('%s.%s = %s.%s' % \
|
||||||
|
(backend.quote_name(current_table_alias),
|
||||||
|
backend.quote_name(current_opts.pk.column),
|
||||||
|
rel_table_alias,
|
||||||
|
backend.quote_name(current_opts.object_name.lower() + '_id')))
|
||||||
|
# Optimization: In the case of primary-key lookups, we
|
||||||
|
# don't have to do an extra join.
|
||||||
|
if lookup_list and lookup_list[0] == f.rel.to._meta.pk.name and lookup_type == 'exact':
|
||||||
|
where.append(get_where_clause(lookup_type, rel_table_alias+'.',
|
||||||
|
f.rel.to._meta.object_name.lower()+'_id', kwarg_value))
|
||||||
|
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
||||||
|
lookup_list.pop()
|
||||||
|
param_required = False
|
||||||
|
else:
|
||||||
|
new_table_alias = 't%s' % table_count
|
||||||
|
tables.append('%s %s' % (backend.quote_name(f.rel.to._meta.db_table),
|
||||||
|
backend.quote_name(new_table_alias)))
|
||||||
|
join_where.append('%s.%s = %s.%s' % \
|
||||||
|
(backend.quote_name(rel_table_alias),
|
||||||
|
backend.quote_name(f.rel.to._meta.object_name.lower() + '_id'),
|
||||||
|
backend.quote_name(new_table_alias),
|
||||||
|
backend.quote_name(f.rel.to._meta.pk.column)))
|
||||||
|
current_table_alias = new_table_alias
|
||||||
|
param_required = True
|
||||||
|
current_opts = f.rel.to._meta
|
||||||
|
raise StopIteration
|
||||||
|
for f in current_opts.fields:
|
||||||
|
# Try many-to-one relationships...
|
||||||
|
if f.rel and f.name == current:
|
||||||
|
# Optimization: In the case of primary-key lookups, we
|
||||||
|
# don't have to do an extra join.
|
||||||
|
if lookup_list and lookup_list[0] == f.rel.to._meta.pk.name and lookup_type == 'exact':
|
||||||
|
where.append(get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value))
|
||||||
|
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
||||||
|
lookup_list.pop()
|
||||||
|
param_required = False
|
||||||
|
# 'isnull' lookups in many-to-one relationships are a special case,
|
||||||
|
# because we don't want to do a join. We just want to find out
|
||||||
|
# whether the foreign key field is NULL.
|
||||||
|
elif lookup_type == 'isnull' and not lookup_list:
|
||||||
|
where.append(get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value))
|
||||||
|
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
||||||
|
else:
|
||||||
|
new_table_alias = 't%s' % table_count
|
||||||
|
tables.append('%s %s' % \
|
||||||
|
(backend.quote_name(f.rel.to._meta.db_table), backend.quote_name(new_table_alias)))
|
||||||
|
join_where.append('%s.%s = %s.%s' % \
|
||||||
|
(backend.quote_name(current_table_alias), backend.quote_name(f.column),
|
||||||
|
backend.quote_name(new_table_alias), backend.quote_name(f.rel.to._meta.pk.column)))
|
||||||
|
current_table_alias = new_table_alias
|
||||||
|
param_required = True
|
||||||
|
current_opts = f.rel.to._meta
|
||||||
|
raise StopIteration
|
||||||
|
# Try direct field-name lookups...
|
||||||
|
if f.name == current:
|
||||||
|
where.append(get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value))
|
||||||
|
params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value))
|
||||||
|
param_required = False
|
||||||
|
raise StopIteration
|
||||||
|
# If we haven't hit StopIteration at this point, "current" must be
|
||||||
|
# an invalid lookup, so raise an exception.
|
||||||
|
throw_bad_kwarg_error(kwarg)
|
||||||
|
except StopIteration:
|
||||||
|
continue
|
||||||
|
return tables, join_where, where, params, table_count
|
||||||
Loading…
Reference in New Issue