import json
import pandas as pd
import requests
from pypinyin import lazy_pinyin, Style


def get_initial(name):
    """
    :param name: 字符串
    :return: 返回首字母缩写(小写)
    """
    initial = ''.join(lazy_pinyin(name, style=Style.FIRST_LETTER))
    return initial.lower()


# 邀请码
invitationCode = ""
# 请求地址
url = "http://10.226.8.14:9091/jx-welcome/registerBatch"
# params参数
params = {
    "assessKey": "",
    "secretKey": ""
}
# 请求头
headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.71 Safari/537.36',
    'Content-Type': 'application/json'
}
# excel数据
data = pd.read_excel("./人员名单.xlsx")
# 错误消息列表
error_list = []
success_list = []

for index, row in data.iterrows():
    nickName = row["姓名"]
    username = row["学号"]
    # 自定义账号密码策略(学号后6位+姓名首字母小写)
    password = str(username)[-6:] + get_initial(nickName)
    confirmPassword = password
    payload_dict = {
        "invitationCode": invitationCode,
        "username": username,
        "password": password,
        "confirmPassword": password,
        "clasz": row["班级"],
        "nickName": nickName,
        "studentId": row["学号"],
        "phonenumber": row["手机号码"],
        "email": row["邮箱"]
    }
    payload = json.dumps(payload_dict)
    # 发送请求
    response = requests.request("POST", url=url, headers=headers, params=params, data=payload)
    json_response = response.json()
    info = {"请求参数": payload_dict, "返回的消息": json_response['msg']}
    if json_response['code'] != 200:
        error_list.append(info)
    else:
        success_list.append(info)

# 记录信息
with open(file="注册错误信息记录.txt", mode="w") as f:
    if len(error_list) == 0:
        f.write("无错误信息")
    else:
        for error in error_list:
            f.write(str(error))
            f.write("\n")

with open(file="注册成功人员名单信息.txt", mode="w") as f:
    for success in success_list:
        f.write(str(success))
        f.write("\n")