网络通信编程学习（11）/ FTP项目（5） ——文件上传和上传断点续存功能

服务端 lib 文件夹下的 main.py 状态码的变化

STATUS_CODE = {
        200: "User login succeeded !", # 用户登陆成功
        201: "Error: wrong username or wrong password !", # 错误用户名或密码
        202: "Action_type does not exist !", # 没有该函数可以执行
        300: "Check Dir !",
        301: "Dir does not exist !",
        400: "Ready to send file !",
        401: "File does not exist !",
        402: "Ready to re_send !",
        500: "File upload done !",
        501: "File upload failed !",
        502: "File exist !"
    }

文件上传功能

服务端 lib 文件夹下的 main.py

    def _put(self, header_dic):
        """从客户端接收文件"""
        filename = header_dic["filename"]
        total_size = header_dic["file_size"]
        res_szie = 0

        # 循环接收文件
        with open(self.current_dir + "\\%s.upload" % filename, "wb") as f:
            while res_szie < total_size:
                data = self.conn.recv(self.MSG_SIZE)
                res_szie += len(data)
                f.write(data)
        if os.path.isfile(self.current_dir + "\\" + filename): # 如果文件存在，文件名加上时间戳
            os.rename(self.current_dir + "\\%s.upload" % filename, self.current_dir + "\\" + filename + "." +str(time.time()))
        else:
            os.rename(self.current_dir + "\\%s.upload" % filename, self.current_dir + "\\" + filename)
       self.send_response(500)

客户端 client 文件夹下的 FTPClient.py

    def put(self, command):
        """向服务端上传文件"""
        # 当前功能只实现一次一个文件上传，所以至少要传入两个参数，一个参数是文件名，一个参数应该是文件在客户端的路径，多余参数不予考虑
        if self.parameter_length_judgment(command=command, min_size=2):
            filename = command[0] # 文件名
            file_path = command[1] # 文件路径
            # 判断文件是否存在
            if os.path.isfile(file_path): # 如果存在，上传文件
                file_size = os.path.getsize(file_path)
                self.create_header_send(action_type="put", filename=filename, file_size=file_size)

                # 记录未上传完成的文件
                print(self.current_dir)
                self.shelve_obj_upload[filename] = (file_path, file_size, self.current_dir) # 记录文件在客户端路径，文件总大小, 上传文件在服务端的路径

                # 进度条
                progress_bar = self.progress_bar(total_size=file_size)
                progress_bar.__next__()
                send_size = 0

                # 发送文件
                with open(file_path, "rb") as f:
                    for line in f:
                        send_size += len(line)
                        progress_bar.send(send_size)
                        self.client.send(line)

                response = self.get_response()

                # 上传成功
                if response.get("status_code") == 500:
                    print(response.get("status_msg"))
                    # 上传成功，删除在  中的数据
                    del self.shelve_obj_upload[filename]

            else: # 文件不存在
                print("File does not exist !")

文件上传断点续存功能

服务端 lib 文件夹下的 main.py

    def _re_put(self, header_dic):
        """重新上传"""
        filename = header_dic.get("filename")
        total_size = header_dic.get("total_size")
        server_path = header_dic.get("server_path")

        if os.path.isfile("%s\\%s.upload" % (server_path, filename)): # 如果未完成上传的文件存在
            recv_size = os.path.getsize("%s\\%s.upload" % (server_path, filename))
            self.send_response(status_code=502, recv_size=recv_size) # 发送已接收的大小

            # 循环接收未上传完成的文件
            with open("%s\\%s.upload" % (server_path, filename), "ab") as f:
                while recv_size < total_size:
                    data = self.conn.recv(self.MSG_SIZE)
                    recv_size += len(data)
                    f.write(data)
            if os.path.isfile("%s\\%s" % (server_path, filename)): # 如果文件存在，文件名加上时间戳
                os.rename("%s\\%s.upload" % (server_path, filename), "%s\\%s.%s" % (server_path, filename, str(time.time())))
            else:
                os.rename("%s\\%s.upload" % (server_path, filename), "%s\\%s" % (server_path, filename))

        else: # 文件不存在
            self.send_response(status_code=401)

客户端 client 文件夹下的 FTPClient.py

    def interactive(self): # 代码增加部分
        """交互指令"""
        if self.auth(): # 登陆成功
            self.unfinished_download_check()
            self.unfinished_upload_check()
        ........

    def unfinished_upload_check(self):
        """上传断点续存"""
        if list(self.shelve_obj_upload.keys()): # 如果有未上传完成的文件
            print("-------Unfinished upload list------")
            for index, filename in enumerate(list(self.shelve_obj_upload.keys())):
                print("%s    %s    %s" % (index, filename,
                                          self.shelve_obj_upload[filename][0]))

            while True:
                choice = input("[select file index to re-put]>>: ").strip()
                if not choice: continue # 如果为空
                if choice == "back": break # 退出
                if choice.isdigit(): # 判断是否是数字
                    choice = int(choice) # 转化为整型
                    if choice <= index: # 选择在范围内
                        filename = list(self.shelve_obj_upload.keys())[choice]

                        file_path = self.shelve_obj_upload[filename][0]
                        total_size = self.shelve_obj_upload[filename][1]
                        server_path = self.shelve_obj_upload[filename][2]

                        if os.path.isfile(file_path) and total_size == os.path.getsize(file_path): # 如果文件存在且文件大小相同
                            # 发送信息
                            self.create_header_send(action_type="re_put", total_size=total_size, server_path=server_path, filename=filename)

                            # 接收消息，包括已接收的文件大小
                            response = self.get_response()
                            if response.get("status_code") == 401: # 文件不存在
                                print(response.get("status_msg"))
                                del self.shelve_obj_upload[filename]
                            else:
                                recv_size = response.get("recv_size") # 准备继续上传

                                # 进度条
                                progress_bar = self.progress_bar(total_size=total_size, res_size=recv_size, last_size=recv_size)
                                progress_bar.__next__()

                                # 继续上传
                                with open(file_path, "rb") as f:
                                    f.seek(recv_size)
                                    for line in f:
                                        self.client.send(line)
                                        recv_size += len(line)
                                        progress_bar.send(recv_size)
                                print("File re_upload done !")
                                del self.shelve_obj_upload[filename]
                        else:
                            print("File does not exist !")
                            del self.shelve_obj_upload[filename]

运行结果