注意:代码环境为服务端 客户端都在本地上,且都在同一个目录下进行编写的,大家可以根据自己的需要进行修改
大体思路:首先将文件的基本信息如文件名称,文件大小,文件的MD5值,封装一下,让后序列化,发送到服务端,服务端对MD5进行检查是否存在自己当前的目录,如果没有存在就行重传,如果已经存在就进行断点续传,中间使用MD5作为文件名进行临时存储文件的内容,这样可以避免同名不同内容的文件的上传,然后等到传输完毕,则将文件名改为自己需要的文件名
客户端代码
import socket
import os
import hashlib
import json
#创建socket对象
up_client = socket.socket()
#连接服务端
up_client.connect(("127.0.0.1",8001))
#求文件的md5值
def file_md5(file_path):
obj = hashlib.md5()
with open(file_path,mode="rb") as f:
for line in f:
obj.update(line)
return obj.hexdigest()
#文件信息
#文件路径
#这里上传的是自己电脑上的文件路径,也可以对代码进行修改,将路径修改为需要输入的路径值,这样就可以进行上传,同时也要注意在服务端也进行修改
file_path = "01.txt"
#获取文件大小
file_size = os.stat(file_path).st_size
file_md5 = file_md5(file_path)
file_name = os.path.basename(file_path)
#自定义协议
file_infor = {"file_name":file_name,"file_size":file_size,"file_md5":file_md5}
up_client.sendall(json.dumps(file_infor).encode("utf-8"))
serve_msg = up_client.recv(1024).decode("utf-8")
if serve_msg=="1002":
print("开始发送")
with open(file_path,mode="rb") as f:
for line in f:
up_client.sendall(line)
print("发送完毕")
else:
exits_size = up_client.recv(1024).decode("utf-8")
print("开始发送")
with open(file_path,mode="rb") as f:
f.seek(exits_size)
for line in f:
up_client.sendall(line)
print("发送完毕")
up_client.close()
服务端代码
import socketserver
import os
import json
import shutil
class UPserver(socketserver.BaseRequestHandler):
def handle(self):
ret = self.request.recv(1024)
file_msg = json.loads(ret)
file_name = file_msg["file_name"]
file_size = file_msg["file_size"]
file_md5 = file_msg["file_md5"]
# 判断上传的文件是否已经存在服务端
#这里的路径修改自己修改就行,这里我用的是我自己电脑上的路径,大家根据自己的需要修改就行
file_exit = os.path.exists("/".join(["PythonStudy", file_md5]))
#1001表示存在,1002表示不存在
file_code = {"exits":"1001","no_exits":"1002"}
# 如果没有存在服务端则进行上传
if not file_exit:
recv_size = 0
total_size = file_size
self.request.sendall(file_code["no_exits"].encode("utf-8"))
print("开始接受")
f = open(file_md5,mode="wb")
while recv_size<total_size:
data = self.request.recv(1024)
f.write(data)
f.flush()
recv_size+=len(data)
f.close()
shutil.move(file_md5,file_name+"2")
print("接受完毕")
#如果存在服务端则进行断点续传
else:
#这里的路径自己进行修改就行,这里我用的是我自己电脑上的路径,大家根据自己的需要修改就行
exit_size = os.stat("/".join("PythonStudy",file_md5))
total_size = file_size
self.request.sendall(exit_size.encode("utf-8"))
f = open(file_md5, mode="ab")
while exit_size<total_size:
data = self.request.recv(1024)
f.write(data)
f.flush()
exit_size+=len(data)
f.close()
shutil.move(file_md5,file_name+"2")
serve = socketserver.ThreadingTCPServer(("127.0.0.1",8001),UPserver)
serve.serve_forever()