diff options
Diffstat (limited to 'lib/sqlalchemy')
| -rw-r--r-- | lib/sqlalchemy/sql/clause_compare.py | 3 | ||||
| -rw-r--r-- | lib/sqlalchemy/sql/selectable.py | 25 | ||||
| -rw-r--r-- | lib/sqlalchemy/sql/util.py | 23 |
3 files changed, 34 insertions, 17 deletions
diff --git a/lib/sqlalchemy/sql/clause_compare.py b/lib/sqlalchemy/sql/clause_compare.py index 50b1df99e..30a90348c 100644 --- a/lib/sqlalchemy/sql/clause_compare.py +++ b/lib/sqlalchemy/sql/clause_compare.py @@ -208,6 +208,9 @@ class StructureComparatorStrategy(object): else isinstance(right.name, elements._anonymous_label) ) + def compare_cte(self, elements, left, right, **kw): + raise NotImplementedError("TODO") + def compare_extract(self, left, right, **kw): return left.field == right.field diff --git a/lib/sqlalchemy/sql/selectable.py b/lib/sqlalchemy/sql/selectable.py index 97c49f8fc..00d3826b2 100644 --- a/lib/sqlalchemy/sql/selectable.py +++ b/lib/sqlalchemy/sql/selectable.py @@ -1247,13 +1247,10 @@ class AliasedReturnsRows(FromClause): self.element._generate_fromclause_column_proxies(self) def _copy_internals(self, clone=_clone, **kw): - # don't apply anything to an aliased Table - # for now. May want to drive this from - # the given **kw. - if isinstance(self.element, TableClause): - return - self._reset_exported() - self.element = clone(self.element, **kw) + element = clone(self.element, **kw) + if element is not self.element: + self._reset_exported() + self.element = element def get_children(self, column_collections=True, **kw): if column_collections: @@ -1509,6 +1506,9 @@ class CTE(Generative, HasSuffixes, AliasedReturnsRows): [clone(elem, **kw) for elem in self._restates] ) + def _cache_key(self, *arg, **kw): + raise NotImplementedError("TODO") + def alias(self, name=None, flat=False): """Return an :class:`.Alias` of this :class:`.CTE`. @@ -2738,10 +2738,7 @@ class GenerativeSelect(SelectBase): raise NotImplementedError() def _copy_internals(self, clone=_clone, **kw): - if self._limit_clause is not None: - self._limit_clause = clone(self._limit_clause, **kw) - if self._offset_clause is not None: - self._offset_clause = clone(self._offset_clause, **kw) + raise NotImplementedError() class CompoundSelect(GenerativeSelect): @@ -2987,12 +2984,13 @@ class CompoundSelect(GenerativeSelect): return self.selects[0].selected_columns def _copy_internals(self, clone=_clone, **kw): - super(CompoundSelect, self)._copy_internals(clone, **kw) self._reset_memoizations() self.selects = [clone(s, **kw) for s in self.selects] if hasattr(self, "_col_map"): del self._col_map for attr in ( + "_limit_clause", + "_offset_clause", "_order_by_clause", "_group_by_clause", "_for_update_arg", @@ -3568,7 +3566,6 @@ class Select(HasPrefixes, HasSuffixes, GenerativeSelect): return False def _copy_internals(self, clone=_clone, **kw): - super(Select, self)._copy_internals(clone, **kw) # Select() object has been cloned and probably adapted by the # given clone function. Apply the cloning function to internal @@ -3614,6 +3611,8 @@ class Select(HasPrefixes, HasSuffixes, GenerativeSelect): # present here. self._raw_columns = [clone(c, **kw) for c in self._raw_columns] for attr in ( + "_limit_clause", + "_offset_clause", "_whereclause", "_having", "_order_by_clause", diff --git a/lib/sqlalchemy/sql/util.py b/lib/sqlalchemy/sql/util.py index dd2c7c1fb..fe83b163c 100644 --- a/lib/sqlalchemy/sql/util.py +++ b/lib/sqlalchemy/sql/util.py @@ -806,10 +806,25 @@ class ClauseAdapter(visitors.ReplacingCloningVisitor): return newcol def replace(self, col): - if isinstance(col, FromClause) and self.selectable.is_derived_from( - col - ): - return self.selectable + if isinstance(col, FromClause): + if self.selectable.is_derived_from(col): + return self.selectable + elif isinstance(col, Alias) and isinstance( + col.element, TableClause + ): + # we are a SELECT statement and not derived from an alias of a + # table (which nonetheless may be a table our SELECT derives + # from), so return the alias to prevent futher traversal + # or + # we are an alias of a table and we are not derived from an + # alias of a table (which nonetheless may be the same table + # as ours) so, same thing + return col + else: + # other cases where we are a selectable and the element + # is another join or selectable that contains a table which our + # selectable derives from, that we want to process + return None elif not isinstance(col, ColumnElement): return None elif self.include_fn and not self.include_fn(col): |