From b0f1e3af857999dd9e4fd6f6c6dbd379dac85ad2 Mon Sep 17 00:00:00 2001 From: Konsta Vesterinen Date: Tue, 21 Jan 2014 22:38:33 +0200 Subject: [PATCH] Refactor update_query --- sqlalchemy_utils/aggregates.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/sqlalchemy_utils/aggregates.py b/sqlalchemy_utils/aggregates.py index 22b2a7e..dd2fbff 100644 --- a/sqlalchemy_utils/aggregates.py +++ b/sqlalchemy_utils/aggregates.py @@ -386,13 +386,9 @@ class AggregatedValue(object): {self.attr: self.aggregate_query} ) if len(self.relationships) == 1: - remote_pairs = self.relationships[-1].property.local_remote_pairs + prop = self.relationships[-1].property - return query.where( - remote_pairs[0][0].in_( - getattr(obj, remote_pairs[0][1].key) for obj in objects - ) - ) + return query.where(self.local_condition(prop, objects)) else: # Builds query such as: #