diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2009-01-27 01:05:20 +0000 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2009-01-27 01:05:20 +0000 |
| commit | 7e7aa8f7c28628c4b5de7428c33ed63552e8f5b9 (patch) | |
| tree | 8e3a9f3cd2449100e88e76f51ef30fa1c85c76e3 /lib/sqlalchemy | |
| parent | 332be8c3962f57206680b79f064b1995884a5a2d (diff) | |
| download | sqlalchemy-7e7aa8f7c28628c4b5de7428c33ed63552e8f5b9.tar.gz | |
- Query now implements __clause_element__() which produces
its selectable, which means a Query instance can be accepted
in many SQL expressions, including col.in_(query),
union(query1, query2), select([foo]).select_from(query),
etc.
- the __selectable__() interface has been replaced entirely
by __clause_element__().
Diffstat (limited to 'lib/sqlalchemy')
| -rw-r--r-- | lib/sqlalchemy/ext/sqlsoup.py | 6 | ||||
| -rw-r--r-- | lib/sqlalchemy/orm/query.py | 22 | ||||
| -rw-r--r-- | lib/sqlalchemy/sql/expression.py | 26 |
3 files changed, 26 insertions, 28 deletions
diff --git a/lib/sqlalchemy/ext/sqlsoup.py b/lib/sqlalchemy/ext/sqlsoup.py index 37b9d8fa8..f2754793b 100644 --- a/lib/sqlalchemy/ext/sqlsoup.py +++ b/lib/sqlalchemy/ext/sqlsoup.py @@ -397,7 +397,7 @@ class SelectableClassType(type): def update(cls, whereclause=None, values=None, **kwargs): _ddl_error(cls) - def __selectable__(cls): + def __clause_element__(cls): return cls._table def __getattr__(cls, attr): @@ -442,7 +442,7 @@ def _selectable_name(selectable): return x def class_for_table(selectable, **mapper_kwargs): - selectable = expression._selectable(selectable) + selectable = expression._clause_element_as_expr(selectable) mapname = 'Mapped' + _selectable_name(selectable) if isinstance(mapname, unicode): engine_encoding = selectable.metadata.bind.dialect.encoding @@ -531,7 +531,7 @@ class SqlSoup: def with_labels(self, item): # TODO give meaningful aliases - return self.map(expression._selectable(item).select(use_labels=True).alias('foo')) + return self.map(expression._clause_element_as_expr(item).select(use_labels=True).alias('foo')) def join(self, *args, **kwargs): j = join(*args, **kwargs) diff --git a/lib/sqlalchemy/orm/query.py b/lib/sqlalchemy/orm/query.py index 6a26d30b4..87019521b 100644 --- a/lib/sqlalchemy/orm/query.py +++ b/lib/sqlalchemy/orm/query.py @@ -320,18 +320,16 @@ class Query(object): return self._compile_context(labels=self._with_labels).statement._annotate({'_halt_adapt': True}) - @property - def _nested_statement(self): - return self.with_labels().enable_eagerloads(False).statement.correlate(None) - def subquery(self): """return the full SELECT statement represented by this Query, embedded within an Alias. Eager JOIN generation within the query is disabled. """ - return self.enable_eagerloads(False).statement.alias() + + def __clause_element__(self): + return self.enable_eagerloads(False).statement @_generative() def enable_eagerloads(self, value): @@ -554,7 +552,7 @@ class Query(object): those being selected. """ - fromclause = self._nested_statement + fromclause = self.with_labels().enable_eagerloads(False).statement.correlate(None) q = self._from_selectable(fromclause) if entities: q._set_entities(entities) @@ -748,7 +746,7 @@ class Query(object): """ return self._from_selectable( - expression.union(*([self._nested_statement]+ [x._nested_statement for x in q]))) + expression.union(*([self]+ list(q)))) def union_all(self, *q): """Produce a UNION ALL of this Query against one or more queries. @@ -758,7 +756,7 @@ class Query(object): """ return self._from_selectable( - expression.union_all(*([self._nested_statement]+ [x._nested_statement for x in q])) + expression.union_all(*([self]+ list(q))) ) def intersect(self, *q): @@ -769,7 +767,7 @@ class Query(object): """ return self._from_selectable( - expression.intersect(*([self._nested_statement]+ [x._nested_statement for x in q])) + expression.intersect(*([self]+ list(q))) ) def intersect_all(self, *q): @@ -780,7 +778,7 @@ class Query(object): """ return self._from_selectable( - expression.intersect_all(*([self._nested_statement]+ [x._nested_statement for x in q])) + expression.intersect_all(*([self]+ list(q))) ) def except_(self, *q): @@ -791,7 +789,7 @@ class Query(object): """ return self._from_selectable( - expression.except_(*([self._nested_statement]+ [x._nested_statement for x in q])) + expression.except_(*([self]+ list(q))) ) def except_all(self, *q): @@ -802,7 +800,7 @@ class Query(object): """ return self._from_selectable( - expression.except_all(*([self._nested_statement]+ [x._nested_statement for x in q])) + expression.except_all(*([self]+ list(q))) ) @util.accepts_a_list_as_starargs(list_deprecation='pending') diff --git a/lib/sqlalchemy/sql/expression.py b/lib/sqlalchemy/sql/expression.py index b3a7dd8e2..cacf7e7b9 100644 --- a/lib/sqlalchemy/sql/expression.py +++ b/lib/sqlalchemy/sql/expression.py @@ -923,6 +923,12 @@ def _literal_as_text(element): else: return element +def _clause_element_as_expr(element): + if hasattr(element, '__clause_element__'): + return element.__clause_element__() + else: + return element + def _literal_as_column(element): if hasattr(element, '__clause_element__'): return element.__clause_element__() @@ -959,14 +965,6 @@ def _corresponding_column_or_error(fromclause, column, require_embedded=False): % (column, getattr(column, 'table', None), fromclause.description)) return c -def _selectable(element): - if hasattr(element, '__selectable__'): - return element.__selectable__() - elif isinstance(element, Selectable): - return element - else: - raise exc.ArgumentError("Object %r is not a Selectable and does not implement `__selectable__()`" % element) - def is_column(col): """True if ``col`` is an instance of ``ColumnElement``.""" return isinstance(col, ColumnElement) @@ -1415,6 +1413,8 @@ class _CompareMixin(ColumnOperators): return self._in_impl(operators.in_op, operators.notin_op, other) def _in_impl(self, op, negate_op, seq_or_selectable): + seq_or_selectable = _clause_element_as_expr(seq_or_selectable) + if isinstance(seq_or_selectable, _ScalarSelect): return self.__compare( op, seq_or_selectable, negate=negate_op) @@ -2475,8 +2475,8 @@ class Join(FromClause): __visit_name__ = 'join' def __init__(self, left, right, onclause=None, isouter=False): - self.left = _selectable(left) - self.right = _selectable(right).self_group() + self.left = _literal_as_text(left) + self.right = _literal_as_text(right).self_group() if onclause is None: self.onclause = self._match_primaries(self.left, self.right) @@ -3103,6 +3103,8 @@ class CompoundSelect(_SelectBaseMixin, FromClause): # some DBs do not like ORDER BY in the inner queries of a UNION, etc. for n, s in enumerate(selects): + s = _clause_element_as_expr(s) + if not numcols: numcols = len(s.c) elif len(s.c) != numcols: @@ -3368,9 +3370,7 @@ class Select(_SelectBaseMixin, FromClause): """return a new select() construct with the given FROM expression applied to its list of FROM objects.""" - if _is_literal(fromclause): - fromclause = _TextClause(fromclause) - + fromclause = _literal_as_text(fromclause) self._froms = self._froms.union([fromclause]) @_generative |
