python分组和任务分配_Python3批量创建Crowd用户并分配组

背景

迁移 Crowd 完成后(之前采用 LDAP 方式，新迁移 Crowd 不采用)，需要批量创建公司所有员工的用户以及分配组，手工创建以及之前 Postman 的方式还是比较低效。

Python 在 N 多年前入门，写了几个爬虫脚本后，再也没用过，借这个机会顺便再熟悉下 Python 脚本。

归根结底的原因就是：本人很懒~

Crowd Api

如下示例是基于 Crowd 3.2.0 版本的 Api，不同版本间的 Api 稍有差异。

# 添加用户

$ curl -u "application-name:password" -X POST -H "Content-Type: application/json" -H "Accept: application/json" -d "{\"name\" : \"test.user\", \"display-name\" : \"Test User\", \"active\" : true, \"first-name\" : \"Test\", \"email\" : \"test.user@ourdomain.com\", \"last-name\" : \"User\", \"password\" : {\"value\" : \"mypassword\"} }" http://localhost:8095/crowd/rest/usermanagement/1/user

# 用户添加到组

$ curl -u "application-name:password" -X POST -H "Content-Type: application/json" -d "{\"name\" : \"all-users\"}" http://localhost:8095/crowd/rest/usermanagement/1/user/group/direct\?username\=daodaotest

注意：此处-u的参数为 Crowd 中应用(Application)的用户名和密码，Crowd 的管理员是不能添加用户。

Python 实现脚本

实现添加 Crowd 用户，用户添加到指定组，读取 csv 文件批量添加用户和设定的多个组。

crowdUsers.csv 用户数据 csv 文件

name,displayName,email

daodaotest1,daodaotest1,daodaotest1@daodaotest.com

daodaotest2,daodaotest2,daodaotest2@daodaotest.com

daodaotest3,daodaotest3,daodaotest3@daodaotest.com

......

addCrowdUsers.py 批量添加 Crowd 用户和用户组脚本

#!/usr/bin/python

# -*- coding: UTF-8 -*-

#

# Filename addCrowdUsers.py

# Revision 0.0.1

# Date 2020/5/14

# Author jiangliheng

# Email jiang_liheng@163.com

# Website https://jiangliheng.github.io/

# Description 批量添加 Crowd 用户和用户组

import requests

from requests.auth import HTTPBasicAuth

import csv

from itertools import islice

# 请求 headers

headers = {

'Accept': 'application/json',

'Content-type': 'application/json',

}

# crowd 访问基础路径

base_url='http://localhost:8095'

# 添加用户的默认用户组和密码

auth_username='application-name'

auth_password='password'

# 用户默认密码

password='daodaotest'

def addUser(name,displayName,email):

"""

添加单用户

:param name: 登录用户，建议拼音全称，如：jiangliheng

:param displayName: 显示名称，建议中文全称，如：蒋李恒

:param email: 邮箱地址

:return: status_code 状态码,text 响应报文信息

"""

# 请求 json 数据

data = '{ \

"name" :"' + name + '", \

"email" : "' + email + '", \

"active" : true, \

"first-name" : "' + displayName + '", \

"last-name" : "' + displayName + '", \

"display-name" : "'+ displayName + '", \

"password" : { \

"value" : "' + password + '" \

} \

}'

# 发起请求

# 解决中文乱码问题 data.encode("utf-8").decode("latin1")

response = requests.post(

base_url + '/crowd/rest/usermanagement/1/user',

headers=headers,

auth=HTTPBasicAuth(auth_username,auth_password),

data=data.encode("utf-8").decode("latin1")

)

# 状态码

status_code=response.status_code

# 响应报文信息

text=response.text

# 状态判断

if str(status_code).startswith("2"):

print("%s 用户添加成功，状态码：%s ，响应报文信息：%s" % (name,status_code,text))

else:

print("%s 用户添加失败，状态码：%s ，响应报文信息：%s" % (name,status_code,text))

# 返回 状态码，响应报文信息

return status_code,text

def addGroup(username,groupname):

"""

用户添加到组

:param username: 登录用户，建议拼音全称，如：jiangliheng

:param groups: 用户组，用逗号隔开，如：bitbucket-users,bamboo-users

:return: status_code 状态码,text 响应报文信息

"""

# 请求 json 数据

data = '{ \

"name" :"' + groupname + '" \

}'

# 发起请求

response = requests.post(

base_url + '/crowd/rest/usermanagement/1/user/group/direct?username='+username,

headers=headers,

auth=HTTPBasicAuth(auth_username,auth_password),

data=data

)

# 状态码

status_code=response.status_code

# 响应报文信息

text=response.text

# 状态判断

if str(status_code).startswith("2"):

print("%s 用户添加组 %s 成功，状态码：%s ，响应报文信息：%s" % (username,groupname,status_code,text))

else:

print("%s 用户添加组 %s 失败，状态码：%s ，响应报文信息：%s" % (username,groupname,status_code,text))

# 返回 状态码，响应报文信息

return status_code,text

def addUserByCsv(csvfile):

"""

通过 CSV 文件批量添加用户，并加到组

:param filename: Crowd 用户 csv 文件

"""

# 批量读取 csv 的用户

with open(csvfile, 'r', encoding='utf-8') as f:

fieldnames = ("name", "displayName", "email")

reader = csv.DictReader(f, fieldnames)

for row in islice(reader, 1, None):

print("批量添加用户 %s" % (row["name"]))

# 添加用户

addUser(row["name"],row["displayName"],row["email"])

# 添加多个组

addGroup(row["name"],"all-users")

addGroup(row["name"],"bitbucket-users")

addGroup(row["name"],"confluence-users")

addGroup(row["name"],"jira-software-users")

addGroup(row["name"],"sonar-users")

f.close()

def main():

# 通过 CSV 文件批量添加用户，并加到组

addUserByCsv("crowdUsers.csv")

# 添加单用户

# addUser("daodaotest","叨叨软件测试","daodaotest@daodaotest.com")

# 添加用户到组

# addGroup("daodaotest","all-users")

if __name__ == "__main__":

main()

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。