Fix bug in HBAC and netgroup plugin get_primary_key_from_dn methods.

This commit is contained in:
Pavel Zuna
2009-10-06 15:50:43 +02:00
committed by Rob Crittenden
parent 0dd92f10cc
commit 0373166002
2 changed files with 8 additions and 2 deletions

View File

@@ -96,7 +96,10 @@ class hbac(LDAPObject):
def get_primary_key_from_dn(self, dn):
pkey = self.primary_key.name
(dn, entry_attrs) = self.backend.get_entry(dn, [pkey])
return entry_attrs.get(pkey, '')
try:
return entry_attrs[pkey][0]
except (KeyError, IndexError):
return ''
api.register(hbac)

View File

@@ -81,7 +81,10 @@ class netgroup(LDAPObject):
def get_primary_key_from_dn(self, dn):
pkey = self.primary_key.name
(dn, entry_attrs) = self.backend.get_entry(dn, [pkey])
return entry_attrs.get(pkey, '')
try:
return entry_attrs[pkey][0]
except (KeyError, IndexError):
return ''
api.register(netgroup)