python实现FTP文件上传

注意：代码环境为服务端 客户端都在本地上，且都在同一个目录下进行编写的，大家可以根据自己的需要进行修改

大体思路：首先将文件的基本信息如文件名称，文件大小，文件的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()