name_classify/client.py

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='姓名')