- [feature] ORM entities can be passed

to select() as well as the select_from(),
correlate(), and correlate_except()
methods, where they will be unwrapped
into selectables. [ticket:2245]

5 files changed, 20 insertions, 16 deletions

diff --git a/lib/sqlalchemy/orm/attributes.py b/lib/sqlalchemy/orm/attributes.py
index 47349e64a..045a9465d 100644
--- a/lib/sqlalchemy/orm/attributes.py
+++ b/lib/sqlalchemy/orm/attributes.py
@@ -112,9 +112,11 @@ PASSIVE_ONLY_PERSISTENT = util.symbol("PASSIVE_ONLY_PERSISTENT",
 )
 
 
-class QueryableAttribute(interfaces.PropComparator):
+class QueryableAttribute(interfaces._InspectionAttr, interfaces.PropComparator):
     """Base class for class-bound attributes. """
 
+    is_attribute = True
+
     def __init__(self, class_, key, impl=None,
                         comparator=None, parententity=None,
                         of_type=None):
@@ -149,6 +151,10 @@ class QueryableAttribute(interfaces.PropComparator):
         # TODO: conditionally attach this method based on clause_element ?
         return self
 
+    @property
+    def expression(self):
+        return self.comparator.__clause_element__()
+
     def __clause_element__(self):
         return self.comparator.__clause_element__()
 
@@ -191,10 +197,7 @@ class QueryableAttribute(interfaces.PropComparator):
     def property(self):
         return self.comparator.property
 
-
-@inspection._inspects(QueryableAttribute)
-def _get_prop(source):
-    return source.property
+inspection._self_inspects(QueryableAttribute)
 
 class InstrumentedAttribute(QueryableAttribute):
     """Class bound instrumented attribute which adds descriptor methods."""
diff --git a/lib/sqlalchemy/orm/interfaces.py b/lib/sqlalchemy/orm/interfaces.py
index 84c75525a..d0732b913 100644
--- a/lib/sqlalchemy/orm/interfaces.py
+++ b/lib/sqlalchemy/orm/interfaces.py
@@ -62,6 +62,8 @@ class _InspectionAttr(object):
     is_instance = False
     is_mapper = False
     is_property = False
+    is_attribute = False
+    is_clause_element = False
 
 class MapperProperty(_InspectionAttr):
     """Manage the relationship of a ``Mapper`` to a single class
diff --git a/lib/sqlalchemy/orm/properties.py b/lib/sqlalchemy/orm/properties.py
index 4533bbdb0..53ee1b5fd 100644
--- a/lib/sqlalchemy/orm/properties.py
+++ b/lib/sqlalchemy/orm/properties.py
@@ -102,6 +102,7 @@ class ColumnProperty(StrategizedProperty):
         else:
             self.strategy_class = strategies.ColumnLoader
 
+
     @property
     def expression(self):
         """Return the primary column or expression for this ColumnProperty.
diff --git a/lib/sqlalchemy/orm/query.py b/lib/sqlalchemy/orm/query.py
index 35c70d51e..0a345f284 100644
--- a/lib/sqlalchemy/orm/query.py
+++ b/lib/sqlalchemy/orm/query.py
@@ -26,10 +26,11 @@ from . import (
     )
 from .util import (
     AliasedClass, ORMAdapter, _entity_descriptor, PathRegistry,
-    _is_aliased_class, _is_mapped_class, _orm_columns, _orm_selectable,
+    _is_aliased_class, _is_mapped_class, _orm_columns,
     join as orm_join,with_parent, aliased
     )
-from .. import sql, util, log, exc as sa_exc, inspect
+from .. import sql, util, log, exc as sa_exc, inspect, inspection
+from ..sql.expression import _interpret_as_from
 from ..sql import (
         util as sql_util,
         expression, visitors
@@ -539,6 +540,9 @@ class Query(object):
 
         return self.enable_eagerloads(False).statement.as_scalar()
 
+    @property
+    def selectable(self):
+        return self.__clause_element__()
 
     def __clause_element__(self):
         return self.enable_eagerloads(False).with_labels().statement
@@ -798,7 +802,8 @@ class Query(object):
          """
 
         self._correlate = self._correlate.union(
-                                        _orm_selectable(s)
+                                        _interpret_as_from(s)
+                                        if s is not None else None
                                         for s in args)
 
     @_generative()
@@ -2672,7 +2677,6 @@ class Query(object):
             statement.append_order_by(*context.eager_order_by)
         return statement
 
-
     def _adjust_for_single_inheritance(self, context):
         """Apply single-table-inheritance filtering.
 
@@ -2696,6 +2700,7 @@ class Query(object):
     def __str__(self):
         return str(self._compile_context().statement)
 
+inspection._self_inspects(Query)
 
 class _QueryEntity(object):
     """represent an entity column returned within a Query result."""
diff --git a/lib/sqlalchemy/orm/util.py b/lib/sqlalchemy/orm/util.py
index 8f340d366..27d9b1b69 100644
--- a/lib/sqlalchemy/orm/util.py
+++ b/lib/sqlalchemy/orm/util.py
@@ -1074,13 +1074,6 @@ def _orm_columns(entity):
     else:
         return [entity]
 
-def _orm_selectable(entity):
-    insp = inspection.inspect(entity, False)
-    if hasattr(insp, 'selectable'):
-        return insp.selectable
-    else:
-        return entity
-
 def has_identity(object):
     state = attributes.instance_state(object)
     return state.has_identity