客户已存在_白名单批量导入.py

# -*- encoding=utf8 -*-
__author__ = "liguangyu"

"""
case_tag:cmdc_api,cmdc客户已存在_白名单批量导入,2299,2299-29,sit,bs
主数据平台:后台运营系统白名单批量导入接口
"""

from common.common_func import commonFuc
from air_case.cmdc_login.后台管理系统登录.后台管理系统登录 import CmdcDoLogin
from common.handle_excel import HandleExcel
import requests
import random
import json
import os

module = "cmdc_area"

# 第一步登录后台运营系统获取token
username = commonFuc().get_business_data(module, "username2")
password = commonFuc().get_business_data(module, "password2")
# 获取登录后Cmdc_access_token
cmdc_access_token = CmdcDoLogin(username, password).get_token()
headers = commonFuc().get_business_data(module, "json_headers_file", cmdc_access_token)

# 获取文件
BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
file_path = BASE_DIR + "/data/cmdc_files/白名单导入模板.xlsx"

# 第二步获取站点开放白名单列表
url = commonFuc().get_api_url() + commonFuc().get_business_data(module, "url12")
request_body = commonFuc().get_business_data(module, "payload12")
# 发送请求
result = requests.post(url, json=request_body, headers=headers)
result = json.loads(result.content)
# 判断开放白名单列表中是否有值
if result["data"]["list"]:
    # 生成随机数
    user_random = random.randint(0, len(result["data"]["list"]) - 1)
    # 获取白名单信息
    userName = result["data"]["list"][user_random]["userName"]
    iphone = result["data"]["list"][user_random]["iphone"]
    companyName = result["data"]["list"][user_random]["companyName"]

    # 将信息写入文件中
    excel = HandleExcel(file_path, "Sheet1")
    excel.write_data(row=2, column=1, value=userName)
    excel.write_data(row=2, column=2, value=iphone)
    excel.write_data(row=2, column=3, value=companyName)

    # 第二步进行白名单批量导入
    url = commonFuc().get_api_url() + commonFuc().get_business_data(module, "url29")

    # 获取文件
    file = open(file_path, "rb")
    files = {"file": file}

    """
     场景: 客户已存在_白名单批量导入
     用例名称：客户已存在_白名单批量导入
     输出：{"success":false,"code":"siteopen","message":"第2行用户白名单或申请已存在","data":null,"freshToken":null}
    """

    # 发送请求
    result = requests.post(url, files=files, headers=headers)
    # 获取接口响应时间
    api_time = float(result.elapsed.total_seconds())

    # 将接口响应时间添加至result
    result = json.loads(result.content)