import csv
import xlrd
import requests

"""
pip3 install xlrd==1.2.0

"""


class ReaderFile:
    """
        读取csv文件
        filePath:文件路径
    """

    def getCsvValue(self, filePath):
        # 用于存放返回dict的list
        dictList = []
        # 打开指定路径的文件
        with open(filePath) as file:
            # 读取csv文件
            datareader = csv.reader(file)
            # 读取的csv文件内容转换成list
            csvList = list(datareader)
            # 获取csv的第一列为dict的key值
            keyList = csvList[0]
            # 将csv取出的数据处理成dict形式
            for value in range(1, len(csvList)):
                # dict必须声明在此位置,后面的dictList.append()时里面的dict是不同的对象,若声明全局变量,那dictList中的dict的指针全是指向同一个对象,值完全一致
                csvDict = {}
                for item in range(0, len(keyList)):
                    # 向dict中添加key和value
                    csvDict[keyList[item]] = csvList[value][item]
                # 将处理完成的dict追加到dictList中
                dictList.append(csvDict)
        return dictList

    """
        读取excel文件
        filePath:文件路径
        sheetName:要读取的sheet工作表的名称
    """

    def getExcelValue(self, filePath, sheetName):
        # 用于存放获取封装的dict的list
        allList = []
        # 读取excel文件
        workbook = xlrd.open_workbook(filePath)
        # 根据sheet名称读取sheet内容
        sheetBook = workbook.sheet_by_name(sheetName)
        # 获取第一行的内容作为key
        keyList = sheetBook.row_values(0)
        # 对第二行及之后的内容进行遍历,与keyList组合并封装成dict
        for value in range(1, sheetBook.nrows):
            listDict = {}
            for key in range(0, len(keyList)):
                listDict[keyList[key]] = sheetBook.row_values(value)[key]
            allList.append(listDict)
        return allList


sample_list = ReaderFile().getExcelValue("/Users/alvin/Downloads/AI测试/测试样例(1).xlsx", "catboost_test")
for x in sample_list:
    url = "http://gateway.aimpdev.k5.bigtree.tech/aimp-manager/out_api/ai_catboost1/1.3"
    post = requests.post(url, json={"org_no": x['risk_buss_no']})
    print(post.json())