接口自动化查删

import requests
from urllib3 import encode_multipart_formdata
import json
import make_js_file as mkf
import config

headers = {"Content-Type": "multipart/form-data"}

filepath = "D:/1、项目资料/RPA/3、We法/协议测试文件/标准保密协议.docx"


def add_user():

    import requests

    formdata = {
            "file_id": "",
            "file_name": "",
            "file_real_name": "",
            "user_id": "admin"
        }
    headers = {
        'User-Agent': 'Chrome/71.0.3578.98 Safari/537.36'}
    files = {
        'file': ('file', open("D:/1、项目资料/RPA/3、We法/协议测试文件/标准保密协议.docx", 'rb'))
    }
    url = config.url + "/xyjc/upload_file/"
    print(headers)
    for i in range(100):
        formdata["file_id"] = "test_file" + str(i + 1)
        formdata["file_name"] = "测试协议" + str(i + 1)
        formdata["file_real_name"] = "标准保密协议"
        r = requests.post(url,data=formdata, headers=headers, files=files)
    r.raise_for_status()
    print(r.text)
    print(r.status_code)

def search_user():
    search_data = {
        "file_id": None,
        "file_name": None,
        "search_type": "0",
        "list_min": 1,
        "list_max": 2000,
    }
    url = config.url + "/xyjc/search_file/"
    response = requests.post(url, data=search_data, headers=config.headers)
    data_dict = response.json()
    user_list = []
    for one in data_dict['data']:
        if "测试协议" in one['file_name']:
            print(one)
            user_list.append(one['x_uid'])
    print(user_list)
    return user_list

def del_user():
    url = config.url + "/xyjc/delete_file/"
    del_data = {
        "x_uid": None
    }
    del_user_list = search_user()
    for one in del_user_list:
        del_data["x_uid"] = one
        response = requests.post(url, data=del_data, headers=config.headers)
        print(response.json())

if __name__ == '__main__':
    add_user()
    # search_user()
    # del_user()
    # mkf.make_json_file("data1.json", data_dict)