X-Git-Url: https://git.stderr.nl/gitweb?a=blobdiff_plain;f=ldapdb%2Fmodels%2Fquery.py;h=7d952d45b23aca76514f787f7c22e87d9498345d;hb=81037a75f672596982475a7081946bfd9b4fd286;hp=6eaecb117fcdb6232a0d7af33b1f796e2a310d48;hpb=5b07fe4646e8171565701bef9884d257f18cf65c;p=matthijs%2Fupstream%2Fdjango-ldapdb.git diff --git a/ldapdb/models/query.py b/ldapdb/models/query.py index 6eaecb1..7d952d4 100644 --- a/ldapdb/models/query.py +++ b/ldapdb/models/query.py @@ -37,7 +37,12 @@ class WhereNode(BaseWhereNode): if isinstance(item, WhereNode): bits.append(item.as_sql()) continue - table, column, type, x, y, values = item + if len(item) == 4: + # django 1.1 + (table, column, type), x, y, values = item + else: + # django 1.0 + table, column, type, x, y, values = item if self.negated: bits.append('(!(%s=%s))' % (column,values[0])) else: @@ -54,14 +59,14 @@ class WhereNode(BaseWhereNode): class Query(BaseQuery): def results_iter(self): # FIXME: use all object classes - filterstr = '(objectClass=%s)' % self.model._meta.object_classes[0] + filterstr = '(objectClass=%s)' % self.model.object_classes[0] filterstr += self.where.as_sql() filterstr = '(&%s)' % filterstr attrlist = [ x.db_column for x in self.model._meta.local_fields if x.db_column ] try: vals = ldapdb.connection.search_s( - self.model._meta.dn, + self.model.base_dn, ldap.SCOPE_SUBTREE, filterstr=filterstr, attrlist=attrlist, @@ -86,9 +91,12 @@ class Query(BaseQuery): # process results for dn, attrs in vals: - row = [dn] + row = [] for field in iter(self.model._meta.fields): - row.append(attrs.get(field.db_column, None)) + if field.attname == 'dn': + row.append(dn) + else: + row.append(attrs.get(field.db_column, None)) yield row class QuerySet(BaseQuerySet):