58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
|
import requests
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
# 配置
|
|||
|
|
BASE_URL = "http://localhost:8080"
|
|||
|
|
LOGIN_URL = f"{BASE_URL}/login/test"
|
|||
|
|
|
|||
|
|
# 登录获取token
|
|||
|
|
print("登录系统...")
|
|||
|
|
login_data = {
|
|||
|
|
"username": "admin",
|
|||
|
|
"password": "admin123"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = requests.post(LOGIN_URL, json=login_data)
|
|||
|
|
token = response.json().get("token")
|
|||
|
|
|
|||
|
|
# 测试不同的导入方式
|
|||
|
|
headers = {'Authorization': f'Bearer {token}'}
|
|||
|
|
|
|||
|
|
print("\n测试1: 直接POST请求（无文件）")
|
|||
|
|
response = requests.post(
|
|||
|
|
f"{BASE_URL}/dpc/intermediary/importPersonData",
|
|||
|
|
headers=headers
|
|||
|
|
)
|
|||
|
|
print(f"状态码: {response.status_code}")
|
|||
|
|
print(f"响应: {response.text[:200]}")
|
|||
|
|
|
|||
|
|
print("\n测试2: 带文件的POST请求（URL参数）")
|
|||
|
|
files = {
|
|||
|
|
'file': ('test.xlsx', open('doc/个人中介黑名单测试数据_1000条.xlsx', 'rb'), 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
|
|||
|
|
}
|
|||
|
|
response = requests.post(
|
|||
|
|
f"{BASE_URL}/dpc/intermediary/importPersonData?updateSupport=false",
|
|||
|
|
files=files,
|
|||
|
|
headers=headers
|
|||
|
|
)
|
|||
|
|
print(f"状态码: {response.status_code}")
|
|||
|
|
print(f"响应: {response.text[:500]}")
|
|||
|
|
files['file'][1].close()
|
|||
|
|
|
|||
|
|
print("\n测试3: 带文件的POST请求（Form数据）")
|
|||
|
|
files = {
|
|||
|
|
'file': ('test.xlsx', open('doc/个人中介黑名单测试数据_1000条.xlsx', 'rb'), 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
|
|||
|
|
}
|
|||
|
|
data = {
|
|||
|
|
'updateSupport': 'false'
|
|||
|
|
}
|
|||
|
|
response = requests.post(
|
|||
|
|
f"{BASE_URL}/dpc/intermediary/importPersonData",
|
|||
|
|
files=files,
|
|||
|
|
data=data,
|
|||
|
|
headers=headers
|
|||
|
|
)
|
|||
|
|
print(f"状态码: {response.status_code}")
|
|||
|
|
print(f"响应: {response.text[:500]}")
|
|||
|
|
files['file'][1].close()
|