# 票 3.0 测试
import json

import pandas as pd
import requests

appId = "38bb3c94f3004d979b283657e4befb69"
secret = "kKL7HP0DHxcK1LO5wJhHykDp5w5_Vp70"
default_header = {
    "appId": appId,
    "secret": secret
}
url = "http://aimp.btaimp.k7.bigtree.tech/api/aimp-platform/out_api/processInvoke/ai_piao_model_v3"

# 读取csv文件，首行为标题
datas = pd.read_csv('/Users/alvin/Downloads/模型测试_全量样本_for田云.csv')

# datas转json数组
datas = datas.to_json(orient='records')

datas = json.loads(datas)
# datas 取前10条
# datas = datas[:2]
length = len(datas)
print("datas长度" + str(len(datas)))  # datas长度
a = 0
for data in datas:
    a += 1
    res = requests.post(url, json=data, headers=default_header).json()
    # 获取p和score  {'code': 10000200, 'data': {'output': {'p': '0.005806804', 'score': 1000.0, 'level': 'F'}}}
    if res['code'] == 10000200:
        p_ = res['data']['output']['p']
        score_ = res['data']['output']['score']
        data['p'] = p_
        data['score'] = score_
        print(length - a)
    else:
        print("error")
        print(res)
# 将datas转为DataFrame，写入csv文件，首行为标题
datas_df = pd.DataFrame(datas)
# ["risk_buss_no", "p", ["score"] 为要写入的列
datas_df[["risk_buss_no", "p", "score"]].to_csv('/Users/alvin/Downloads/模型测试_全量样本_for田云_new.csv', index=False)