12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- import requests, os
- from config import *
- def cls_process_openai(path, client_id, one_key, name_column, api_key="sk-iREtaVNjamaBArOTlc_2BfGFJVPiU-9EjSFMUspIPBT3BlbkFJxS0SMmKZD9L9UumPczee4VKawCwVeGBQAr9MgsWGkA",
- proxy=False, chunk_size=100):
- # 定义请求的数据
- data = {
- "path": path,
- "client_id": client_id,
- "one_key": one_key,
- "name_column": name_column,
- "proxy": proxy,
- "chunk_size":chunk_size
- }
- # 发送 POST 请求
- response = requests.post(f"http://10.41.1.57:8071/classify_openai/", json=data)
- # 处理响应
- if response.status_code == 200:
- result = response.json()
- print("接口响应:", result)
- else:
- print(f"请求失败,状态码: {response.status_code}")
- print("错误信息:", response.text)
- def cls_process_bert(path, client_id, name_column):
- # 定义请求的数据
- data = {
- "path": path,
- "client_id": client_id,
- "name_column": name_column
- }
- # 发送 POST 请求
- response = requests.post(f"http://10.41.1.57:8071/classify_bert/", json=data)
- # 处理响应
- if response.status_code == 200:
- result = response.json()
- print("接口响应:", result)
- else:
- print(f"请求失败,状态码: {response.status_code}")
- print("错误信息:", response.text)
- def upload_process(file_path, client_id):
- # 表单数据
- data = {
- 'client_id': client_id
- }
- # 发送 POST 请求
- with open(file_path, 'rb') as file:
- name = os.path.basename(file_path)
- files = {
- 'file': (name, file, 'text/plain') # 'text/plain' 是文件类型,可以根据需要修改
- }
- response = requests.post(f"http://10.41.1.57:8071/uploadfile/", data=data, files=files)
- # 处理响应
- if response.status_code == 200:
- result = response.json()
- print("接口响应:", result)
- return result.get('data').get('file_path')
- else:
- print(f"请求失败,状态码: {response.status_code}")
- print("错误信息:", response.text)
- return None
- def both_process_api(path, client_id, one_key, name_column):
- file_path = upload_process(file_path=path, client_id=client_id)
- if file_path:
- cls_process_openai(path=file_path, client_id=client_id, one_key=one_key, name_column=name_column)
- def both_process_bert(path, client_id, name_column):
- file_path = upload_process(file_path=path, client_id=client_id)
- print(file_path)
- if file_path:
- cls_process_bert(path=file_path, client_id=client_id, name_column=name_column)
-
- if __name__ == "__main__":
- # both_process(path="E:/code/name_classify/data_before/人群分类-自建站2024.08.xlsx", client_id='wangdalin', one_key='网店单号', name_column='收货姓名')
- # upload_process(file_path="E:/code/name_classify/data_before/人群分类-自建站2024.08.xlsx", client_id='wangdalin')
- # cls_process_openai(path="./process/wangdalin\\人群分类-自建站2024.08.xlsx",client_id='wangdalin', one_key='网店单号', name_column='收货姓名')
- # cls_process_bert(path="./process/wangdalin/人群分类-自建站2024.08.xlsx",client_id='wangdalin', name_column='收货姓名')
- # both_process_bert(path='/dalin/name_classify/process/wangdalin/1销售Shopify人群分析0909-0915.xlsx', client_id='wangdalin', name_column='姓名')
- both_process_api(path='/dalin/name_classify/test.csv', client_id='wangdalin', one_key='one_key', name_column='name')
- # both_process_bert(path='/dalin/name_classify/test.csv', client_id='wangdalin', name_column='name')
- # cls_process_bert(path='/mnt/e/code/name_classify/process/wangdalin/3销售Shopify人群分析0902-0908_classify_error.xlsx', client_id='wangdalin', name_column='姓名')
-
|