#!/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 訪問基礎(chǔ)路徑
base_url='http://localhost:8095'

# 添加用戶的默認(rèn)用戶組和密碼
auth_username='application-name'
auth_password='password'

# 用戶默認(rèn)密碼
password='daodaotest'

def addUser(name,displayName,email):
  """
  添加單用戶

  :param name: 登錄用戶，建議拼音全稱，如：jiangliheng
  :param displayName: 顯示名稱，建議中文全稱，如：蔣李恒
  :param email: 郵箱地址
  :return: status_code 狀態(tài)碼,text 響應(yīng)報(bào)文信息
  """

  # 請求 json 數(shù)據(jù)
  data = '{ \
    "name" :"' + name + '", \
    "email" : "' + email + '", \
    "active" : true, \
    "first-name" : "' + displayName + '", \
    "last-name" : "' + displayName + '", \
    "display-name" : "'+ displayName + '", \
    "password" : { \
      "value" : "' + password + '" \
    } \
  }'

  # 發(fā)起請求
  # 解決中文亂碼問題 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")
  )

  # 狀態(tài)碼
  status_code=response.status_code
  # 響應(yīng)報(bào)文信息
  text=response.text

  # 狀態(tài)判斷
  if str(status_code).startswith("2"):
    print("%s 用戶添加成功，狀態(tài)碼：%s ，響應(yīng)報(bào)文信息：%s" % (name,status_code,text))
  else:
    print("%s 用戶添加失敗，狀態(tài)碼：%s ，響應(yīng)報(bào)文信息：%s" % (name,status_code,text))

  # 返回 狀態(tài)碼，響應(yīng)報(bào)文信息
  return status_code,text

def addGroup(username,groupname):
  """
  用戶添加到組

  :param username: 登錄用戶，建議拼音全稱，如：jiangliheng
  :param groups: 用戶組，用逗號隔開，如：bitbucket-users,bamboo-users
  :return: status_code 狀態(tài)碼,text 響應(yīng)報(bào)文信息
  """

  # 請求 json 數(shù)據(jù)
  data = '{ \
    "name" :"' + groupname + '" \
  }'

  # 發(fā)起請求
  response = requests.post(
    base_url + '/crowd/rest/usermanagement/1/user/group/direct?username='+username,
    headers=headers,
    auth=HTTPBasicAuth(auth_username,auth_password),
    data=data
  )

  # 狀態(tài)碼
  status_code=response.status_code
  # 響應(yīng)報(bào)文信息
  text=response.text

  # 狀態(tài)判斷
  if str(status_code).startswith("2"):
    print("%s 用戶添加組 %s 成功，狀態(tài)碼：%s ，響應(yīng)報(bào)文信息：%s" % (username,groupname,status_code,text))
  else:
    print("%s 用戶添加組 %s 失敗，狀態(tài)碼：%s ，響應(yīng)報(bào)文信息：%s" % (username,groupname,status_code,text))

  # 返回 狀態(tài)碼，響應(yīng)報(bào)文信息
  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"])
      # 添加多個(gè)組
      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()