python 获取AD信息

原创

mob64ca12dfd1d5 2024-08-02 12:16:13 ©著作权

文章标签 Python CSV Server 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12dfd1d5的原创作品，请联系作者获取转载授权，否则将追究法律责任

在Python中获取Active Directory（AD）信息的指南

在当前的企业环境中，Active Directory（AD）是管理用户和资源的重要工具。对于开发者而言，从AD中获取信息不仅能帮助管理用户、组及其权限，也能为开发自动化脚本提供支持。在这篇文章中，我将指导你如何用Python获取AD信息。为了简化流程，我将分步讲解。

流程概述

以下是获取AD信息的基本流程：

步骤	描述
1	设置环境，安装必要库
2	连接到Active Directory
3	执行查询以获取用户或组的信息
4	处理查询结果并输出

步骤详解

步骤 1：设置环境，安装必要库

要与Active Directory交互，首先需要安装一些库。我们将使用ldap3库，这是一个用于与LDAP（轻量级目录访问协议）服务交互的Python库。

pip install ldap3

此命令会安装ldap3库，我们后续会用到它。

步骤 2：连接到Active Directory

我们需要使用适当的凭证连接到Active Directory。请确保你有访问AD服务器的权限。

下面是连接到AD的代码示例：

from ldap3 import Server, Connection, ALL, NTLM

# 定义伺服器地址和域名
server = Server('your_ad_server', get_info=ALL)  # 替换为你的AD服务器地址
username = 'DOMAIN\\username'  # 使用DOMAIN\\username格式
password = 'your_password'  # 密码

# 连接到AD
conn = Connection(server, user=username, password=password, authentication=NTLM)

# 检查连接
if not conn.bind():
    print('连接失败:', conn.result)
else:
    print('成功连接到AD')

代码说明：

Server：创建连接AD服务器的实例。
Connection：创建连接并进行身份验证，使用NTLM协议。
conn.bind()：尝试连接，如果返回True则连接成功。

步骤 3：执行查询以获取用户或组的信息

连接成功后，可以执行LDAP查询来获取所需的信息。这里是获取所有用户的示例代码：

# 定义搜索基础和过滤条件
search_base = 'dc=your_domain,dc=com'  # 替换为你的域名
search_filter = '(objectClass=user)'  # 查询所有用户

# 执行搜索
conn.search(search_base, search_filter, attributes=['cn', 'sAMAccountName', 'mail'])

# 检查是否有结果
if len(conn.entries) > 0:
    for entry in conn.entries:
        print(f'姓名: {entry.cn}, 登录名: {entry.sAMAccountName}, 邮箱: {entry.mail}')
else:
    print('未找到任何用户信息')

代码说明：

search_base：定义LDAP搜索的起始点。
search_filter：设置查询过滤器，这里我们寻找所有用户。
attributes：指定要检索的字段。
conn.search()：执行查询。
conn.entries：存储查询结果。

步骤 4：处理查询结果并输出

在第三步中，我们已经处理了查询结果并将其打印出来。根据你的需求，可以进一步扩展此部分，比如保存到CSV文件或数据库中。以下是将结果保存到CSV文件的示例代码：

import csv

# 打开CSV文件并写入数据
with open('ad_users.csv', mode='w', newline='', encoding='utf-8') as file:
    writer = csv.writer(file)
    writer.writerow(['姓名', '登录名', '邮箱'])  # 写入表头

    for entry in conn.entries:
        writer.writerow([entry.cn, entry.sAMAccountName, entry.mail])  # 写入每个用户的信息
    
print('用户信息已保存到 ad_users.csv')

代码说明：