|
|
@@ -0,0 +1,67 @@
|
|
|
+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())
|
