100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
import urllib.request
|
|
import argparse
|
|
import sys
|
|
|
|
def openfile(filename):
|
|
try:
|
|
with open(filename, "r", encoding='utf-8') as f:
|
|
content = f.read()
|
|
print(f"✅ 文件读取成功，内容长度: {len(content)} 字符")
|
|
return content
|
|
except FileNotFoundError as e:
|
|
print(f"❌ 错误：指定的文件不存在 -> {filename}")
|
|
raise e
|
|
except PermissionError as e:
|
|
print(f"❌ 错误：没有权限读取文件 -> {filename}")
|
|
raise e
|
|
except Exception as e:
|
|
print(f"❌ 错误：读取文件失败 -> {str(e)}")
|
|
raise e
|
|
|
|
def httppost(url: str, msg: str) -> str:
|
|
try:
|
|
# 核心转换：字符串 → 字节流（原生库必须要求）
|
|
post_body_bytes = msg.encode("utf-8")
|
|
# 发送POST请求，timeout=10秒防止卡死
|
|
response = urllib.request.urlopen(url, data=post_body_bytes, timeout=2)
|
|
|
|
# 解析响应结果
|
|
if response.getcode() == 200:
|
|
resp_content = response.read().decode("utf-8")
|
|
print("\n🎉 请求上传成功！HTTP状态码: 200")
|
|
print(f"🎉 服务端响应内容:\n{resp_content}")
|
|
else:
|
|
resp_content = f'---{str(response.getcode())}---'
|
|
print(f"\n⚠️ 请求上传完成，但服务端返回异常状态码: {response.getcode()}")
|
|
print(f"⚠️ 服务端响应内容: {resp_content}")
|
|
|
|
return resp_content
|
|
|
|
except urllib.error.URLError as e:
|
|
print(f"\n❌ 请求失败：网络错误/连接超时/端口未开放 -> {str(e)}")
|
|
raise e
|
|
except urllib.error.HTTPError as e:
|
|
print(f"\n❌ 请求失败：HTTP错误 -> 状态码: {e.code}, 详情: {e.read().decode('utf-8')}")
|
|
raise e
|
|
except Exception as e:
|
|
print(f"\n❌ 请求失败：未知错误 -> {str(e)}")
|
|
raise e
|
|
|
|
def upload_file(file: str, host: str, port: int) -> str:
|
|
"""抽象的URL访问函数，用于上传文件到指定服务器
|
|
|
|
Args:
|
|
file: 待上传的文件路径
|
|
host: 目标服务器IP
|
|
port: 目标服务器端口
|
|
"""
|
|
try:
|
|
# 读取文件内容
|
|
msg = openfile(file)
|
|
# 构造URL
|
|
url = f"http://{host}:{port}/api/v1/mybot-sample"
|
|
# 发送POST请求
|
|
resp = httppost(url, msg)
|
|
except Exception as e:
|
|
print(f"❌ 错误：上传文件失败 -> {str(e)}")
|
|
resp = str(e)
|
|
return resp
|
|
|
|
def upload_file_and_train(file: str, host: str, port: int) -> str:
|
|
"""抽象的URL访问函数，用于上传文件到指定服务器
|
|
|
|
Args:
|
|
file: 待上传的文件路径
|
|
host: 目标服务器IP
|
|
port: 目标服务器端口
|
|
"""
|
|
try:
|
|
# 读取文件内容
|
|
msg = openfile(file)
|
|
# 构造URL
|
|
url = f"http://{host}:{port}/api/v1/mybot-nn"
|
|
# 发送POST请求
|
|
resp = httppost(url, msg)
|
|
except Exception as e:
|
|
print(f"❌ 错误：上传文件失败 -> {str(e)}")
|
|
resp = str(e)
|
|
return resp
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="python 远程上传文本内容")
|
|
parser.add_argument("file", type=str, help="待上传的文件")
|
|
parser.add_argument('--port', type=int, default=9203, help="目标服务器监听端口，默认9203")
|
|
parser.add_argument("--host", type=str, default="127.0.0.1", help="目标服务器的IP，默认本地")
|
|
args =parser.parse_args()
|
|
resp = upload_file(args.file, args.host, args.port)
|
|
#resp = upload_file_and_train(args.file, args.host, args.port)
|
|
print(resp)
|