small refactorization

Bauble · Jan 1, 2015 · e9b202c · e9b202c
1 parent 0c9717f
commit e9b202c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/bauble/search.py b/bauble/search.py
@@ -96,11 +96,12 @@ def evaluate(self, env):
         query = env.session.query(env.domain)
         if len(self.value) == 1:
             # identifier is an attribute of the table being queried
-            attr = getattr(env.domain, self.value[0])
+            cls = env.domain
         elif len(self.value) > 1:
             # identifier is an attribute of a joined table
             query = query.join(*self.value[:-1], aliased=True)
-            attr = getattr(query._joinpoint['_joinpoint_entity'], self.value[-1])
+            cls = query._joinpoint['_joinpoint_entity']
+        attr = getattr(cls, self.value[-1])
         return query, attr
 
     def needs_join(self, env):