Remove toolz refs

This commit is contained in:
Konsta Vesterinen
2014-03-03 18:57:11 +02:00
parent 128c39bbb7
commit d2e01b354d

View File

@@ -1,7 +1,5 @@
from functools import partial from functools import partial
from operator import attrgetter from operator import attrgetter
from toolz import first
import six
import sqlalchemy as sa import sqlalchemy as sa
from sqlalchemy import inspect from sqlalchemy import inspect
from sqlalchemy.orm.attributes import InstrumentedAttribute from sqlalchemy.orm.attributes import InstrumentedAttribute
@@ -105,7 +103,6 @@ def remote_column_names(prop):
yield remote.name yield remote.name
def query_labels(query): def query_labels(query):
""" """
Return all labels for given SQLAlchemy query object. Return all labels for given SQLAlchemy query object.
@@ -155,7 +152,7 @@ def query_entities(query):
def get_query_entity_by_alias(query, alias): def get_query_entity_by_alias(query, alias):
entities = query_entities(query) entities = query_entities(query)
if not alias: if not alias:
return first(entities) return list(entities)[0]
for entity in entities: for entity in entities:
if isinstance(entity, AliasedInsp): if isinstance(entity, AliasedInsp):