openstack/deb-python-sqlalchemy-utils
Code Issues Proposed changes

Fix query_entities with labeled subqueries

Browse Source
This commit is contained in:
Konsta Vesterinen
2014-03-20 11:15:18 +02:00
parent d47d56525f
commit d615225c43
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
sqlalchemy_utils/functions/orm.py
View File

@@ -132,16 +132,21 @@ def query_entities(query):
Example:: Example::
query = session.query( query = session.query(Category)
Category
) query_entities(query) # <Category>
query = session.query(Category.id)
query_entities(query) # <Category>
query_entities(query) # ('Category', )
:param query: SQLAlchemy Query object :param query: SQLAlchemy Query object
""" """
for entity in query._entities: for entity in query._entities:
yield entity.entity_zero.class_ if entity.entity_zero:
yield entity.entity_zero.class_
for entity in query._join_entities: for entity in query._join_entities:
if isinstance(entity, Mapper): if isinstance(entity, Mapper):