Zionetrix::Git
Repositories
Help
Report an Issue
python-mylib
Code
Commits
Branches
Tags
Search
Tree:
a525bd5
Branches
Tags
master
python-mylib
LdapServer.py
Fix/improve error reporting
Benjamin Renard
commited
a525bd5
at 2018-02-13 14:30:38
LdapServer.py
Blame
History
Raw
#!/usr/bin/python import ldap import ldap.modlist as modlist import logging class LdapServer(object): uri = None dn = None pwd = None v2 = None con = 0 def __init__(self,uri,dn=None,pwd=None,v2=None,raiseOnError=False): self.uri = uri self.dn = dn self.pwd = pwd self.raiseOnError = raiseOnError if v2: self.v2=True def _error(self,error,level=logging.WARNING): if self.raiseOnError: raise LdapServerException(error) else: logging.log(level,error) def connect(self): if self.con == 0: try: con = ldap.initialize(self.uri) if self.v2: con.protocol_version = ldap.VERSION2 else: con.protocol_version = ldap.VERSION3 if self.dn: con.simple_bind_s(self.dn,self.pwd) self.con = con return True except ldap.LDAPError, e: self._error('LdapServer - Error connecting and binding to LDAP server : %s' % e,logging.CRITICAL) return False return True def search(self,basedn,filter,attrs,sizelimit=0): res_id = self.con.search(basedn,ldap.SCOPE_SUBTREE,filter,attrs) ret = {} c=0 while 1: res_type, res_data = self.con.result(res_id,0) if res_data == [] or sizelimit!=0 and c>sizelimit: break else: if res_type == ldap.RES_SEARCH_ENTRY: ret[res_data[0][0]]=res_data[0][1] c=c+1 return ret def add_object(self,dn,attrs): ldif = modlist.addModlist(attrs) try: logging.debug("LdapServer - Add %s" % dn) self.con.add_s(dn,ldif) return True except ldap.LDAPError, e: self._error("LdapServer - Error adding %s : %s" % (dn,e), logging.error) return False def update_object(self,dn,old,new): ldif = modlist.modifyModlist(old,new) if ldif == []: return True try: self.con.modify_s(dn,ldif) return True except ldap.LDAPError, e: self._error("LdapServer - Error updating %s : %s\nOld : %s\nNew : %s" % (dn, e, old, new), logging.error) return False def rename_object(self,dn,new_rdn): try: logging.debug("LdapServer - Rename %s in %s" % (dn,new_rdn)) self.con.rename_s(dn,new_rdn) return True except ldap.LDAPError, e: self._error("LdapServer - Error renaming %s in %s : %s" % (dn,new_rdn,e), logging.error) return False def drop_object(self,dn): try: logging.debug("LdapServer - Delete %s" % dn) self.con.delete_s(dn) return True except ldap.LDAPError, e: self._error("LdapServer - Error deleting %s : %s" % (dn,e), logging.error) return False def get_dn(self,obj): return obj[0][0] def get_attr(self,obj,attr,all=None,default=None): if attr not in obj: for k in obj: if k.lower() == attr.lower(): attr = k break if all is not None: if attr in obj: return obj[attr] else: return default or [] else: if attr in obj: return obj[attr][0] else: return default class LdapServerException(BaseException): def __init__(self,msg): BaseException.__init__(self, msg)
