adding the possibility to get a list of attributes in get_user

kakwa · May 22, 2015 · ab8c380 · ab8c380
1 parent f4906bb
commit ab8c380
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ldapcherry/backend/backendLdap.py b/ldapcherry/backend/backendLdap.py
@@ -56,7 +56,7 @@ def add_user(self, username):
     def del_user(self, username):
         pass
 
-    def get_user(self, username):
+    def get_user(self, username, attrs=None):
         ldap_client = self._connect()
         try:
             ldap_client.simple_bind_s(self.binddn, self.bindpassword)
@@ -80,8 +80,9 @@ def get_user(self, username):
 
         r = ldap_client.search_s(self.userdn,
                 ldap.SCOPE_SUBTREE,
-                user_filter
-                 )
+                user_filter,
+                attrlist=attrs
+            )
         if len(r) == 0:
             ldap_client.unbind_s()
             return False