LDAP-PYTHON文档

功能：
   1.用户登录 (LDAP账号密码登录)
   2.用户搜索（查询相关用户信息）
   3.用户信息修改（增加删除修改相关用户信息）
   4.用户密码修改

代码示范：
# coding=utf-8
import ldap

LDAP_CONFIG = {
   'HOST': 'ldap://172.16.0.250',        # ad服务器地址
   'HOST_SSL': 'ldaps://172.16.0.250:636',   # 使用加密协议的ad服务器地址
   'BASE': 'ou=ZY,dc=casvip,dc=com',         #  基准dn，也就是ad的搜索域
   'USER_DN': 'administrator@casvip.com',    #  bind_dn, ad的管理员账号,可以通过此账号对ad字段信息进行查询，修改。
   'PASSWORD': 'Admin123***',                #  管理员密码
   'LOGIN_ATTR': 'userPrincipalName'         #  登录所用的字段 此字段格式为id@casvip.com eg：17081111@casvip.com
}

class MyLdap(object):
   def __init__(self, server_url, user='', passwd=''):
       '''
       :param server_url:
       :param user:
       :param passwd:
       initail
       '''
       self.user = user
       self.passwd = passwd
       self.server_url = server_url
       self.ldap_obj = None

def ldap_connect(self):

"""
用户登录
       connenct ldap-server
       :return: success: true or erro : flase msg
       """

ldap.set_option(ldap.OPT_X_TLS_REQUIRE_CERT, ldap.OPT_X_TLS_NEVER)
       lp = ldap.initialize(self.server_url)
       lp.set_option(ldap.OPT_REFERRALS, 0)
       lp.set_option(ldap.OPT_PROTOCOL_VERSION, 3)
       lp.set_option(ldap.OPT_X_TLS, ldap.OPT_X_TLS_DEMAND)
       lp.set_option(ldap.OPT_X_TLS_DEMAND, True)
       lp.set_option(ldap.OPT_DEBUG_LEVEL, 255)

# try:
       #     conn.start_tls_s()
       # except ldap.LDAPError as exc:
       #     raise APIException(exc.message)
       if self.user and not self.passwd:
           return False, u"请输入LDAP密码"
       try:
           rest = lp.simple_bind_s(self.user, self.passwd)
       except ldap.SERVER_DOWN:
           return False, u"无法连接到LDAP"
       except ldap.INVALID_CREDENTIALS:
           return False, u"LDAP账号错误"
       except Exception, ex:
           # return False, type(ex)
           return False, u"其他错误"
       if rest[0] != 97:  # 97 表示success
           return False, u"帐号或密码错误"
       self.ldap_obj = lp
       return True

def ldap_search(self, base=LDAP_CONFIG['BASE'], keyword=None, rdn=LDAP_CONFIG['LOGIN_ATTR']):
       """
用户搜索
       base: 域 ou=ZY,dc=casvip,dc=com
       keyword: 搜索的用户
       rdn: userPrincipalName
       """
       scope = ldap.SCOPE_SUBTREE
       # filter = "%s=%s" % (rdn, keyword)
       retrieve_attributes = None
       try:
           result_id = self.ldap_obj.search(base, scope, filter, retrieve_attributes)
           result_type, result_data = self.ldap_obj.result(result_id)
           if not result_data:
               return False, []
       except ldap.LDAPError, error_message:
           return False, error_message
       return True, result_data

   def modify_user(self, dn, attr_list):
       """
修改用户信息
       MOD_ADD: 如果属性存在，这个属性可以有多个值，那么新值加进去，旧值保留
       MOD_DELETE ：如果属性的值存在，值将被删除
       MOD_REPLACE ：这个属性所有的旧值将会被删除，这个值被加进去

dn: cn=test, ou=magicstack,dc=test, dc=com
       attr_list: [( ldap.MOD_REPLACE, 'givenName', 'Francis' ),
                   ( ldap.MOD_ADD, 'cn', 'Frank Bacon' )
                  ]
       """
       try:
           result = self.ldap_obj.modify_s(dn, attr_list)
       except ldap.LDAPError, error_message:
           return False, error_message
       else:
           if result[0] == 103:
               return True, []
           else:
               return False, result[1]

   def get_dn(self, base=LDAP_CONFIG['BASE'], keyword=None, rdn=LDAP_CONFIG['LOGIN_ATTR']):
       '''
获取DN
       :param base:
       :param keyword:
       :param rdn:
       :return: user dn
       '''
       success, result = self.ldap_search(base, keyword, rdn)
       if not success:
           return []
       return result[0][0]

   def ldap_update_pass(self, new_pass, user=''):
       '''
用户密码修改
       password change need host_ssl_url and new_passwd
       :param new_pass:
       :param user:
       :return: True or False,msg
       '''

try:
           dn = self.get_dn(user)
           unicode_pass = unicode("\"" + str(new_pass) + "\"", "iso-8859-1")
           password_value = unicode_pass.encode("utf-16-le")
           add_pass = [(ldap.MOD_REPLACE, 'unicodePwd', [password_value])]
           self.ldap_obj.modify_s(dn, add_pass)
           self.login_out()
       except Exception, e:
           return False, e

   def login_out(self):
       self.ldap_obj.unbind_s()
       del self.ldap_obj
       return True

def main():
   ld = MyLdap(LDAP_CONFIG['HOST'], LDAP_CONFIG['USER_DN'], LDAP_CONFIG['PASSWORD'])
   print ld.ldap_connect() #连接成功 返回True
   search_result = ld.ldap_search(LDAP_CONFIG["BASE"], '123@casvip.com', rdn=LDAP_CONFIG["LOGIN_ATTR"])
   print search_result #返回搜索结果
   attr_list = [(ldap.MOD_REPLACE, 'userpassword', '123456')]
   dn = ld.get_dn(LDAP_CONFIG["BASE"], '123@casvip.com', rdn=LDAP_CONFIG["LOGIN_ATTR"])
   print ld.modify_user(dn, attr_list) #修改userpassword 这一属性

if __name__ == "__main__":
   main()